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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 操作系统开发 > 查看源码
PCBOOTP.C
资源名称：ertos.rar　[点击查看]
上传用户：sunrenlu
上传日期：2022-06-13
资源大小：1419k
文件大小：16k
源码类别：

操作系统开发

开发平台：

DOS

PCBOOTP.C：源码内容
  1. /*
  2.  *
  3.  *   BOOTP - Boot Protocol (RFC 854)
  4.  *
  5.  *   These extensions get called if _bootphost is set to an IP address or
  6.  *   to 0xffffffff.
  7.  *
  8.  *   Version
  9.  *
  10.  *   0.5 : Oct 29, 2001 : J. Toman - fixed DHCP packet size and moved DHCP
  11.  *                                   lifetime variable so it doesn't clobber
  12.  *                                   results from the server
  13.  *   0.4 : May 11, 1999 : S. Lawson - added DHCP
  14.  *   0.3 : Feb  1, 1992 : J. Dent - patched up various things
  15.  *   0.2 : May 22, 1991 : E.J. Sutcliffe - added RFC_1048 vendor fields
  16.  *   0.1 : May  9, 1991 : E. Engelke - made part of the library
  17.  *   0.0 : May  3, 1991 : E. Engelke - original program as an application
  18.  *
  19.  */
  21. #include <copyright.h>
  22. #include <stdio.h>
  23. #include <wattcp.h>
  24. #include <mem.h>
  25. #include <bootp.h>
  26. #include <conio.h>
  27. #include <string.h> // S. Lawson
  29. /* global variables */
  30. longword _bootphost = 0xffffffffL;
  31. longword _dhcphost = 0xffffffffL;       // S. Lawson
  32. word _bootptimeout = 30;
  33. word _bootpon = 0;
  35. static longword dhcp_life;         // S. Lawson
  36. static char dhcp_seen, dhcp_bind; // S. Lawson
  37. static char got_ns, got_gw, got_cy; // S. Lawson
  38. static udp_Socket bsock; // S. Lawson
  40. // S. Lawson - we now use these
  41. char *getdomainname( char *name, int length );
  42. char *setdomainname( char *string );
  43. char *sethostname( char *name );
  45. #define VM_RFC1048 0x63825363L /* I think this is correct */
  47. /*
  48.  * _dobootpc - Checks global variables _bootptimeout, _bootphost
  49.  *             if no host specified, the broadcast address
  50.  *             returns 0 on success and sets ip address
  51.  */
  52. int _dobootp( void )
  53. {
  54. // S. Lawson - moved    udp_Socket bsock;
  55.     extern char defaultdomain[]; // S. Lawson - in pcconfig.c
  56.     longword sendtimeout, bootptimeout;
  57.     word magictimeout;
  58.     word len, templen;
  59.     struct bootp sendbootp;     /* outgoing data */
  60.     struct bootp _bootp;        /* incoming data */
  61.     int status;
  62.     longword xid;
  63.     unsigned char *p ,*t;       // R. Whitby - uncomment *t
  64.     longword *l;        // S. Lawson
  65. /*
  66.     const char dhcp_opt[]={DHCP_VN_TYPE,1,DHCP_TY_REQ,DHCP_VN_OPTS,6,1,
  67.                            3,6,42,12,15,0}; */  /* S. Lawson - keep even! */
  68. /*    const char dhcp_opt[]={DHCP_VN_OPTS,6,1,3,6,42,12,15,0}; */ /* Shaun Jackman 2000/3/10 */
  69.     const char dhcp_opt[]={DHCP_VN_OPTS,6,1,3,6,42,12,15}; /* Michael Temari 2002/1/17, keep even */
  71. //    if ( _pktdevclass == PD_SLIP ) return( -1 );
  74.     /* We must get Waterloo TCP to use IP address 0 for sending */
  75.     xid = set_timeout(1);   /* random... well, actually time based */
  76.     my_ip_addr = 0;
  77.     dhcp_seen=dhcp_bind=0;       // S. Lawson
  78.     got_ns=got_gw=got_cy=0; // S. Lawson
  79.     dhcp_life = 0L;                             // JCT
  81.     if (!udp_open( &bsock, IPPORT_BOOTPC, _bootphost, IPPORT_BOOTPS, NULL )) {
  82. outs("nrUnable to resolve bootp servernr");
  83. return( -1 );
  84.     }
  86.     bootptimeout = set_timeout( _bootptimeout );
  87.     // magictimeout = (xid & 7) + 7;  /* between 7 and 14 seconds */
  88.     magictimeout = ((word)(xid & 7)) + 7;/* JCT forced to word, 7-14 s */
  90.     memset( &sendbootp, 0, sizeof( struct bootp ));
  91.     sendbootp.bp_op = BOOTREQUEST;
  92.     sendbootp.bp_htype = _pktdevclass;
  93.     /* Copy into position the Magic Number used by Bootp */
  94.     /* avoid static storage and pushf/call assembler instructions */
  95.     *(longword *)(&sendbootp.bp_vend) = intel(VM_RFC1048);
  97.     // S. Lawson - append DHCPDISCOVER to vendor field.  I'm expecting that
  98.     // a BOOTP server configured with this machines MAC address will simply
  99.     // ignore it and answer anyway, possibly a bad assumption?
  100.     p=&sendbootp.bp_vend[4];
  101.     *p++=DHCP_VN_TYPE;
  102.     *p++=1;
  103.     *p++=DHCP_TY_DSC;
  104.     *p=255;
  106.     if (_pktdevclass == PD_ETHER) sendbootp.bp_hlen = 6;
  108.     sendbootp.bp_xid = xid;
  109.     sendbootp.bp_secs = intel16( 1 );
  111. //    movmem( &_eth_addr, &sendbootp.bp_chaddr, sizeof(eth_address));
  112. // 94.11.19 ??
  113.     movmem( &_eth_addr, sendbootp.bp_chaddr, sizeof(eth_address));
  115.     while ( 1 ) {
  116. sock_fastwrite( (sock_type*)&bsock, (byte *)&sendbootp, sizeof( struct bootp ));
  117. sendbootp.bp_secs = intel16( intel16( sendbootp.bp_secs ) + magictimeout );      /* for next time */
  118.         sendtimeout = set_timeout( magictimeout += ((word)(xid >> 5)) & 7 );  // JCT
  120. while ( !chk_timeout( sendtimeout )) {
  122.     if (chk_timeout( bootptimeout))
  123. goto give_up;
  124.     kbhit();
  125.     sock_tick( (sock_type*)&bsock, &status );
  126.     status = status;    /* get rid of warning */
  127.     if ((len = sock_dataready( (sock_type*)&bsock)) != 0 ) {
  129. /* got a response, lets consider it */
  130. templen = sock_fastread( (sock_type*)&bsock, (byte *)&_bootp, sizeof( struct bootp ));
  131. // S. Lawson - vendor area increased (bootp: 64, dhcp: 312)
  132. // if ( templen < sizeof( struct bootp )) {
  134. /*
  135.  *              if ( templen < sizeof( struct bootp ) - (312-64) ) {
  136.  * JCT that's not right size either
  137.  */
  138.                 if ( templen < sizeof( struct bootp ) - 312 ) {
  139.                    /* too small, not a bootp packet */
  142.                     memset( &_bootp, 0, sizeof( struct bootp ));
  143.     continue;
  144. }
  146. /* we must see if this is for us */
  147. if (_bootp.bp_xid != sendbootp.bp_xid) {
  148.     memset( &_bootp, 0, sizeof( struct bootp ));
  149.     continue;
  150. }
  152. /* we must have found it */
  153. my_ip_addr = intel( _bootp.bp_yiaddr );
  155. // S. Lawson - just in case
  156. if (!my_ip_addr) {
  157.   outs("Rejecting zero addressrn");
  158.   memset( &_bootp, 0, sizeof( struct bootp ));
  159.   continue;
  160. }
  162. dhcp_seen = 0;          // S. Lawson
  163.                 /* dhcp_life=0L;  wrong place */  // S. Lawson
  165. if ( intel( *(longword*)(&_bootp.bp_vend)) == VM_RFC1048 ) {
  166.     /*RFC1048 complient BOOTP vendor field */
  167.     /* Based heavily on NCSA Telnet BOOTP */
  169.     p = &_bootp.bp_vend[4]; /* Point just after vendor field */
  171. // S. Lawson - vendor area increased for dhcp (bootp: 64, dhcp: 312)
  172. //     while ((*p!=255) && (p <= &_bootp.bp_vend[63])) {
  173.     while ((*p!=255) && (p <= &_bootp.bp_vend[311])) {
  174. switch(*p) {
  175.   case 0: /* Nop Pad character */
  176.  p++;
  177.  break;
  178.   case 1: /* Subnet Mask */
  179.  sin_mask = intel( *(longword *)( &p[2] ));
  180.  /* and fall through */
  181.   case 2: /* Time offset */
  182.  p += *(p+1) + 2;
  183.  break;
  184.   case 3: /* gateways */
  185.   /* only add first */
  186.   if (!got_gw)   // S. Lawson
  187.      _arp_add_gateway( NULL,
  188. intel( *(longword*)(&p[2])));
  189.   p +=*(p+1)+2;
  190.   got_gw=1; // S. Lawson
  191.   break;
  192.   /* and fall through */
  193.   case 4: /*time servers */
  194.   /* fall through */
  195.   case 5: /* IEN=116 name server */
  196.  p +=*(p+1)+2;
  197.  break;
  198.   case 6: /* Domain Name Servers (BIND) */
  199. if (!got_ns)  // S. Lawson
  201.                                 //for ( len = 0; len < *(p+1) ; len += 4 )
  202.                                 // JCT make promotion to int explicit
  203.                                 for ( len = 0 ; len < (int)*(p+1) ; len += 4 )
  204.     _add_server( &_last_nameserver,
  205. MAX_NAMESERVERS, def_nameservers,
  206.     intel( *(longword*)(&p[2+len])));
  207. got_ns=1; // S. Lawson
  208. /* and fall through */
  209.   case 7: /* log server */
  210.  p += *(p+1)+2;
  211.  break;
  212.   case 8: /* cookie server */
  213.  if (!got_cy)  // S. Lawson
  214.                                    // for ( len = 0; len < *(p+1) ; len += 4 )
  215.                                    // JCT make promotion to int explicit
  216.                                    for ( len = 0 ;  len < (int)*(p+1); len += 4 )
  217.      _add_server( &_last_cookie, MAX_COOKIES,
  218. _cookie, intel( *(longword*)(&p[2+len])));
  219.   /* and fall through */
  220.   p +=*(p+1)+2;
  221.   got_cy=1; // S. Lawson
  222.   break;
  223.   case 9: /* lpr server */
  224.   case 10: /* impress server */
  225.   case 11: /* rlp server */
  226.    p +=*(p+1)+2;
  227.    break;
  229. #ifdef NOTUSED // R. Whitby - handle bootp/dhcp setting of domain name too
  230.   case 12: /* Client Hostname */
  231.   movmem( &p[2] , _hostname, MAX_STRING );
  232. /* bootpfix - 94.09.30 mdurkin@tsoft.net */
  233.        /* hostname field is *not* null terminated; need to use
  234.     length byte.  Extra bytes used to be tacked on the
  235.     end, usually just a single 0xFF char (the end tag).
  236.     Technically should also probably make the
  237.     _hostname buffer 256 chars long so we'll never
  238.     truncate the name arbitrarily... see CMU bootpd. */
  239.   if( *(p+1) < MAX_STRING ) {
  240.       _hostname[ *(p+1) ] = '';
  241.   } else _hostname[ MAX_STRING - 1 ] = '';
  242.   p += *(p+1)+2;
  243.   break;
  244. #else
  245.   case 12: /* Client Hostname */
  246.   len = *(++p);
  247.   movmem(p+1, p, len); *(p+len) = 0;
  248.   if ((t = strchr((char *)p, '.')) != 0) {
  249.      *t++ = 0;
  250.      if (!getdomainname(NULL, 0)) {
  251. strncpy(defaultdomain, (char *)t, MAX_STRING);
  252. defaultdomain[MAX_STRING-1] = 0;
  253. setdomainname(defaultdomain);
  254.      }
  255.   }
  256.   strncpy(_hostname, (char *)p, MAX_STRING);
  257.   _hostname[MAX_STRING-1] = 0;
  258.   sethostname(_hostname);
  259.   p += len+1;
  260.   break;
  262.   case 15: /* Client Domain */
  263.   len = *(++p);
  264.   if (!getdomainname(NULL, 0)) {
  265.      movmem(p+1, p, len); *(p+len) = 0;
  266.      strncpy(defaultdomain, (char *)p, MAX_STRING);
  267.      defaultdomain[MAX_STRING-1] = 0;
  268.      setdomainname(defaultdomain);
  269.   }
  270.   p += len+1;
  271.   break;
  272. #endif
  274.   // S. Lawson - handle DHCP type
  275.   case DHCP_VN_TYPE:
  276.   dhcp_seen=1;
  277.   if (!dhcp_bind) {
  278.      if (*(p+2)!=DHCP_TY_OFR) my_ip_addr=0L;
  279.   } else {
  280.      if (*(p+2)!=DHCP_TY_ACK) my_ip_addr=0L;
  281.   }
  282.   p += *(p+1)+2;
  283.   break;
  285.   // S. Lawson - handle server id
  286.   case DHCP_VN_SRVRID:
  287.   _dhcphost=intel(*(longword*)(&p[2]));
  288.   p += *(p+1)+2;
  289.   break;
  291.   // S. Lawson - handle lease expiration time
  292.   case DHCP_VN_T2TIME:
  293.   dhcp_life=intel(*(longword*)(&p[2]));
  294.   p += *(p+1)+2;
  295.   break;
  297.   case 255:
  298.    break;
  299.   default:
  300.    p +=*(p+1)+2;
  301.    break;
  302. } /* end of switch */
  303.      } /* end of while */
  304. }/* end of RFC_1048 if */
  305. if (dhcp_seen && my_ip_addr==0L) continue;  // S. Lawson
  306. if (!dhcp_seen || dhcp_bind)                // S. Lawson
  307.    goto give_up;
  309. // S. Lawson - append DHCPREQUEST and server/offered IP
  310. // addresses to vendor field.  We should get get a DHCPACK
  311. // in response
  312. dhcp_bind=1;
  313. p=&sendbootp.bp_vend[4];
  314. *p++=DHCP_VN_TYPE;
  315. *p++=1;
  316. *p++=DHCP_TY_REQ;
  317. movmem(dhcp_opt,p,sizeof(dhcp_opt));
  318. p+=sizeof(dhcp_opt);
  319. *p++=DHCP_VN_REQIP;
  320. *p++=4;
  321. l=(longword *) p, *l++=intel(my_ip_addr), p=(char *) l;
  322. *p++=DHCP_VN_SRVRID;
  323. *p++=4;
  324. l=(longword *) p, *l++=intel(_dhcphost), p=(char *) l;
  325. *p=255;
  326. my_ip_addr=0L;
  327. break;
  329.     }
  330. }
  331.     }
  332. give_up:
  334.     sock_close( (sock_type *)&bsock );
  336.     return (my_ip_addr == 0 );  /* return 0 on success */
  338. sock_err:
  339.     /* major network error if UDP fails */
  340.     sock_close( (sock_type *)&bsock );
  341.     return( -1 );
  342. }
  344. // S. Lawson - countdown the DHCP lease, renew or expire it as needed (mostly
  345. //             copied from above and trimmed down alot)
  346. int dhcp_expired(void) {
  347.    static char nested=false;
  348.    static char expired=false;
  349.    static char sockopen=false;
  350.    static longword timeout;
  351.    static longword xid=0;
  352.    static longword seconds=0, renew=0;
  353.    longword ltime;
  354.    word len;
  355.    char acknak; // -1 NAK, 0 unseen, 1 ACK
  356.    struct bootp _bootp;
  357.    unsigned char *p;
  358.    longword *l;
  360.    // once expired, stop talking
  361.    if (expired) return true;
  363.    // skip if nested, not dhcp, or the lease is permanent
  364.    if (nested || !dhcp_bind || dhcp_life==0xFFFFFFFFL) return false;
  365.    nested=true; // avoid tcp_tick() recursion
  367.    // on first pass, init the renewal time and start the timer
  368.    if (seconds==0) {
  369.       renew=dhcp_life>>1; // renew at 1/2 lifetime
  370.       if ((seconds=renew)>3600) seconds=3600;
  371.       timeout=set_timeout((int) seconds);
  372.       nested=false;
  373.       return false;
  374.    }
  376.    // if the bootp socket is open, check for a response
  377.    if (sockopen && (len = sock_dataready( (sock_type*)&bsock)) != 0 ) {
  378.      memset( &_bootp, 0, sizeof( struct bootp ));
  379.      len = sock_fastread( (sock_type*)&bsock, (byte *)&_bootp, sizeof( struct bootp ));
  380.      // (312-64) gives size of original bootp packet - struct bigger for dhcp
  381.      // if ( len >= sizeof( struct bootp ) - (312-64) &&
  382.      // JCT this is wrong size, fixed as
  383.      if ( len >= sizeof( struct bootp) - 312 &&
  385.   _bootp.bp_xid == xid &&
  386.   my_ip_addr == intel( _bootp.bp_yiaddr ) &&
  387.   intel( *(longword*)(&_bootp.bp_vend)) == VM_RFC1048 ) {
  388.        acknak=0;
  389.        ltime=0;
  390.        p=&_bootp.bp_vend[4];
  391.        while (*p!=255 && p<=&_bootp.bp_vend[311]) {
  392.  switch(*p) {
  393.    case 0:  /* Nop Pad character */
  394.      p++;
  395.      break;
  397.    case DHCP_VN_TYPE:   /* DHCP type */
  398.      if (*(p+2)==DHCP_TY_ACK) acknak=1;
  399.      if (*(p+2)==DHCP_TY_NAK) acknak=-1;
  400.      p += *(p+1)+2;
  401.      break;
  403.    case DHCP_VN_T2TIME: /* expiration time */
  404.      ltime=intel(*(longword*)(&p[2]));
  405.      p += *(p+1)+2;
  406.      break;
  408.    default:
  409.      p +=*(p+1)+2;
  410.      break;
  411.  } /* end of switch */
  412.        } /* end of while */
  413.        if (acknak==(char)-1) {                        // NAK!
  414.  dhcp_life=renew=seconds=0; //   fall through and expire
  415.        } else if (acknak==1 && ltime) { // ACK!
  416.  sock_close((sock_type*)&bsock);
  417.  sockopen=false;
  418.  dhcp_life=ltime;
  419.  renew=dhcp_life>>1; //   renew at 1/2 lifetime
  420.  if ((seconds=renew)>3600) seconds=3600;
  421.          timeout=set_timeout((int)seconds);
  422.  nested=false;
  423.  return false;
  424.        }
  425.      }
  426.    }
  428.    // count down the timeout till it expires
  429.    if (!chk_timeout(timeout)) {
  430.       nested=false;
  431.       return false;
  432.    }
  434.    // keep counting every hour (or less) till renewal time
  435.    renew-=seconds;
  436.    dhcp_life-=seconds;
  437.    if (renew>0) {
  438.       if ((seconds=renew)>3600) seconds=3600;
  439.       timeout=set_timeout((int) seconds);
  440.       nested=false;
  441.       return false;
  442.    }
  444.    // expire if dhcp_life gets below 2 seconds
  445.    if (dhcp_life<2) {
  446.       outs("nrDHCP lease expired, network disablednr");
  447.       if (sockopen) {
  448.  sock_close((sock_type*)&bsock);
  449.  sockopen=false;
  450.       }
  451.       // we really should should terminate all connections here
  452.       expired=true;
  453.       nested=false;
  454.       return true;
  455.    }
  457.    // send out a DHCPREQUEST to renew the lease and restart timer
  458.    if (!sockopen && !udp_open( &bsock, IPPORT_BOOTPC, _dhcphost, IPPORT_BOOTPS, NULL )) {
  459.      outs("nrUnable to resolve dhcp servernr");
  460.    } else {
  461.      sockopen=true;
  462.      memset( &_bootp, 0, sizeof( struct bootp ));
  463.      _bootp.bp_op = BOOTREQUEST;
  464.      _bootp.bp_htype = _pktdevclass;
  465.      if (_pktdevclass == PD_ETHER) _bootp.bp_hlen = 6;
  466.      _bootp.bp_ciaddr=intel(my_ip_addr);
  467.      if (xid==0) xid=intel(my_ip_addr);
  468.      _bootp.bp_xid = ++xid;
  469.      _bootp.bp_secs = intel16( 1 );
  470.      *(longword *)(&_bootp.bp_vend) = intel(VM_RFC1048);
  471.      p=&_bootp.bp_vend[4];
  472.      *p++=DHCP_VN_TYPE;
  473.      *p++=1;
  474.      *p++=DHCP_TY_REQ;
  475.      *p++=0;
  476.      *p++=DHCP_VN_REQIP;
  477.      *p++=4;
  478.      l=(longword *) p, *l++=intel(my_ip_addr), p=(char *) l;
  479.      *p++=DHCP_VN_SRVRID;
  480.      *p++=4;
  481.      l=(longword *) p, *l++=intel(_dhcphost), p=(char *) l;
  482.      *p=255;
  483.      movmem( &_eth_addr, _bootp.bp_chaddr, sizeof(eth_address));
  484.      sock_fastwrite( (sock_type*)&bsock, (byte *)&_bootp, sizeof( struct bootp ));
  485.    }
  486.    renew=dhcp_life>>1; // renew at 1/2 lifetime
  487.    if ((seconds=renew)>3600) seconds=3600;
  488.    timeout=set_timeout((int)seconds);
  489.    nested=false;
  490.    return false;
  491. }