protocol.c
上传用户:lgb322
上传日期:2013-02-24
资源大小:30529k
文件大小:4k
源码类别:

嵌入式Linux

开发平台:

Unix_Linux

  1. /*
  2.  * INET An implementation of the TCP/IP protocol suite for the LINUX
  3.  * operating system.  INET is implemented using the  BSD Socket
  4.  * interface as the means of communication with the user level.
  5.  *
  6.  * INET protocol dispatch tables.
  7.  *
  8.  * Version: $Id: protocol.c,v 1.14 2001/05/18 02:25:49 davem Exp $
  9.  *
  10.  * Authors: Ross Biro, <bir7@leland.Stanford.Edu>
  11.  * Fred N. van Kempen, <waltje@uWalt.NL.Mugnet.ORG>
  12.  *
  13.  * Fixes:
  14.  * Alan Cox : Ahah! udp icmp errors don't work because
  15.  *   udp_err is never called!
  16.  * Alan Cox : Added new fields for init and ready for
  17.  *   proper fragmentation (_NO_ 4K limits!)
  18.  * Richard Colella : Hang on hash collision
  19.  * Vince Laviano : Modified inet_del_protocol() to correctly
  20.  *   maintain copy bit.
  21.  *
  22.  * This program is free software; you can redistribute it and/or
  23.  * modify it under the terms of the GNU General Public License
  24.  * as published by the Free Software Foundation; either version
  25.  * 2 of the License, or (at your option) any later version.
  26.  */
  27. #include <asm/uaccess.h>
  28. #include <asm/system.h>
  29. #include <linux/types.h>
  30. #include <linux/kernel.h>
  31. #include <linux/sched.h>
  32. #include <linux/string.h>
  33. #include <linux/config.h>
  34. #include <linux/socket.h>
  35. #include <linux/in.h>
  36. #include <linux/inet.h>
  37. #include <linux/netdevice.h>
  38. #include <linux/timer.h>
  39. #include <linux/brlock.h>
  40. #include <net/ip.h>
  41. #include <net/protocol.h>
  42. #include <net/tcp.h>
  43. #include <linux/skbuff.h>
  44. #include <net/sock.h>
  45. #include <net/icmp.h>
  46. #include <net/udp.h>
  47. #include <net/ipip.h>
  48. #include <linux/igmp.h>
  49. #define IPPROTO_PREVIOUS NULL
  50. #ifdef CONFIG_IP_MULTICAST
  51. static struct inet_protocol igmp_protocol = {
  52. handler: igmp_rcv,
  53. next: IPPROTO_PREVIOUS,
  54. protocol: IPPROTO_IGMP,
  55. name: "IGMP"
  56. };
  57. #undef  IPPROTO_PREVIOUS
  58. #define IPPROTO_PREVIOUS &igmp_protocol
  59. #endif
  60. static struct inet_protocol tcp_protocol = {
  61. handler: tcp_v4_rcv,
  62. err_handler: tcp_v4_err,
  63. next: IPPROTO_PREVIOUS,
  64. protocol: IPPROTO_TCP,
  65. name: "TCP"
  66. };
  67. #undef  IPPROTO_PREVIOUS
  68. #define IPPROTO_PREVIOUS &tcp_protocol
  69. static struct inet_protocol udp_protocol = {
  70. handler: udp_rcv,
  71. err_handler: udp_err,
  72. next: IPPROTO_PREVIOUS,
  73. protocol: IPPROTO_UDP,
  74. name: "UDP"
  75. };
  76. #undef  IPPROTO_PREVIOUS
  77. #define IPPROTO_PREVIOUS &udp_protocol
  78. static struct inet_protocol icmp_protocol = {
  79. handler: icmp_rcv,
  80. next: IPPROTO_PREVIOUS,
  81. protocol: IPPROTO_ICMP,
  82. name: "ICMP"
  83. };
  84. #undef  IPPROTO_PREVIOUS
  85. #define IPPROTO_PREVIOUS &icmp_protocol
  86. struct inet_protocol *inet_protocol_base = IPPROTO_PREVIOUS;
  87. struct inet_protocol *inet_protos[MAX_INET_PROTOS];
  88. /*
  89.  * Add a protocol handler to the hash tables
  90.  */
  91. void inet_add_protocol(struct inet_protocol *prot)
  92. {
  93. unsigned char hash;
  94. struct inet_protocol *p2;
  95. hash = prot->protocol & (MAX_INET_PROTOS - 1);
  96. br_write_lock_bh(BR_NETPROTO_LOCK);
  97. prot ->next = inet_protos[hash];
  98. inet_protos[hash] = prot;
  99. prot->copy = 0;
  100. /*
  101.  * Set the copy bit if we need to. 
  102.  */
  103.  
  104. p2 = (struct inet_protocol *) prot->next;
  105. while (p2) {
  106. if (p2->protocol == prot->protocol) {
  107. prot->copy = 1;
  108. break;
  109. }
  110. p2 = (struct inet_protocol *) p2->next;
  111. }
  112. br_write_unlock_bh(BR_NETPROTO_LOCK);
  113. }
  114. /*
  115.  * Remove a protocol from the hash tables.
  116.  */
  117.  
  118. int inet_del_protocol(struct inet_protocol *prot)
  119. {
  120. struct inet_protocol *p;
  121. struct inet_protocol *lp = NULL;
  122. unsigned char hash;
  123. hash = prot->protocol & (MAX_INET_PROTOS - 1);
  124. br_write_lock_bh(BR_NETPROTO_LOCK);
  125. if (prot == inet_protos[hash]) {
  126. inet_protos[hash] = (struct inet_protocol *) inet_protos[hash]->next;
  127. br_write_unlock_bh(BR_NETPROTO_LOCK);
  128. return 0;
  129. }
  130. p = (struct inet_protocol *) inet_protos[hash];
  131. if (p != NULL && p->protocol == prot->protocol)
  132. lp = p;
  133. while (p) {
  134. /*
  135.  * We have to worry if the protocol being deleted is
  136.  * the last one on the list, then we may need to reset
  137.  * someone's copied bit.
  138.  */
  139. if (p->next && p->next == prot) {
  140. /*
  141.  * if we are the last one with this protocol and
  142.  * there is a previous one, reset its copy bit.
  143.  */
  144. if (prot->copy == 0 && lp != NULL)
  145. lp->copy = 0;
  146. p->next = prot->next;
  147. br_write_unlock_bh(BR_NETPROTO_LOCK);
  148. return 0;
  149. }
  150. if (p->next != NULL && p->next->protocol == prot->protocol) 
  151. lp = p->next;
  152. p = (struct inet_protocol *) p->next;
  153. }
  154. br_write_unlock_bh(BR_NETPROTO_LOCK);
  155. return -1;
  156. }