authuser.c
上传用户:zibowangxu
上传日期:2007-01-04
资源大小:331k
文件大小:6k
源码类别:

Ftp客户端

开发平台:

Unix_Linux

  1. /****************************************************************************  
  2.  
  3.   Copyright (c) 1999 WU-FTPD Development Group.  
  4.   All rights reserved.
  5.   
  6.   Portions Copyright (c) 1980, 1985, 1988, 1989, 1990, 1991, 1993, 1994
  7.     The Regents of the University of California.
  8.   Portions Copyright (c) 1993, 1994 Washington University in Saint Louis.
  9.   Portions Copyright (c) 1996, 1998 Berkeley Software Design, Inc.
  10.   Portions Copyright (c) 1989 Massachusetts Institute of Technology.
  11.   Portions Copyright (c) 1998 Sendmail, Inc.
  12.   Portions Copyright (c) 1983, 1995, 1996, 1997 Eric P.  Allman.
  13.   Portions Copyright (c) 1997 by Stan Barber.
  14.   Portions Copyright (c) 1997 by Kent Landfield.
  15.   Portions Copyright (c) 1991, 1992, 1993, 1994, 1995, 1996, 1997
  16.     Free Software Foundation, Inc.  
  17.  
  18.   Use and distribution of this software and its source code are governed 
  19.   by the terms and conditions of the WU-FTPD Software License ("LICENSE").
  20.  
  21.   If you did not receive a copy of the license, it may be obtained online
  22.   at http://www.wu-ftpd.org/license.html.
  23.  
  24.   $Id: authuser.c,v 1.12 1999/09/28 15:03:06 wuftpd Exp $
  25.  
  26. ****************************************************************************/
  27. #include "../src/config.h"
  28. #ifdef USE_RFC931
  29. #include <stdlib.h>
  30. #include <stdio.h>
  31. #include <ctype.h>
  32. #include <errno.h>
  33. #include <fcntl.h>
  34. #include <limits.h>
  35. #include <netdb.h>
  36. #include <netinet/in.h>
  37. #ifdef AIX
  38. #include <netinet/if_ether.h>
  39. #include <net/if_dl.h>
  40. #endif
  41. #include <arpa/inet.h>
  42. #include <sys/socket.h>
  43. #include <sys/stat.h>
  44. #include <signal.h>
  45. extern unsigned int timeout_rfc931;
  46. extern int errno;
  47. #include "authuser.h"
  48. unsigned short auth_tcpport = 113;
  49. #define SIZ 500 /* various buffers */
  50. static int usercmp(register char *u, register char *v)
  51. {
  52.     /* is it correct to consider Foo and fOo the same user? yes */
  53.     /* but the function of this routine may change later */
  54.     while (*u && *v)
  55. if (tolower(*u) != tolower(*v))
  56.     return tolower(*u) - tolower(*v);
  57. else
  58.     ++u, ++v;
  59.     return *u || *v;
  60. }
  61. static char authline[SIZ];
  62. char *auth_xline(register char *user, register int fd, register long unsigned int *in)
  63. {
  64.     unsigned short local;
  65.     unsigned short remote;
  66.     register char *ruser;
  67.     if (auth_fd(fd, in, &local, &remote) == -1)
  68. return 0;
  69.     ruser = auth_tcpuser(*in, local, remote);
  70.     if (!ruser)
  71. return 0;
  72.     if (!user)
  73. user = ruser; /* forces X-Auth-User */
  74.     (void) sprintf(authline,
  75.     (usercmp(ruser, user) ? "X-Forgery-By: %s" : "X-Auth-User: %s"),
  76.    ruser);
  77.     return authline;
  78. }
  79. int auth_fd(register int fd, register long unsigned int *in, register short unsigned int *local, register short unsigned int *remote)
  80. {
  81.     struct sockaddr_in sa;
  82. #if defined(UNIXWARE) || defined(AIX)
  83.     size_t dummy;
  84. #else
  85.     int dummy;
  86. #endif
  87.     dummy = sizeof(sa);
  88.     if (getsockname(fd, (struct sockaddr *) &sa, &dummy) == -1)
  89. return -1;
  90.     if (sa.sin_family != AF_INET) {
  91. errno = EAFNOSUPPORT;
  92. return -1;
  93.     }
  94.     *local = ntohs(sa.sin_port);
  95.     dummy = sizeof(sa);
  96.     if (getpeername(fd, (struct sockaddr *) &sa, &dummy) == -1)
  97. return -1;
  98.     *remote = ntohs(sa.sin_port);
  99.     *in = sa.sin_addr.s_addr;
  100.     return 0;
  101. }
  102. static char ruser[SIZ];
  103. static char realbuf[SIZ];
  104. static char *buf;
  105. static int fdAuth = -1;
  106. static void timout(int sig)
  107. {
  108.     if (fdAuth != -1) {
  109. close(fdAuth);
  110. fdAuth = -1;
  111.     }
  112. }
  113. char *auth_tcpuser(register long unsigned int in, register short unsigned int local, register short unsigned int remote)
  114. {
  115.     struct sockaddr_in sa;
  116.     register int buflen;
  117.     register int w;
  118.     register int saveerrno;
  119.     char ch;
  120.     unsigned short rlocal;
  121.     unsigned short rremote;
  122.     extern struct sockaddr_in ctrl_addr;
  123.     int on = 1;
  124.     buf = realbuf;
  125.     (void) sprintf(buf, "%u , %urn", (unsigned int) remote, (unsigned int) local);
  126.     /* note the reversed order---the example in the RFC is misleading */
  127.     buflen = strlen(buf);
  128.     if ((fdAuth = socket(AF_INET, SOCK_STREAM, 0)) == -1)
  129. return 0;
  130.     setsockopt(fdAuth, SOL_SOCKET, SO_REUSEADDR, (char *) &on, sizeof(on)); /*Try */
  131.     sa = ctrl_addr;
  132.     sa.sin_port = htons(0);
  133.     bind(fdAuth, (struct sockaddr *) &sa, sizeof(sa)); /* may as well try ... */
  134.     sa.sin_family = AF_INET;
  135.     sa.sin_port = htons(auth_tcpport);
  136.     sa.sin_addr.s_addr = in;
  137.     signal(SIGALRM, timout);
  138. #if defined(LINUX)
  139.     /*
  140.      * The default behaviour of signal() under modern glibc implementations is
  141.      * to have SA_RESTART on by default.  This prevents the alarm() call from
  142.      * interrupting the connect() call, which then, in the case of a firewall,
  143.      * causes the connect() to take minutes to time out, in turn leading to
  144.      * frustrated inbound clients.
  145.      */
  146.     {
  147. struct sigaction old, new;
  148. sigaction(SIGALRM, NULL, &old);
  149. new = old;
  150. new.sa_flags &= ~SA_RESTART;
  151. sigaction(SIGALRM, &new, &old);
  152.     }
  153. #endif
  154.     alarm(timeout_rfc931);
  155.     if (connect(fdAuth, (struct sockaddr *) &sa, sizeof(sa)) == -1) {
  156. saveerrno = errno;
  157. alarm(0);
  158. if (fdAuth != -1) {
  159.     close(fdAuth);
  160.     fdAuth = -1;
  161. }
  162. errno = saveerrno;
  163. return 0;
  164.     }
  165.     while ((w = write(fdAuth, buf, buflen)) < buflen)
  166. if (w == -1) { /* should we worry about 0 as well? */
  167.     saveerrno = errno;
  168.     alarm(0);
  169.     if (fdAuth != -1) {
  170. close(fdAuth);
  171. fdAuth = -1;
  172.     }
  173.     errno = saveerrno;
  174.     return 0;
  175. }
  176. else {
  177.     buf += w;
  178.     buflen -= w;
  179. }
  180.     buf = realbuf;
  181.     while ((w = read(fdAuth, &ch, 1)) == 1) {
  182. *buf = ch;
  183. if ((ch != ' ') && (ch != 't') && (ch != 'r'))
  184.     ++buf;
  185. if ((buf - realbuf == sizeof(realbuf) - 1) || (ch == 'n'))
  186.     break;
  187.     }
  188.     saveerrno = errno;
  189.     alarm(0);
  190.     if (fdAuth != -1) {
  191. close(fdAuth);
  192. fdAuth = -1;
  193.     }
  194.     errno = saveerrno;
  195.     if (w == -1)
  196. return 0;
  197.     *buf = '';
  198. /* H* fix: limit scanf of returned identd string. */
  199.     if (sscanf(realbuf, "%hd,%hd: USERID :%*[^:]:%400s",
  200.        &rremote, &rlocal, ruser) < 3) {
  201. errno = EIO;
  202. /* makes sense, right? well, not when USERID failed to match
  203.    ERROR but there's no good error to return in that case */
  204. return 0;
  205.     }
  206.     if ((remote != rremote) || (local != rlocal)) {
  207. errno = EIO;
  208. return 0;
  209.     }
  210.     /* XXX: we're not going to do any backslash processing */
  211.     return ruser;
  212. }
  213. #else /* USE_RFC931 */
  214. int auth_dummy_variable = 0; /* To keep compilers quiet */
  215. #endif /* USE_RFC931 */