ident.c
上传用户:pycemail
上传日期:2007-01-04
资源大小:329k
文件大小:4k
源码类别:

Ftp客户端

开发平台:

Unix_Linux

  1. /*
  2.  * ProFTPD - FTP server daemon
  3.  * Copyright (c) 1997, 1998 Public Flood Software
  4.  *  
  5.  * This program is free software; you can redistribute it and/or modify
  6.  * it under the terms of the GNU General Public License as published by
  7.  * the Free Software Foundation; either version 2 of the License, or
  8.  * (at your option) any later version.
  9.  *
  10.  * This program is distributed in the hope that it will be useful,
  11.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13.  * GNU General Public License for more details.
  14.  *
  15.  * You should have received a copy of the GNU General Public License
  16.  * along with this program; if not, write to the Free Software
  17.  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307, USA.
  18.  */
  19. /*
  20.  * Ident (RFC1413) protocol support
  21.  */
  22. #include "conf.h"
  23. static int ident_timeout;
  24. static IOFILE *io;
  25. static int _ident_timeout(CALLBACK_FRAME)
  26. {
  27.   ident_timeout++;
  28.   if(io)
  29.     /* Abort the IOFILE, which will cause io_poll (and thus io_read) to
  30.      * also abort.  This is similar to the way data connects are aborted
  31.      */
  32.     io_abort(io);
  33.     
  34.   return 0;
  35. }
  36. char *get_ident(pool *p,conn_t *c)
  37. {
  38.   char *ret = "UNKNOWN";
  39.   pool *tmpp;
  40.   conn_t *ident_conn,*ident_io;
  41.   char buf[256],*tok,*tmp;
  42.   int timer,i = 0;
  43.   int ident_port = inet_getservport(p,"ident","tcp");
  44.   tmpp = make_sub_pool(p);
  45.   ident_timeout = 0;
  46.   io = NULL;
  47.   if(ident_port == -1) {
  48.     destroy_pool(tmpp);
  49.     return pstrdup(p,ret);    
  50.   }
  51.  
  52.   /* Set up our timer before going any further */
  53.   timer = add_timer(TUNABLE_TIMEOUTIDENT,-1,NULL,(callback_t)_ident_timeout);
  54.   if(timer <= 0) {
  55.     destroy_pool(tmpp);
  56.     return pstrdup(p,ret);
  57.   }
  58.   
  59.   ident_conn = inet_create_connection(tmpp,NULL,-1,c->local_ipaddr,INPORT_ANY,FALSE);
  60.   inet_setnonblock(tmpp,ident_conn);
  61.   i = inet_connect_nowait(tmpp,ident_conn,c->remote_ipaddr,ident_port);
  62.   if(i < 0) {
  63.     remove_timer(timer,NULL);
  64.     inet_close(tmpp,ident_conn);
  65.     destroy_pool(tmpp);
  66.     return pstrdup(p,ret);
  67.   }
  68.   
  69.   if(!i) { /* Not yet connected */
  70.     io = io_open(p,ident_conn->listen_fd,IO_READ);
  71.     io_set_poll_sleep(io,1);
  72.     switch(io_poll(io)) {
  73.     case 1: /* Abort, Timeout? */
  74. if(ident_timeout) {
  75.           remove_timer(timer,NULL);
  76.                  io_close(io);
  77.   destroy_pool(tmpp);
  78.   return pstrdup(p,ret);
  79. }
  80. break;
  81.     case -1: /* Error */
  82. remove_timer(timer,NULL);
  83. io_close(io);
  84. destroy_pool(tmpp);
  85. return pstrdup(p,ret);
  86.     default: /* connected */
  87. ident_conn->mode = CM_OPEN;
  88. inet_get_conn_info(ident_conn,ident_conn->listen_fd);
  89. break;
  90.     }
  91.   }
  92.   ident_io = inet_openrw(tmpp,ident_conn,NULL,-1,-1,-1,FALSE);
  93.   io = ident_io->inf;
  94.   inet_setnonblock(tmpp,ident_io);
  95.   io_set_poll_sleep(ident_io->inf,1);
  96.   io_set_poll_sleep(ident_io->outf,1);
  97.   io_printf(ident_io->outf,"%d, %drn",c->remote_port,c->local_port);
  98.   /* If the timer fires while in io_gets, io_gets will simply return
  99.    * either a partial string, or NULL.  This works because _ident_timeout
  100.    * aborts the IOFILE we are reading on.  io_set_poll_sleep() is used
  101.    * to make sure significant delays don't occur on systems that
  102.    * automatically restart syscalls after the SIGALRM signal.
  103.    */
  104.   
  105.   if(io_gets(buf,sizeof(buf),ident_io->inf)) {
  106.     strip_end(buf,"rn");
  107.     
  108.     tmp = buf;
  109.     tok = get_token(&tmp,":");
  110.     if(tok && (tok = get_token(&tmp,":"))) {
  111.       while(*tok && isspace((UCHAR)*tok)) tok++;
  112.       strip_end(tok," t");
  113.       if(strcasecmp(tok,"ERROR") == 0) {
  114.         if(tmp) {
  115.           while(*tmp && isspace((UCHAR)*tmp)) tmp++;
  116.   strip_end(tmp," t");
  117.           if(strcasecmp(tmp,"HIDDEN-USER") == 0)
  118.             ret = "HIDDEN-USER";
  119.         }
  120.       } else if(strcasecmp(tok,"USERID") == 0) {
  121.         if(tmp && (tok = get_token(&tmp,":"))) {
  122.           if(tmp) {
  123.             while(*tmp && isspace((UCHAR)*tmp)) tmp++;
  124.             strip_end(tmp," t");
  125.             ret = tmp;
  126.           }
  127.         }
  128.       }
  129.     }
  130.   }
  131.   remove_timer(timer,NULL);
  132.   inet_close(tmpp,ident_io);
  133.   inet_close(tmpp,ident_conn);
  134.   destroy_pool(tmpp);
  135.   return pstrdup(p,ret);
  136. }