Category

Development Platform
Home > SourceCode/Document > Internet-Socket-Network > TCP/IP Stack > View Code
ipsock.c
Package: export.zip　[view]
Upload User: hepax88
Upload Date: 2007-01-03
Package Size: 1101k
Code Size: 3k
Category:

TCP/IP Stack

Development Platform:

Visual C++

ipsock.c：Code Content
  1. #include "global.h"
  2. #include "mbuf.h"
  3. #include "ip.h"
  4. #include "usock.h"
  5. #include "socket.h"
  7. char Inet_eol[] = "rn";
  9. static void rip_recv(struct raw_ip *rp);
  10. static void autobind(struct usock *up);
  12. int
  13. so_ip_sock(up,protocol)
  14. struct usock *up;
  15. int protocol;
  16. {
  17. int s;
  19. s = up->index;
  20. up->cb.rip = raw_ip(protocol,rip_recv);
  21. up->cb.rip->user = s;
  22. return 0;
  23. }
  24. int
  25. so_ip_conn(up)
  26. struct usock *up;
  27. {
  28. if(up->name == NULL)
  29. autobind(up);
  30. return 0;
  31. }
  32. int
  33. so_ip_recv(up,bpp,from,fromlen)
  34. struct usock *up;
  35. struct mbuf **bpp;
  36. struct sockaddr *from;
  37. int *fromlen;
  38. {
  39. struct raw_ip *rip;
  40. struct sockaddr_in *remote;
  41. struct ip ip;
  42. int cnt;
  44. while((rip = up->cb.rip) != NULL && rip->rcvq == NULL){
  45. if(up->noblock){
  46. errno = EWOULDBLOCK;
  47. return -1;
  48. } else if((errno = kwait(up)) != 0){
  49. return -1;
  50. }
  51. }
  52. if(rip == NULL){
  53. /* Connection went away */
  54. errno = ENOTCONN;
  55. return -1;
  56. }
  57. *bpp = dequeue(&rip->rcvq);
  58. ntohip(&ip,bpp);
  60. cnt = len_p(*bpp);
  61. if(from != NULL && fromlen != (int *)NULL && *fromlen >= SOCKSIZE){
  62. remote = (struct sockaddr_in *)from;
  63. remote->sin_family = AF_INET;
  64. remote->sin_addr.s_addr = ip.source;
  65. remote->sin_port = 0;
  66. *fromlen = SOCKSIZE;
  67. }
  68. return cnt;
  69. }
  70. int
  71. so_ip_send(
  72. struct usock *up,
  73. struct mbuf **bpp,
  74. struct sockaddr *to
  75. ){
  76. struct sockaddr_in *local,*remote;
  78. if(up->name == NULL)
  79. autobind(up);
  80. local = (struct sockaddr_in *)up->name;
  81. if(to != NULL){
  82. remote = (struct sockaddr_in *)to;
  83. } else if(up->peername != NULL) {
  84. remote = (struct sockaddr_in *)up->peername;
  85. } else {
  86. free_p(bpp);
  87. errno = ENOTCONN;
  88. return -1;
  89. }
  90. ip_send(local->sin_addr.s_addr,remote->sin_addr.s_addr,
  91. (char)up->cb.rip->protocol,0,0,bpp,0,0,0);
  92. return 0;
  93. }
  94. int
  95. so_ip_qlen(up,rtx)
  96. struct usock *up;
  97. int rtx;
  98. {
  99. int len;
  101. switch(rtx){
  102. case 0:
  103. len = len_q(up->cb.rip->rcvq);
  104. break;
  105. case 1:
  106. len = 0;
  107. break;
  108. }
  109. return len;
  110. }
  111. int
  112. so_ip_close(up)
  113. struct usock *up;
  114. {
  115. del_ip(up->cb.rip);
  116. return 0;
  117. }
  118. int
  119. checkipaddr(name,namelen)
  120. struct sockaddr *name;
  121. int namelen;
  122. {
  123. struct sockaddr_in *sock;
  125. sock = (struct sockaddr_in *)name;
  126. if(sock->sin_family != AF_INET || namelen != sizeof(struct sockaddr_in))
  127. return -1;
  128. return 0;
  129. }
  131. /* Raw IP receive upcall routine */
  132. static void
  133. rip_recv(rp)
  134. struct raw_ip *rp;
  135. {
  136. ksignal(itop(rp->user),1);
  137. kwait(NULL);
  138. }
  139. /* Issue an automatic bind of a local address */
  140. static void
  141. autobind(up)
  142. struct usock *up;
  143. {
  144. struct sockaddr_in local;
  145. int s;
  147. s = up->index;
  148. local.sin_family = AF_INET;
  149. local.sin_addr.s_addr = INADDR_ANY;
  150. local.sin_port = Lport++;
  151. bind(s,(struct sockaddr *)&local,sizeof(struct sockaddr_in));
  152. }
  153. char *
  154. ippsocket(p)
  155. struct sockaddr *p;
  156. {
  157. struct sockaddr_in *sp;
  158. struct socket socket;
  159. static char buf[30];
  161. sp = (struct sockaddr_in *)p;
  162. socket.address = sp->sin_addr.s_addr;
  163. socket.port = sp->sin_port;
  164. strcpy(buf,pinet(&socket));
  166. return buf;
  167. }