home *** CD-ROM | disk | FTP | other *** search
/ ftp.muug.mb.ca / 2014.06.ftp.muug.mb.ca.tar / ftp.muug.mb.ca / pub / src / traceroute / vj_traceroute / raw_ip.diff < prev    next >
Text File  |  1989-03-04  |  3KB  |  126 lines
  1. *** rip_output.original    Sat Mar  4 23:08:29 1989
  2. --- rip_output.van    Sat Mar  4 23:08:57 1989
  3. ***************
  4. *** 1,40 ****
  5. ! /*
  6. !  * Generate IP header and pass packet to ip_output.
  7. !  * Tack on options user may have setup with control call.
  8. !  */
  9. ! rip_output(m0, so)
  10. !     struct mbuf *m0;
  11.       struct socket *so;
  12.   {
  13. -     register struct mbuf *m;
  14.       register struct ip *ip;
  15. !     int len = 0, error;
  16.       struct rawcb *rp = sotorawcb(so);
  17.       struct sockaddr_in *sin;
  18. ! 
  19.       /*
  20. !      * Calculate data length and get an mbuf
  21. !      * for IP header.
  22.        */
  23. !     for (m = m0; m; m = m->m_next)
  24. !         len += m->m_len;
  25. !     m = m_get(M_DONTWAIT, MT_HEADER);
  26. !     if (m == 0) {
  27. !         error = ENOBUFS;
  28. !         goto bad;
  29. !     }
  30. !     
  31. !     /*
  32. !      * Fill in IP header as needed.
  33. !      */
  34. !     m->m_off = MMAXOFF - sizeof(struct ip);
  35. !     m->m_len = sizeof(struct ip);
  36. !     m->m_next = m0;
  37. !     ip = mtod(m, struct ip *);
  38. !     ip->ip_tos = 0;
  39. !     ip->ip_off = 0;
  40. !     ip->ip_p = rp->rcb_proto.sp_protocol;
  41. !     ip->ip_len = sizeof(struct ip) + len;
  42.       if (rp->rcb_flags & RAW_LADDR) {
  43.           sin = (struct sockaddr_in *)&rp->rcb_laddr;
  44.           if (sin->sin_family != AF_INET) {
  45. --- 1,51 ----
  46. ! rip_output(m, so)
  47. !     register struct mbuf *m;
  48.       struct socket *so;
  49.   {
  50.       register struct ip *ip;
  51. !     int error;
  52.       struct rawcb *rp = sotorawcb(so);
  53.       struct sockaddr_in *sin;
  54. ! #if BSD>=43
  55. !     short proto = rp->rcb_proto.sp_protocol;
  56. ! #else
  57. !     short proto = so->so_proto->pr_protocol;
  58. ! #endif
  59.       /*
  60. !      * if the protocol is IPPROTO_RAW, the user handed us a 
  61. !      * complete IP packet.  Otherwise, allocate an mbuf for a
  62. !      * header and fill it in as needed.
  63.        */
  64. !     if (proto != IPPROTO_RAW) {
  65. !         /*
  66. !          * Calculate data length and get an mbuf
  67. !          * for IP header.
  68. !          */
  69. !         int len = 0;
  70. !         struct mbuf *m0;
  71. ! 
  72. !         for (m0 = m; m; m = m->m_next)
  73. !             len += m->m_len;
  74. ! 
  75. !         m = m_get(M_DONTWAIT, MT_HEADER);
  76. !         if (m == 0) {
  77. !             m = m0;
  78. !             error = ENOBUFS;
  79. !             goto bad;
  80. !         }
  81. !         m->m_off = MMAXOFF - sizeof(struct ip);
  82. !         m->m_len = sizeof(struct ip);
  83. !         m->m_next = m0;
  84. ! 
  85. !         ip = mtod(m, struct ip *);
  86. !         ip->ip_tos = 0;
  87. !         ip->ip_off = 0;
  88. !         ip->ip_p = proto;
  89. !         ip->ip_len = sizeof(struct ip) + len;
  90. !         ip->ip_ttl = MAXTTL;
  91. !     } else
  92. !         ip = mtod(m, struct ip *);
  93. ! 
  94.       if (rp->rcb_flags & RAW_LADDR) {
  95.           sin = (struct sockaddr_in *)&rp->rcb_laddr;
  96.           if (sin->sin_family != AF_INET) {
  97. ***************
  98. *** 44,53 ****
  99.           ip->ip_src.s_addr = sin->sin_addr.s_addr;
  100.       } else
  101.           ip->ip_src.s_addr = 0;
  102.       ip->ip_dst = ((struct sockaddr_in *)&rp->rcb_faddr)->sin_addr;
  103. !     ip->ip_ttl = MAXTTL;
  104.       return (ip_output(m, rp->rcb_options, &rp->rcb_route, 
  105.          (so->so_options & SO_DONTROUTE) | IP_ALLOWBROADCAST));
  106.   bad:
  107.       m_freem(m);
  108.       return (error);
  109. --- 55,70 ----
  110.           ip->ip_src.s_addr = sin->sin_addr.s_addr;
  111.       } else
  112.           ip->ip_src.s_addr = 0;
  113. + 
  114.       ip->ip_dst = ((struct sockaddr_in *)&rp->rcb_faddr)->sin_addr;
  115. ! 
  116. ! #if BSD>=43
  117.       return (ip_output(m, rp->rcb_options, &rp->rcb_route, 
  118.          (so->so_options & SO_DONTROUTE) | IP_ALLOWBROADCAST));
  119. + #else
  120. +     return (ip_output(m, (struct mbuf *)0, &rp->rcb_route, 
  121. +        (so->so_options & SO_DONTROUTE) | IP_ALLOWBROADCAST));
  122. + #endif
  123.   bad:
  124.       m_freem(m);
  125.       return (error);
  126.