开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > TCP/IP协议栈 > 查看源码
ftpserv.c
资源名称：export.zip　[点击查看]
上传用户：hepax88
上传日期：2007-01-03
资源大小：1101k
文件大小：16k
源码类别：

TCP/IP协议栈

开发平台：

Visual C++

ftpserv.c：源码内容
  1. /* Internet FTP Server
  2.  * Copyright 1991 Phil Karn, KA9Q
  3.  */
  4. #include <stdio.h>
  5. #include <ctype.h>
  6. #include <time.h>
  7. #ifdef __TURBOC__
  8. #include <io.h>
  9. #include <dir.h>
  10. #endif
  11. #include "global.h"
  12. #include "mbuf.h"
  13. #include "proc.h"
  14. #include "socket.h"
  15. #include "dirutil.h"
  16. #include "commands.h"
  17. #include "files.h"
  18. #include "ftp.h"
  19. #include "ftpserv.h"
  20. #include "md5.h"
  22. static void ftpserv(int s,void *unused,void *p);
  23. static int pport(struct sockaddr_in *sock,char *arg);
  24. static void ftplogin(struct ftpserv *ftp,char *pass);
  25. static int sendit(struct ftpserv *ftp,char *command,char *file);
  26. static int recvit(struct ftpserv *ftp,char *command,char *file);
  28. /* Command table */
  29. static char *commands[] = {
  30. "user",
  31. "acct",
  32. "pass",
  33. "type",
  34. "list",
  35. "cwd",
  36. "dele",
  37. "name",
  38. "quit",
  39. "retr",
  40. "stor",
  41. "port",
  42. "nlst",
  43. "pwd",
  44. "xpwd", /* For compatibility with 4.2BSD */
  45. "mkd ",
  46. "xmkd", /* For compatibility with 4.2BSD */
  47. "xrmd", /* For compatibility with 4.2BSD */
  48. "rmd ",
  49. "stru",
  50. "mode",
  51. "syst",
  52. "xmd5",
  53. "xcwd",
  54. NULL
  55. };
  57. /* Response messages */
  58. static char banner[] = "220 %s FTP version %s ready at %sn";
  59. static char badcmd[] = "500 Unknown command '%s'n";
  60. static char binwarn[] = "100 Warning: type is ASCII and %s appears to be binaryn";
  61. static char unsupp[] = "500 Unsupported command or optionn";
  62. static char givepass[] = "331 Enter PASS commandn";
  63. static char logged[] = "230 Logged inn";
  64. static char typeok[] = "200 Type %s OKn";
  65. static char only8[] = "501 Only logical bytesize 8 supportedn";
  66. static char deleok[] = "250 File deletedn";
  67. static char mkdok[] = "200 MKD okn";
  68. static char delefail[] = "550 Delete failed: %sn";
  69. static char pwdmsg[] = "257 "%s" is current directoryn";
  70. static char badtype[] = "501 Unknown type "%s"n";
  71. static char badport[] = "501 Bad port syntaxn";
  72. static char unimp[] = "502 Command not yet implementedn";
  73. static char bye[] = "221 Goodbye!n";
  74. static char nodir[] = "553 Can't read directory "%s": %sn";
  75. static char cantopen[] = "550 Can't read file "%s": %sn";
  76. static char sending[] = "150 Opening data connection for %s %sn";
  77. static char cantmake[] = "553 Can't create "%s": %sn";
  78. static char writerr[] = "552 Write error: %sn";
  79. static char portok[] = "200 Port command okayn";
  80. static char rxok[] = "226 File received OKn";
  81. static char txok[] = "226 File sent OKn";
  82. static char noperm[] = "550 Permission deniedn";
  83. static char noconn[] = "425 Data connection resetn";
  84. static char lowmem[] = "421 System overloaded, try again latern";
  85. static char notlog[] = "530 Please log in with USER and PASSn";
  86. static char userfirst[] = "503 Login with USER first.n";
  87. static char okay[] = "200 Okn";
  88. static char syst[] = "215 %s Type: L%d Version: %sn";
  90. /* Start up FTP service */
  91. int
  92. ftpstart(argc,argv,p)
  93. int argc;
  94. char *argv[];
  95. void *p;
  96. {
  97. uint16 port;
  99. if(argc < 2)
  100. port = IPPORT_FTP;
  101. else
  102. port = atoi(argv[1]);
  104. return start_tcp(port,"FTP Server",ftpserv,2048);
  105. }
  106. static void
  107. ftpserv(s,n,p)
  108. int s; /* Socket with user connection */
  109. void *n;
  110. void *p;
  111. {
  112. struct ftpserv ftp;
  113. char **cmdp,buf[512],*arg,*cp,*cp1,*file,*mode;
  114. long t;
  115. int i;
  116. struct sockaddr_in socket;
  118. memset(&ftp,0,sizeof(ftp)); /* Start with clear slate */
  119. ftp.control = fdopen(s,"r+t");
  120. sockowner(fileno(ftp.control),Curproc); /* We own it now */
  121. setvbuf(ftp.control,NULL,_IOLBF,BUFSIZ);
  122. if(availmem() != 0){
  123. fprintf(ftp.control,lowmem);
  124. fclose(ftp.control);
  125. return;
  126. }
  128. fclose(stdin);
  129. stdin = fdup(ftp.control);
  130. fclose(stdout);
  131. stdout = fdup(ftp.control);
  133. /* Set default data port */
  134. i = SOCKSIZE;
  135. getpeername(fileno(ftp.control),(struct sockaddr *)&socket,&i);
  136. socket.sin_port = IPPORT_FTPD;
  137. ASSIGN(ftp.port,socket);
  139. logmsg(fileno(ftp.control),"open FTP");
  140. time(&t);
  141. cp = ctime(&t);
  142. if((cp1 = strchr(cp,'n')) != NULL)
  143. *cp1 = '';
  144. fprintf(ftp.control,banner,Hostname,Version,cp);
  145. loop: fflush(ftp.control);
  146. if((fgets(buf,sizeof(buf),ftp.control)) == NULL){
  147. /* He closed on us */
  148. goto finish;
  149. }
  150. if(strlen(buf) == 0){
  151. /* Can't be a legal FTP command */
  152. fprintf(ftp.control,badcmd,buf);
  153. goto loop;
  154. }
  155. rip(buf);
  156. /* Translate first word to lower case */
  157. for(cp = buf;*cp != ' ' && *cp != '';cp++)
  158. *cp = tolower(*cp);
  159. /* Find command in table; if not present, return syntax error */
  160. for(cmdp = commands;*cmdp != NULL;cmdp++)
  161. if(strncmp(*cmdp,buf,strlen(*cmdp)) == 0)
  162. break;
  163. if(*cmdp == NULL){
  164. fprintf(ftp.control,badcmd,buf);
  165. goto loop;
  166. }
  167. /* Allow only USER, PASS and QUIT before logging in */
  168. if(ftp.cd == NULL || ftp.path == NULL){
  169. switch(cmdp-commands){
  170. case USER_CMD:
  171. case PASS_CMD:
  172. case QUIT_CMD:
  173. break;
  174. default:
  175. fprintf(ftp.control,notlog);
  176. goto loop;
  177. }
  178. }
  179. arg = &buf[strlen(*cmdp)];
  180. while(*arg == ' ')
  181. arg++;
  183. /* Execute specific command */
  184. switch(cmdp-commands){
  185. case USER_CMD:
  186. free(ftp.username);
  187. ftp.username = strdup(arg);
  188. fprintf(ftp.control,givepass);
  189. break;
  190. case TYPE_CMD:
  191. switch(arg[0]){
  192. case 'A':
  193. case 'a': /* Ascii */
  194. ftp.type = ASCII_TYPE;
  195. fprintf(ftp.control,typeok,arg);
  196. break;
  197. case 'l':
  198. case 'L':
  199. while(*arg != ' ' && *arg != '')
  200. arg++;
  201. if(*arg == '' || *++arg != '8'){
  202. fprintf(ftp.control,only8);
  203. break;
  204. }
  205. ftp.type = LOGICAL_TYPE;
  206. ftp.logbsize = 8;
  207. fprintf(ftp.control,typeok,arg);
  208. break;
  209. case 'B':
  210. case 'b': /* Binary */
  211. case 'I':
  212. case 'i': /* Image */
  213. ftp.type = IMAGE_TYPE;
  214. fprintf(ftp.control,typeok,arg);
  215. break;
  216. default: /* Invalid */
  217. fprintf(ftp.control,badtype,arg);
  218. break;
  219. }
  220. break;
  221. case QUIT_CMD:
  222. fprintf(ftp.control,bye);
  223. goto finish;
  224. case RETR_CMD:
  225. file = pathname(ftp.cd,arg);
  226. switch(ftp.type){
  227. case IMAGE_TYPE:
  228. case LOGICAL_TYPE:
  229. mode = READ_BINARY;
  230. break;
  231. case ASCII_TYPE:
  232. mode = READ_TEXT;
  233. break;
  234. }
  235. if(!permcheck(ftp.path,ftp.perms,RETR_CMD,file)){
  236.   fprintf(ftp.control,noperm);
  237. } else if((ftp.fp = fopen(file,mode)) == NULL){
  238. fprintf(ftp.control,cantopen,file,sys_errlist[errno]);
  239. } else {
  240. logmsg(fileno(ftp.control),"RETR %s",file);
  241. if(ftp.type == ASCII_TYPE && isbinary(ftp.fp)){
  242. fprintf(ftp.control,binwarn,file);
  243. }
  244. sendit(&ftp,"RETR",file);
  245. }
  246. FREE(file);
  247. break;
  248. case STOR_CMD:
  249. file = pathname(ftp.cd,arg);
  250. switch(ftp.type){
  251. case IMAGE_TYPE:
  252. case LOGICAL_TYPE:
  253. mode = WRITE_BINARY;
  254. break;
  255. case ASCII_TYPE:
  256. mode = WRITE_TEXT;
  257. break;
  258. }
  259. if(!permcheck(ftp.path,ftp.perms,STOR_CMD,file)){
  260.   fprintf(ftp.control,noperm);
  261. } else if((ftp.fp = fopen(file,mode)) == NULL){
  262. fprintf(ftp.control,cantmake,file,sys_errlist[errno]);
  263. } else {
  264. logmsg(fileno(ftp.control),"STOR %s",file);
  265. recvit(&ftp,"STOR",file);
  266. }
  267. FREE(file);
  268. break;
  269. case PORT_CMD:
  270. if(pport(&ftp.port,arg) == -1){
  271. fprintf(ftp.control,badport);
  272. } else {
  273. fprintf(ftp.control,portok);
  274. }
  275. break;
  276. #ifndef CPM
  277. case LIST_CMD:
  278. file = pathname(ftp.cd,arg);
  279. if(!permcheck(ftp.path,ftp.perms,RETR_CMD,file)){
  280.   fprintf(ftp.control,noperm);
  281. } else if((ftp.fp = dir(file,1)) == NULL){
  282. fprintf(ftp.control,nodir,file,sys_errlist[errno]);
  283. } else {
  284. sendit(&ftp,"LIST",file);
  285. }
  286. FREE(file);
  287. break;
  288. case NLST_CMD:
  289. file = pathname(ftp.cd,arg);
  290. if(!permcheck(ftp.path,ftp.perms,RETR_CMD,file)){
  291.   fprintf(ftp.control,noperm);
  292. } else if((ftp.fp = dir(file,0)) == NULL){
  293. fprintf(ftp.control,nodir,file,sys_errlist[errno]);
  294. } else {
  295. sendit(&ftp,"NLST",file);
  296. }
  297. FREE(file);
  298. break;
  299. case XCWD_CMD:
  300. case CWD_CMD:
  301. file = pathname(ftp.cd,arg);
  302. if(!permcheck(ftp.path,ftp.perms,RETR_CMD,file)){
  303.   fprintf(ftp.control,noperm);
  304. FREE(file);
  305. #ifdef MSDOS
  306. /* Don'tcha just LOVE %%$#@!! MS-DOS? */
  307. } else if(strcmp(file,"/") == 0 || access(file,0) == 0){
  308. #else
  309. } else if(access(file,0) == 0){ /* See if it exists */
  310. #endif
  311. /* Succeeded, record in control block */
  312. free(ftp.cd);
  313. ftp.cd = file;
  314. fprintf(ftp.control,pwdmsg,file);
  315. } else {
  316. /* Failed, don't change anything */
  317. fprintf(ftp.control,nodir,file,sys_errlist[errno]);
  318. FREE(file);
  319. }
  320. break;
  321. case XPWD_CMD:
  322. case PWD_CMD:
  323. fprintf(ftp.control,pwdmsg,ftp.cd);
  324. break;
  325. #else
  326. case LIST_CMD:
  327. case NLST_CMD:
  328. case CWD_CMD:
  329. case XCWD_CMD:
  330. case XPWD_CMD:
  331. case PWD_CMD:
  332. #endif
  333. case ACCT_CMD:
  334. fprintf(ftp.control,unimp);
  335. break;
  336. case DELE_CMD:
  337. file = pathname(ftp.cd,arg);
  338. if(!permcheck(ftp.path,ftp.perms,DELE_CMD,file)){
  339.   fprintf(ftp.control,noperm);
  340. } else if(unlink(file) == 0){
  341. logmsg(fileno(ftp.control),"DELE %s",file);
  342. fprintf(ftp.control,deleok);
  343. } else {
  344. fprintf(ftp.control,delefail,sys_errlist[errno]);
  345. }
  346. FREE(file);
  347. break;
  348. case PASS_CMD:
  349. if(ftp.username == NULL)
  350. fprintf(ftp.control,userfirst);
  351. else
  352. ftplogin(&ftp,arg);
  353. break;
  354. #ifndef CPM
  355. case XMKD_CMD:
  356. case MKD_CMD:
  357. file = pathname(ftp.cd,arg);
  358. if(!permcheck(ftp.path,ftp.perms,MKD_CMD,file)){
  359. fprintf(ftp.control,noperm);
  360. #ifdef __TURBOC__
  361. } else if(mkdir(file) == 0){
  362. #else
  363. } else if(mkdir(file,0777) == 0){
  364. #endif
  365. logmsg(fileno(ftp.control),"MKD %s",file);
  366. fprintf(ftp.control,mkdok);
  367. } else {
  368. fprintf(ftp.control,cantmake,file,sys_errlist[errno]);
  369. }
  370. FREE(file);
  371. break;
  372. case XRMD_CMD:
  373. case RMD_CMD:
  374. file = pathname(ftp.cd,arg);
  375. if(!permcheck(ftp.path,ftp.perms,RMD_CMD,file)){
  376.   fprintf(ftp.control,noperm);
  377. } else if(rmdir(file) == 0){
  378. logmsg(fileno(ftp.control),"RMD %s",file);
  379. fprintf(ftp.control,deleok);
  380. } else {
  381. fprintf(ftp.control,delefail,sys_errlist[errno]);
  382. }
  383. FREE(file);
  384. break;
  385. case STRU_CMD:
  386. if(tolower(arg[0]) != 'f')
  387. fprintf(ftp.control,unsupp);
  388. else
  389. fprintf(ftp.control,okay);
  390. break;
  391. case MODE_CMD:
  392. if(tolower(arg[0]) != 's')
  393. fprintf(ftp.control,unsupp);
  394. else
  395. fprintf(ftp.control,okay);
  396. break;
  397. case SYST_CMD:
  398. fprintf(ftp.control,syst,System,NBBY,Version);
  399. break;
  400. case XMD5_CMD:
  401. file = pathname(ftp.cd,arg);
  402. switch(ftp.type){
  403. case IMAGE_TYPE:
  404. case LOGICAL_TYPE:
  405. mode = READ_BINARY;
  406. break;
  407. case ASCII_TYPE:
  408. mode = READ_TEXT;
  409. break;
  410. }
  411. if(!permcheck(ftp.path,ftp.perms,RETR_CMD,file)){
  412.   fprintf(ftp.control,noperm);
  413. } else if((ftp.fp = fopen(file,mode)) == NULL){
  414. fprintf(ftp.control,cantopen,file,sys_errlist[errno]);
  415. } else {
  416. uint8 hash[16];
  418. logmsg(fileno(ftp.control),"XMD5 %s",file);
  419. if(ftp.type == ASCII_TYPE && isbinary(ftp.fp))
  420. fprintf(ftp.control,binwarn,file);
  422. md5hash(ftp.fp,hash,ftp.type == ASCII_TYPE);
  423. fclose(ftp.fp);
  424. ftp.fp = NULL;
  425. fprintf(ftp.control,"200 ");
  426. for(i=0;i<16;i++)
  427. fprintf(ftp.control,"%02x",hash[i]);
  428. fprintf(ftp.control," %sn",file);
  429. }
  430. FREE(file);
  431. break;
  432. }
  433. #endif
  434. goto loop;
  435. finish:
  436. logmsg(fileno(ftp.control),"close FTP");
  437. /* Clean up */
  438. fclose(ftp.control);
  439. if(ftp.data != NULL)
  440. fclose(ftp.data);
  441. if(ftp.fp != NULL)
  442. fclose(ftp.fp);
  443. free(ftp.username);
  444. free(ftp.path);
  445. free(ftp.cd);
  446. }
  448. /* Shut down FTP server */
  449. int
  450. ftp0(argc,argv,p)
  451. int argc;
  452. char *argv[];
  453. void *p;
  454. {
  455. uint16 port;
  457. if(argc < 2)
  458. port = IPPORT_FTP;
  459. else
  460. port = atoi(argv[1]);
  462. return stop_tcp(port);
  463. }
  464. static
  465. int
  466. pport(sock,arg)
  467. struct sockaddr_in *sock;
  468. char *arg;
  469. {
  470. int32 n;
  471. int i;
  473. n = 0;
  474. for(i=0;i<4;i++){
  475. n = atoi(arg) + (n << 8);
  476. if((arg = strchr(arg,',')) == NULL)
  477. return -1;
  478. arg++;
  479. }
  480. sock->sin_addr.s_addr = n;
  481. n = atoi(arg);
  482. if((arg = strchr(arg,',')) == NULL)
  483. return -1;
  484. arg++;
  485. n = atoi(arg) + (n << 8);
  486. sock->sin_port = n;
  487. return 0;
  488. }
  490. /* Attempt to log in the user whose name is in ftp->username and password
  491.  * in pass
  492.  */
  493. static void
  494. ftplogin(ftp,pass)
  495. struct ftpserv *ftp;
  496. char *pass;
  497. {
  498. char *path;
  499. int anony = 0;
  501. path = mallocw(200);
  502. if((ftp->perms = userlogin(ftp->username,pass,&path,200,&anony))
  503.    == -1){
  504. fprintf(ftp->control,noperm);
  505. free(path);
  506. return;
  507. }
  508. /* Set up current directory and path prefix */
  509. #if defined(AMIGAGONE)
  510. ftp->cd = pathname("", path);
  511. ftp->path = strdup(ftp->cd);
  512. free(path);
  513. #else
  514. ftp->cd = path;
  515. ftp->path = strdup(path);
  516. #endif
  518. fprintf(ftp->control,logged);
  519. if(!anony)
  520. logmsg(fileno(ftp->control),"%s logged in",ftp->username);
  521. else
  522. logmsg(fileno(ftp->control),"%s logged in, ID %s",ftp->username,pass);
  523. }
  525. #ifdef MSDOS
  526. /* Illegal characters in a DOS filename */
  527. static char badchars[] = ""[]:|<>+=;,";
  528. #endif
  530. /* Return 1 if the file operation is allowed, 0 otherwise */
  531. int
  532. permcheck(path,perms,op,file)
  533. char *path;
  534. int perms;
  535. int op;
  536. char *file;
  537. {
  538. #ifdef MSDOS
  539. char *cp;
  540. #endif
  542. if(file == NULL || path == NULL)
  543. return 0; /* Probably hasn't logged in yet */
  544. #ifdef MSDOS
  545. /* Check for characters illegal in MS-DOS file names */
  546. for(cp = badchars;*cp != '';cp++){
  547. if(strchr(file,*cp) != NULL)
  548. return 0;
  549. }
  550. #endif
  551. #ifndef MAC
  552. /* The target file must be under the user's allowed search path */
  553. if(strncmp(file,path,strlen(path)) != 0)
  554. return 0;
  555. #endif
  557. switch(op){
  558. case RETR_CMD:
  559. /* User must have permission to read files */
  560. if(perms & FTP_READ)
  561. return 1;
  562. return 0;
  563. case DELE_CMD:
  564. case RMD_CMD:
  565. /* User must have permission to (over)write files */
  566. if(perms & FTP_WRITE)
  567. return 1;
  568. return 0;
  569. case STOR_CMD:
  570. case MKD_CMD:
  571. /* User must have permission to (over)write files, or permission
  572.  * to create them if the file doesn't already exist
  573.  */
  574. if(perms & FTP_WRITE)
  575. return 1;
  576. if(access(file,2) == -1 && (perms & FTP_CREATE))
  577. return 1;
  578. return 0;
  579. }
  580. return 0; /* "can't happen" -- keep lint happy */
  581. }
  582. static int
  583. sendit(ftp,command,file)
  584. struct ftpserv *ftp;
  585. char *command;
  586. char *file;
  587. {
  588. long total;
  589. struct sockaddr_in dport;
  590. int s;
  592. s = socket(AF_INET,SOCK_STREAM,0);
  593. dport.sin_family = AF_INET;
  594. dport.sin_addr.s_addr = INADDR_ANY;
  595. dport.sin_port = IPPORT_FTPD;
  596. bind(s,(struct sockaddr *)&dport,SOCKSIZE);
  597. fprintf(ftp->control,sending,command,file);
  598. fflush(ftp->control);
  599. if(connect(s,(struct sockaddr *)&ftp->port,SOCKSIZE) == -1){
  600. fclose(ftp->fp);
  601. ftp->fp = NULL;
  602. close_s(s);
  603. ftp->data = NULL;
  604. fprintf(ftp->control,noconn);
  605. return -1;
  606. }
  607. ftp->data = fdopen(s,"r+");
  608. /* Do the actual transfer */
  609. total = sendfile(ftp->fp,ftp->data,ftp->type,0);
  611. if(total == -1){
  612. /* An error occurred on the data connection */
  613. fprintf(ftp->control,noconn);
  614. shutdown(fileno(ftp->data),2); /* Blow away data connection */
  615. fclose(ftp->data);
  616. } else {
  617. fprintf(ftp->control,txok);
  618. }
  619. fclose(ftp->fp);
  620. ftp->fp = NULL;
  621. fclose(ftp->data);
  622. ftp->data = NULL;
  623. if(total == -1)
  624. return -1;
  625. else
  626. return 0;
  627. }
  628. static int
  629. recvit(ftp,command,file)
  630. struct ftpserv *ftp;
  631. char *command;
  632. char *file;
  633. {
  634. struct sockaddr_in dport;
  635. long total;
  636. int s;
  638. s = socket(AF_INET,SOCK_STREAM,0);
  639. dport.sin_family = AF_INET;
  640. dport.sin_addr.s_addr = INADDR_ANY;
  641. dport.sin_port = IPPORT_FTPD;
  642. bind(s,(struct sockaddr *)&dport,SOCKSIZE);
  643. fprintf(ftp->control,sending,command,file);
  644. fflush(ftp->control);
  645. if(connect(s,(struct sockaddr *)&ftp->port,SOCKSIZE) == -1){
  646. fclose(ftp->fp);
  647. ftp->fp = NULL;
  648. close_s(s);
  649. ftp->data = NULL;
  650. fprintf(ftp->control,noconn);
  651. return -1;
  652. }
  653. ftp->data = fdopen(s,"r+");
  654. /* Do the actual transfer */
  655. total = recvfile(ftp->fp,ftp->data,ftp->type,0);
  657. #ifdef CPM
  658. if(ftp->type == ASCII_TYPE)
  659. putc(CTLZ,ftp->fp);
  660. #endif
  661. if(total == -1) {
  662. /* An error occurred while writing the file */
  663. fprintf(ftp->control,writerr,sys_errlist[errno]);
  664. shutdown(fileno(ftp->data),2); /* Blow it away */
  665. fclose(ftp->data);
  666. } else {
  667. fprintf(ftp->control,rxok);
  668. fclose(ftp->data);
  669. }
  670. ftp->data = NULL;
  671. fclose(ftp->fp);
  672. ftp->fp = NULL;
  673. if(total == -1)
  674. return -1;
  675. else
  676. return 0;
  677. }