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

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

TCP/IP协议栈

开发平台：

Visual C++

bootpcmd.c：源码内容
  1. /*************************************************/
  2. /* Center for Information Technology Integration */
  3. /*           The University of Michigan          */
  4. /*                    Ann Arbor                  */
  5. /*                                               */
  6. /* Dedicated to the public domain.               */
  7. /* Send questions to info@citi.umich.edu         */
  8. /*                                               */
  9. /* BOOTP is documented in RFC 951 and RFC 1048   */
  10. /*************************************************/
  14. #include <stdio.h>
  15. #include <systypes.h>
  16. #include <sysstat.h>
  17. #include <time.h>
  18. #include <ctype.h>
  19. #include <stdarg.h>
  20. #include "global.h"
  21. #include "config.h"
  22. #include "cmdparse.h"
  23. #include "bootpd.h"
  24. #include "netuser.h"
  25. #include "iface.h"
  26. #include "udp.h"
  27. #include "arp.h"
  29. #define BP_DEFAULT_TAB "bootptab"
  30. #define BP_DEFAULT_LOG "bootplog"
  31. #define BP_DEFAULT_DIR "bpfiles"
  32. #define BP_DEFAULT_FILE "boot"
  34. static char    *bootptab = BP_DEFAULT_TAB;
  35. static FILE    *bootfp;                 /* bootptab fp */
  36. static long     modtime;           /* last modification time of bootptab */
  38. static char    bootplog[64] = BP_DEFAULT_LOG;
  39. static int LogInFile = 0;   /* Should bp_log log in a file? */
  40. static int LogOnScreen = 0; /* Should bp_log log on screen? */
  42. static char    *line;            /* line buffer for reading bootptab */
  43. static int     linenum;          /* current ilne number in bootptab */
  45. extern int Nhosts;          /* number of hosts in host structure */
  46. extern struct host hosts[MHOSTS];
  48. extern char    homedir[64];      /* bootfile homedirectory */
  49. extern char    defaultboot[64];  /* default file to boot */
  50. extern int32   bp_DefaultDomainNS[BP_MAXDNS]; /* default domain name server */
  51. extern int Nhosts;
  52. extern struct udp_cb *Bootpd_cb;
  58. static int bp_Homedir(int argc,char *argv[],void *p);
  59. static int bp_DefaultFile(int argc,char *argv[],void *p);
  60. static int bp_DynamicRange(int argc,char *argv[],void *p);
  61. static int bp_donothing(int argc,char *argv[],void *p);
  62. static int bp_Host(int argc,char *argv[],void *p);
  63. static int bp_rmHost(int argc,char *argv[],void *p);
  64. static int bp_DomainNS(int argc,char *argv[],void *p);
  65. static int bp_Start(int argc,char *argv[],void *p);
  66. static int bp_Stop(int argc,char *argv[],void *p);
  67. static int bp_logFile(int argc,char *argv[],void *p);
  68. static int bp_logScreen(int argc,char *argv[],void *p);
  69. static void dumphosts(void);
  71. void bootpd(struct iface *iface, struct udp_cb *sock, int cnt);
  73. static struct cmds BootpdCmds[] = {
  74.         "",             bp_donothing,           0, 0, NULL,
  75. "start",  bp_Start, 0, 0, NULL,
  76. "stop", bp_Stop, 0, 0, NULL,
  77. "dns", bp_DomainNS, 0, 0, NULL,
  78.         "dynip",     bp_DynamicRange,      0, 0, NULL,
  79. "host", bp_Host, 0, 0, NULL,
  80. "rmhost", bp_rmHost, 0, 0, NULL,
  81.         "homedir",      bp_Homedir,           0, 0, NULL,
  82.         "defaultfile",  bp_DefaultFile,       0, 0, NULL,
  83. "logfile", bp_logFile, 0, 0, NULL,
  84. "logscreen", bp_logScreen, 0, 0, NULL,
  85.         NULL,       NULL,                 0, 0, NULL
  86. };
  90. int 
  91. bootpdcmd (argc, argv, p)
  92. int argc;
  93. char *argv[];
  94. void *p;
  95. {
  96. return subcmd (BootpdCmds, argc, argv, p);
  97. }
  100. /* Start up bootp service */
  101. static int
  102. bp_Start (argc,argv,p)
  103. int argc;
  104. char *argv[];
  105. void *p;
  106. {
  108.         struct socket lsock;
  109.         time_t tloc;
  110. char *usage = "bootpd startn";
  112. if (argc != 1) {
  113. printf (usage);
  114. return (-1);
  115. }
  117. time(&tloc);
  118.         bp_log ("nn####BOOTP server starting at %sn", ctime(&tloc));
  120.         lsock.address = INADDR_ANY;
  121.         lsock.port = IPPORT_BOOTPS;
  123.         /* This way is better than recvfrom because it passes the iface in bootpd call */
  124.        /* Listen doesn't work for datagrams. */
  126.         if (Bootpd_cb == NULL) {
  127.                 if ((Bootpd_cb = open_udp(&lsock, bootpd)) == NULL) {
  128. printf ("bootpd: can't open_udpn");
  129. return (-1);
  130. }
  131.         }
  133.         /*
  134.          * Read the bootptab file once immediately upon startup.
  135.          */
  137.         da_init();
  139.         readtab();
  141.         return (0);
  142. }
  146. /* Stop bootp service */
  147. static int
  148. bp_Stop (argc, argv, p)
  149. int argc;
  150. char *argv[];
  151. void *p;
  152. {
  153. time_t now;
  154. char *usage = "bootpd stopn";
  156. if (argc != 1) {
  157. printf (usage);
  158. return -1;
  159. }
  161. time (&now);
  163.         Nhosts = 0;
  164.         da_shut();
  165.         readtab_shut();
  166.         del_udp (Bootpd_cb);
  167.         Bootpd_cb = NULL;
  169. bp_log ("Bootpd shutdown %s", ctime (&now));
  170.         return (0);
  171. };
  175. static int
  176. bp_logFile (argc, argv, p)
  177. int argc;
  178. char *argv[];
  179. void *p;
  180. {
  181. int i;
  182. time_t now;
  183. char *usage = "bootpd logfile [<file_name> | default] [on | off] n"; 
  185. time (&now);
  187. if (argc == 1) {
  188. if (LogInFile)
  189.                  printf ("Bootpd logging to file '%s' turned on.n", bootplog);
  190. else 
  191.                  printf ("Bootpd logging to file '%s' turned off.n", bootplog);
  192. }
  193. else {
  194. for (i = 1; i < argc; i++) {
  196. if (strcmp ("?", argv[i]) == 0) 
  197. printf (usage);
  199. else if (strcmp ("off", argv[i]) == 0) {
  200. bp_log ("Stopping file logging at %s", ctime(&now));
  201. LogInFile = 0;
  202. }
  203. else if (strcmp ("on", argv[i]) == 0) {
  204. LogInFile = 1;
  205. bp_log ("Starting file logging at %s", ctime(&now));
  206. }
  207. else if (strcmp ("default", argv[i]) == 0) {
  208. strcpy (bootplog, BP_DEFAULT_LOG);
  209. bp_log ("File for logging set to %sn", bootplog);
  210. }
  211. else {
  212. strcpy (bootplog, argv[1]);
  213. bp_log ("File for logging set to %sn", bootplog);
  214. }
  215. }
  216. }
  217. return 0;
  218. }
  221. static int
  222. bp_logScreen (argc, argv, p)
  223. int argc;
  224. char *argv[];
  225. void *p;
  226. {
  227. char *usage = "bootpd logscreen [on | off]n";
  229.         if (argc == 1)
  230. if (LogOnScreen)
  231.                  printf ("Bootpd logging on screen turned on.n");
  232. else 
  233.                  printf ("Bootpd logging on screen turned off.n");
  235.         else if (argc == 2)  {
  236.                 if  (strcmp ("on", argv[1]) == 0)
  237.                         LogOnScreen = 1;
  238.                 else if  (strcmp ("off", argv[1]) == 0)
  239.                         LogOnScreen = 0;
  240. else printf (usage);
  241. }
  242. else printf (usage);
  243. return 0;
  244. }
  249. static int
  250. bp_DomainNS (argc, argv, p)
  251. int argc;
  252. char *argv[];
  253. void *p;
  254. {
  255. int a0, a1, a2, a3;
  256. int i;
  257. char *usage = "bootpd dns [<IP addr of domain name server>...]n";
  259. if (argc == 1) {
  260. printf ("Bootp domain name servers: ");
  261. for (i=0; (i < BP_MAXDNS) && (bp_DefaultDomainNS[i] != 0); i++) 
  262. printf (" %s", inet_ntoa (bp_DefaultDomainNS[i]));
  263. printf ("n");
  264. return (0);
  265. }
  267. if (argc > 1) {
  268. if ((argc == 2) && (strcmp ("?", argv[1]) == 0)) {
  269. printf (usage);
  270. return 0;
  271. }
  273. /* A list of name servers has been given */
  274. /* reset the domain name server list */
  275. for (i= 0; i < BP_MAXDNS; i++) 
  276. bp_DefaultDomainNS[i] = 0;
  278. /* get ip address */
  279. for (i = 1; (i < argc) && (i < BP_MAXDNS); i++) {
  280.                  if (4 != sscanf (argv[i], "%d.%d.%d.%d", &a0, &a1, &a2, &a3)) {
  281.                           printf("bad internet address: %sn", argv[1], linenum);
  282. return  -1;
  283.                  }
  284.          bp_DefaultDomainNS[i-1] = aton(argv[i]);
  285. }
  286. }
  287. /* record for the loggers sake */
  288. bp_log ("Bootp domain name servers: ");
  289. for (i=0; (i < BP_MAXDNS) && (bp_DefaultDomainNS[i] != 0); i++) 
  290. bp_log (" %s", inet_ntoa (bp_DefaultDomainNS[i]));
  291. bp_log ("n");
  292. return 0;
  293. }
  297. static int
  298. bp_rmHost (argc, argv, p)
  299. int argc;
  300. char *argv[];
  301. void *p;
  302. {
  303. int i;
  304. struct host *hp = NULL;
  305. struct host *cp = NULL;
  306. char *usage = "bootpd rmhost <host name>n";
  309. if (argc == 2) {
  311. /* Find the host record */
  312. for (i=0; i < Nhosts; i++) {
  313. if (strcmp (hosts[i].name, argv[1]) == 0) {
  314. hp = &(hosts[i]);
  315. break;
  316. }
  317. }
  318. /* Return if not found */
  319. if (hp == NULL) {
  320. printf ("Host %s not in host tables.n", argv[1]);
  321. return -1;
  322. }
  323. bp_log ("Host %s removed from host tablen", hp->name);
  324. cp = &(hosts [Nhosts - 1]);
  325. if (hp < cp) 
  326. memcpy(hp,cp,sizeof(struct host));
  327. Nhosts--;
  328. return 0;
  329. }
  330. else printf (usage);
  331. return 0;
  332. }
  335. /*
  336.  * Printout the hosts table.
  337. */
  338. static void
  339. dumphosts()
  340. {
  341.         int i;
  342.         struct host *hp;
  343. struct arp_type *at;
  345.         printf ("nnStatus of host tablen");
  347.         if (Nhosts == 0) {
  348.                 printf ("     No hosts in host tablen");
  349.                 return;
  350.         }
  351.         for (i = 0; i <= Nhosts-1; i++) {
  352.                 hp = &hosts[i];
  353. at = &Arp_type[hp->htype];
  355.                 printf ("%s  %s  %s  %s  '%s'n",
  356.                         hp->name, ArpNames[hp->htype], (*at->format)(bp_ascii, hp->haddr),
  357.                         inet_ntoa ((int32)hp->iaddr.s_addr),
  358.                         hp->bootfile);
  360.         }
  361. }
  364. static int
  365. bp_Host (argc, argv, p)
  366. int argc;
  367. char *argv[];
  368. void *p;
  369. {
  370. struct host *hp;
  371. int a0, a1, a2, a3;
  372. struct arp_type *at;
  373. char *usage = "bootpd host [<hostname> <hardware type> <hardware addr> <ip addr> [boot file]]n";
  375. switch (argc) {
  376. case 1:
  377. dumphosts();
  378. break;
  379. case 5:
  380. case 6:
  382. hp = &hosts[Nhosts];
  384. /* get host name */
  385. strncpy (hp->name, argv[1], sizeof (hp->name));
  387. /* get hardware type */
  388. /* This code borrowed from Phil Karn's arpcmd.c */
  389. /* This is a kludge. It really ought to be table driven */
  390. switch(tolower(argv[2][0])){
  391. case 'n':       /* Net/Rom pseudo-type */
  392. hp->htype = ARP_NETROM;
  393. break;
  394. case 'e': /* "ether" */
  395. hp->htype = ARP_ETHER;
  396. break;
  397. case 'a': /* "ax25" */
  398. hp->htype = ARP_AX25;
  399. break;
  400. case 'm': /* "mac appletalk" */
  401. hp->htype = ARP_APPLETALK;
  402. break;
  403. default:
  404. printf("unknown hardware type "%s"n",argv[2]);
  405. return -1;
  406. }
  408. at = &Arp_type[hp->htype];
  409. if(at->scan == NULL){
  410. return 1;
  411. }
  412. /* Destination address */
  413. (*at->scan)(hp->haddr,argv[3]);
  416. /* get ip address */
  417.                 if (4 != sscanf (argv[4], "%d.%d.%d.%d", &a0, &a1, &a2, &a3))
  418.                 {
  419.                         printf("bad internet address: %sn", argv[1], linenum);
  420.                         return (0);
  421.                 }
  422.         hp->iaddr.s_addr = aton(argv[4]);
  424. /* get the bootpfile */
  425. if (argc == 6) strncpy (hp->bootfile, argv[5], sizeof (hp->bootfile));
  426. else hp->bootfile[0] = 0;
  428.                 bp_log ("Host added: %s  %s  %s  %s  '%s'n",
  429.                         hp->name, ArpNames[hp->htype], (*at->format)(bp_ascii, hp->haddr),
  430.                         inet_ntoa ((int32)hp->iaddr.s_addr),
  431.                         hp->bootfile);
  434. Nhosts++;
  435. break;
  437. default:
  438. printf (usage);
  439. break;
  440. }
  441. return 0;
  442. }
  446. static int
  447. bp_Homedir (argc, argv, p)
  448. int argc;
  449. char *argv[];
  450. void *p;
  451. {
  452. char *usage = "bootpd homedir [<name of home directory> | default]n";
  454. if (argc == 1) 
  455. printf ("Bootp home directory: '%s'n", homedir);
  456. else if (argc == 2) {
  457. if (strcmp (argv[1], "?") == 0)
  458. printf (usage);
  459. else if (strcmp (argv[1], "default") == 0) {
  460. strcpy (homedir, BP_DEFAULT_DIR);
  461. bp_log ("Bootp home directory set to: '%s'n", homedir);
  462. }
  463. else {
  464. strcpy (homedir, argv[1]);
  465. bp_log ("Bootp home directory set to: '%s'n", homedir);
  466. }
  467. }
  468. else printf (usage);
  469. return (0);
  470. };
  474. static int
  475. bp_DefaultFile (argc, argv, p)
  476. int argc;
  477. char *argv[];
  478. void *p;
  479. {
  480. char *usage = "bootpd defaultfile [<name of default boot file> | default]n";
  482.         if (argc == 1)
  483.                 printf ("Bootp default boot file:  '%s'n", defaultboot);
  484.         else if (argc == 2) {
  485. if (strcmp (argv[1], "?") == 0)
  486.                         printf (usage);
  487.                 else if (strcmp (argv[1], "default") == 0)
  488.                         strcpy (defaultboot, BP_DEFAULT_FILE);
  489.                 else {
  490.                         strcpy (defaultboot, argv[1]);
  491.                  bp_log ("Bootp default boot file set to:  '%s'n", defaultboot);
  492. }
  493.         }
  494. else
  495.                 printf (usage);
  497. return  (0);
  498. };
  501. static int
  502. bp_DynamicRange (argc, argv, p)
  503. int argc;
  504. char *argv[];
  505. void *p;
  506. {
  507. int i0, i1, i2, i3;
  508. int32 start, end;
  509. struct iface *iface;
  510. char *usage = "bootpd dynip [<net name> | <netname>  <IP address> <IP address> | <netname> off]n";
  512. if (argc == 1) {
  513. da_status (NULL);
  514. return 0;
  515. }
  516. if ((argc == 2) && (strcmp ("?", argv[1]) == 0)) {
  517. printf (usage);
  518. return 0;
  519. }
  521. /* get the interface */
  522. iface = if_lookup (argv[1]);
  523. if (iface == NULL) {
  524. printf ("network '%s' not foundn", argv[1]);
  525. return  (-1);
  526. }
  527. if (argc == 2) {
  528. da_status (iface);
  529. return 0;
  530. }
  531. if (argc == 3) {
  532. if (strcmp ("off", argv[2]) == 0) 
  533. da_done_net (iface);
  534. else printf (usage);
  535. }
  536. else if (argc == 4) {
  538.         /* get ip address */
  539.                 /* check the ip address - isaddr isn't a ka9q function */
  540.                 if ((4 != sscanf (argv[2], "%d.%d.%d.%d", &i0, &i1, &i2, &i3)) || 
  541. (i0 > 255) || (i1 > 255) || (i2 > 255) || (i3 > 255)
  542. )
  543.                 {
  544.                  printf("bad internet address: %sn", argv[2], linenum);
  545.                         return (-1);
  546.                 }
  548.                 if ((4 != sscanf (argv[3], "%d.%d.%d.%d", &i0, &i1, &i2, &i3)) || 
  549. (i0 > 255) || (i1 > 255) || (i2 > 255) || (i3 > 255)
  550. )
  551.                 {
  552.                  printf("bad internet address: %sn", argv[3], linenum);
  553.                         return (-1);
  554.                 }
  556.                 start = aton(argv[2]);
  557.              end = aton(argv[3]);
  559. da_serve_net (iface, start, end);
  561. }
  562. else {
  563. printf (usage);
  564. return (0);
  565. }
  567. return (0);
  568. };
  571. static int
  572. bp_donothing (argc, argv, p)
  573. int argc;
  574. char *argv[];
  575. void *p;
  576. {
  577.         return (0);
  578. }
  581. /*
  582.  * Read bootptab database file.  Avoid rereading the file if the
  583.  * write date hasn't changed since the last time we read it.
  584.  */
  585. int
  586. readtab()
  587. {
  588.         struct stat st;
  590.         /* If the file hasn't been opened, open it. */
  591.         if (bootfp == 0) {
  592.                 if ((bootfp = fopen(bootptab, "r")) == NULL) {
  593.                         bp_log("Can't open bootptab file: %sn", bootptab);
  594.                         return (-1);
  595.                 }
  596.         }
  598.         /* Only reread if modified */
  599.         stat (bootptab, &st);
  600.         if (st.st_mtime == modtime && st.st_nlink) {
  601.                 return (0); /* hasnt been modified or deleted yet */
  602.         }
  603.         /* It's been changed, reread. */
  605.         if ((bootfp = fopen(bootptab, "r")) == NULL) {
  606.                 bp_log("Can't open %sn", bootptab);
  607.                 return (-1);
  608.         }
  609.         fstat(fileno(bootfp), &st);
  610.         bp_log("(re)reading %sn", bootptab);
  611.         modtime = st.st_mtime;
  613.         /*
  614.          * read and parse each line in the file.
  615.          */
  617. line = mallocw(BUFSIZ);
  619. while (fgets(line, BUFSIZ, bootfp) != NULL) {
  620. linenum++;
  623. if ((line[0] == 0) || (line[0] == '#') || (line[0] == ' '))
  624. continue;
  626.                 if (cmdparse (BootpdCmds, line, NULL) ==  -1)
  627.                          continue;
  629.         }
  630.         fclose(bootfp);
  631. free (line);
  632. return (0);
  633. }
  635. void
  636. readtab_shut()
  637. {
  638. modtime = 0;
  639. }
  641. /*
  642.  * log an error message
  643.  *
  644.  */
  645. void
  646. bp_log(char *fmt,...)
  647. {
  648.         FILE *fp;
  649. va_list ap;
  651.         if (LogOnScreen) {
  652. va_start(ap,fmt);
  653. vprintf(fmt, ap);
  654. va_end(ap);
  655. fflush (stdout);
  656. }
  657.         if (LogInFile) {
  658.                 if ((fp = fopen(bootplog, "a+")) == NULL) {
  659.                         printf ("Cannot open bootplog.n");
  660.                         return;
  661.                 }
  662. va_start(ap,fmt);
  663. vfprintf(fp, fmt, ap);
  664. va_end(ap);
  665.                 fflush(fp);
  666.                 fclose(fp);
  667.         }
  668. }