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

Ftp客户端

开发平台:

Unix_Linux

  1. /**************************************************************************** 
  2.    Copyright (c) 1999 WU-FTPD Development Group. 
  3.    All rights reserved.
  4.    
  5.    Portions Copyright (c) 1980, 1985, 1988, 1989, 1990, 1991, 1993, 1994 
  6.    The Regents of the University of California.  Portions Copyright (c) 
  7.    1993, 1994 Washington University in Saint Louis.  Portions Copyright 
  8.    (c) 1996, 1998 Berkeley Software Design, Inc.  Portions Copyright (c) 
  9.    1998 Sendmail, Inc.  Portions Copyright (c) 1983, 1995, 1996, 1997 Eric 
  10.    P. Allman.  Portions Copyright (c) 1989 Massachusetts Institute of 
  11.    Technology.  Portions Copyright (c) 1997 by Stan Barber.  Portions 
  12.    Copyright (C) 1991, 1992, 1993, 1994, 1995, 1996, 1997 Free Software 
  13.    Foundation, Inc.  Portions Copyright (c) 1997 by Kent Landfield. 
  14.  
  15.    Use and distribution of this software and its source code are governed 
  16.    by the terms and conditions of the WU-FTPD Software License ("LICENSE"). 
  17.  
  18.    $Id: privatepw.c,v 1.9 1999/10/03 22:24:25 wuftpd Exp $
  19.  
  20. ****************************************************************************/
  21. /*
  22.    Subsystem:  WU-FTPD FTP Server
  23.    Purpose:    Change WU-FTPD Guest Passwords
  24.    File Name:  privatepw.c               
  25.    usage: privatepw [-c] [-f passwordfile] [-g group] accessgroup
  26.    privatepw [-d] [-f passwordfile] accessgroup
  27.    privatepw [-l] [-f passwordfile] 
  28.    -c:           creates a new file.
  29.    -d:           deletes specified accessgroup.
  30.    -l:           list contents of ftpgroups file.
  31.    -f ftpgroups: updates the specified file.
  32.    -g group:     set real group to the specified group.
  33.    This software was initially written by Kent Landfield (kent@landfield.com)
  34.  */
  35. #include <sys/types.h>
  36. #include <sys/signal.h>
  37. #include <sys/stat.h>
  38. #include <string.h>
  39. #include <stdio.h>
  40. #include <stdlib.h>
  41. #include <time.h>
  42. #include <grp.h>
  43. #include <unistd.h>
  44. #include "../../src/config.h"
  45. #include "../../src/pathnames.h"
  46. #define BUFLEN 256
  47. #define GROUPLEN 8
  48. char *tmp;
  49. char line[BUFLEN];
  50. FILE *fp;
  51. int verbose = 0;
  52. static unsigned char itoa64[] = /* 0 ... 63 => ascii - 64 */
  53. "./0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
  54. void print_copyright(void);
  55. static void usage(void)
  56. {
  57.     fprintf(stderr, "usage: privatepw [-c] [-f ftpgroups] [-g group] accessgroupn");
  58.     fprintf(stderr, "       privatepw [-d] [-f ftpgroups] accessgroupn");
  59.     fprintf(stderr, "       privatepw [-l] [-f ftpgroups]n");
  60.     fprintf(stderr, "tt-c:           creates a new file.n");
  61.     fprintf(stderr, "tt-d:           deletes specified accessgroup.n");
  62.     fprintf(stderr, "tt-l:           list contents of ftpgroups file.n");
  63.     fprintf(stderr, "tt-f ftpgroups: updates the specified file.n");
  64.     fprintf(stderr, "tt-g group:     set real group to the specified group.n");
  65.     exit(1);
  66. }
  67. static void to64(register char *s, register long v, register int n)
  68. {
  69.     while (--n >= 0) {
  70. *s++ = itoa64[v & 0x3f];
  71. v >>= 6;
  72.     }
  73. }
  74. static void terminate(void)
  75. {
  76.     if (tmp)
  77. unlink(tmp);
  78.     exit(1);
  79. }
  80. static void catchintr(void)
  81. {
  82.     fprintf(stderr, "Interrupted.n");
  83.     terminate();
  84. }
  85. static char *savit(char *s)
  86. {
  87.     char *d;
  88.     if ((d = (char *) malloc(strlen(s) + 1)) == NULL) {
  89. fprintf(stderr, "Whoa... Malloc failed.n");
  90. terminate();
  91.     }
  92.     strcpy(d, s);
  93.     return (d);
  94. }
  95. static int confirmed(char *accessgroup)
  96. {
  97.     register int ch;
  98.     printf("Delete %s: Are your sure ? (y/n) ", accessgroup);
  99.     ch = getc(stdin);
  100.     if (ch == 'y')
  101. return (1);
  102.     return (0);
  103. }
  104. static char *getgroup(char *msg)
  105. {
  106.     register int ch;
  107.     register char *p;
  108.     static char buf[GROUPLEN + 1];
  109.     fputs(msg, stderr);
  110.     rewind(stderr); /* implied flush */
  111.     for (p = buf; (ch = getc(stdin)) != EOF && ch != 'n';)
  112. if (p < buf + GROUPLEN)
  113.     *p++ = ch;
  114.     *p = '';
  115.     if (getgrnam(buf) == NULL) {
  116. fprintf(stderr, "Invalid group '%s' specifiedn", buf);
  117. terminate();
  118.     }
  119.     return (buf);
  120. }
  121. static void addrecord(char *accessgroup, char *sysgroup, char *msg, FILE *f)
  122. {
  123.     char *pw, *cpw, salt[3];
  124. #ifndef NO_CRYPT_PROTO
  125.     extern char *crypt(const char *, const char *);
  126. #endif
  127.     char *getpass(const char *prompt);
  128.     printf("%s %sn", msg, accessgroup);
  129.     if (sysgroup[0] == '')
  130. strcpy(sysgroup, getgroup("Real System Group to use: "));
  131.     pw = savit((char *) getpass("New password: "));
  132.     if (strcmp(pw, (char *) getpass("Re-type new password: "))) {
  133. fprintf(stderr, "They don't match, sorry.n");
  134. if (tmp)
  135.     unlink(tmp);
  136. exit(1);
  137.     }
  138.     srand((int) time((time_t *) NULL));
  139.     to64(&salt[0], rand(), 2);
  140.     cpw = crypt(pw, salt);
  141.     free(pw);
  142.     fprintf(f, "%s:%s:%sn", accessgroup, cpw, sysgroup);
  143. }
  144. static void list_privatefile(char *privatefile)
  145. {
  146.     if (verbose)
  147. fprintf(stderr, "Private File: %s file.n", privatefile);
  148.     if ((fp = fopen(privatefile, "r")) == NULL) {
  149. fprintf(stderr, "Could not open %s file.n", privatefile);
  150. exit(1);
  151.     }
  152.     printf("nWU-FTPD Private file: %sn", privatefile);
  153.     printf("accessgroup : password : system groupn");
  154.     printf("-------n");
  155.     while (fgets(line, BUFLEN, fp) != NULL)
  156. fputs(line, stdout);
  157.     printf("-------n");
  158. }
  159. int main(int argc, char **argv)
  160. {
  161.     extern void (*signal(int sig, void (*disp) (int))) (int);
  162.     extern int getopt(int argc, char *const *argv, const char *optstring);
  163.     extern char *optarg;
  164.     extern int optind;
  165.     extern int opterr;
  166.     struct stat stbuf;
  167.     char realgroup[BUFLEN];
  168.     char *passwdpath;
  169.     char *cp;
  170.     char accessgroup[BUFLEN];
  171.     char w[BUFLEN];
  172.     char command[BUFLEN];
  173.     int create;
  174.     int delete;
  175.     int list;
  176.     int found;
  177.     int lineno;
  178.     int c;
  179.     FILE *tfp;
  180.     opterr = 0;
  181.     create = 0;
  182.     delete = 0;
  183.     list = 0;
  184.     tmp = NULL;
  185.     realgroup[0] = '';
  186.     passwdpath = _PATH_PRIVATE;
  187.     if (argc == 1)
  188. usage();
  189.     while ((c = getopt(argc, argv, "Vvcdf:g:l")) != EOF) {
  190. switch (c) {
  191. case 'd':
  192.     delete++;
  193.     break;
  194. case 'c':
  195.     create++;
  196.     break;
  197. case 'f':
  198.     passwdpath = optarg;
  199.     break;
  200. case 'g':
  201.     strcpy(realgroup, optarg);
  202.     if (getgrnam(realgroup) == NULL) {
  203. fprintf(stderr, "Invalid group '%s' specifiedn", realgroup);
  204. return (1);
  205.     }
  206.     break;
  207. case 'l':
  208.     list++;
  209.     break;
  210. case 'v':
  211.     verbose++;
  212.     break;
  213. case 'V':
  214.     print_copyright();
  215.     return (0);
  216.     /* NOTREACHED */
  217. default:
  218.     usage();
  219. }
  220.     }
  221.     if (list) {
  222. list_privatefile(passwdpath);
  223. return (0);
  224.     }
  225.     if (optind >= argc) {
  226. fprintf(stderr, "Need to specify an accessgroup name.n");
  227. usage();
  228.     }
  229.     signal(SIGINT, (void (*)()) catchintr);
  230.     strcpy(accessgroup, argv[optind]);
  231.     if (create) {
  232. if (stat(passwdpath, &stbuf) == 0) {
  233.     fprintf(stderr, "%s exists, cannot create it.n", passwdpath);
  234.     fprintf(stderr, "Remove -c option or use the -f option to specify another.n");
  235.     return (1);
  236. }
  237. if ((tfp = fopen(passwdpath, "w")) == NULL) {
  238.     fprintf(stderr, "Could not open "%s" for writing.n", passwdpath);
  239.     perror("fopen");
  240.     return (1);
  241. }
  242. tmp = passwdpath;
  243. printf("Creating WU-FTPD Private file: %sn", passwdpath);
  244. addrecord(accessgroup, realgroup, "Adding accessgroup", tfp);
  245. fclose(tfp);
  246. return (0);
  247.     }
  248.     tmp = tmpnam(NULL);
  249.     if ((tfp = fopen(tmp, "w")) == NULL) {
  250. fprintf(stderr, "Could not open temp file.n");
  251. return (1);
  252.     }
  253.     if ((fp = fopen(passwdpath, "r")) == NULL) {
  254. fprintf(stderr, "Could not open %s file.n", passwdpath);
  255. fprintf(stderr, "Use -c option to create new one.n");
  256. return (1);
  257.     }
  258.     lineno = 0;
  259.     found = 0;
  260.     while (fgets(line, BUFLEN, fp) != NULL) {
  261. lineno++;
  262. if (found || (line[0] == '#') || (!line[0])) {
  263.     fputs(line, tfp);
  264.     continue;
  265. }
  266. strcpy(w, line);
  267. if ((cp = strchr(w, ':')) == NULL) {
  268.     fprintf(stderr, "%s: line %d: invalid record format.n", passwdpath, lineno);
  269.     continue;
  270. }
  271. *cp++ = '';
  272. if ((cp = strchr(cp, ':')) == NULL) {
  273.     fprintf(stderr, "%s: line %d: invalid record format.n", passwdpath, lineno);
  274.     continue;
  275. }
  276. *cp++ = '';
  277. if (strcmp(accessgroup, w)) {
  278.     fputs(line, tfp);
  279.     continue;
  280. }
  281. else {
  282.     if (delete) {
  283. if (!confirmed(accessgroup))
  284.     terminate();
  285.     }
  286.     else {
  287. if (realgroup[0] == '') {
  288.     strcpy(realgroup, cp);
  289.     if ((cp = strchr(realgroup, 'n')) != NULL)
  290. *cp = '';
  291. }
  292. addrecord(accessgroup, realgroup, "Updating accessgroup", tfp);
  293.     }
  294.     found = 1;
  295. }
  296.     }
  297.     if (!found && !delete)
  298. addrecord(accessgroup, realgroup, "Adding accessgroup", tfp);
  299.     else if (!found && delete) {
  300. fprintf(stderr, "%s not found in %s.n", accessgroup, passwdpath);
  301. terminate();
  302.     }
  303.     fclose(fp);
  304.     fclose(tfp);
  305.     sprintf(command, "cp %s %s", tmp, passwdpath);
  306.     system(command);
  307.     unlink(tmp);
  308.     return (0);
  309. }