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

嵌入式Linux

开发平台:

Unix_Linux

  1. /*
  2.  * common UDP/RAW code
  3.  * Linux INET6 implementation 
  4.  *
  5.  * Authors:
  6.  * Pedro Roque <roque@di.fc.ul.pt>
  7.  *
  8.  * $Id: datagram.c,v 1.23 2001/09/01 00:31:50 davem Exp $
  9.  *
  10.  * This program is free software; you can redistribute it and/or
  11.  *      modify it under the terms of the GNU General Public License
  12.  *      as published by the Free Software Foundation; either version
  13.  *      2 of the License, or (at your option) any later version.
  14.  */
  15. #include <linux/errno.h>
  16. #include <linux/types.h>
  17. #include <linux/kernel.h>
  18. #include <linux/sched.h>
  19. #include <linux/interrupt.h>
  20. #include <linux/socket.h>
  21. #include <linux/sockios.h>
  22. #include <linux/in6.h>
  23. #include <linux/ipv6.h>
  24. #include <linux/route.h>
  25. #include <net/ipv6.h>
  26. #include <net/ndisc.h>
  27. #include <net/addrconf.h>
  28. #include <net/transp_v6.h>
  29. #include <linux/errqueue.h>
  30. #include <asm/uaccess.h>
  31. void ipv6_icmp_error(struct sock *sk, struct sk_buff *skb, int err, 
  32.      u16 port, u32 info, u8 *payload)
  33. {
  34. struct icmp6hdr *icmph = (struct icmp6hdr *)skb->h.raw;
  35. struct sock_exterr_skb *serr;
  36. if (!sk->net_pinfo.af_inet6.recverr)
  37. return;
  38. skb = skb_clone(skb, GFP_ATOMIC);
  39. if (!skb)
  40. return;
  41. serr = SKB_EXT_ERR(skb);
  42. serr->ee.ee_errno = err;
  43. serr->ee.ee_origin = SO_EE_ORIGIN_ICMP6;
  44. serr->ee.ee_type = icmph->icmp6_type; 
  45. serr->ee.ee_code = icmph->icmp6_code;
  46. serr->ee.ee_pad = 0;
  47. serr->ee.ee_info = info;
  48. serr->ee.ee_data = 0;
  49. serr->addr_offset = (u8*)&(((struct ipv6hdr*)(icmph+1))->daddr) - skb->nh.raw;
  50. serr->port = port;
  51. skb->h.raw = payload;
  52. __skb_pull(skb, payload - skb->data);
  53. if (sock_queue_err_skb(sk, skb))
  54. kfree_skb(skb);
  55. }
  56. void ipv6_local_error(struct sock *sk, int err, struct flowi *fl, u32 info)
  57. {
  58. struct sock_exterr_skb *serr;
  59. struct ipv6hdr *iph;
  60. struct sk_buff *skb;
  61. if (!sk->net_pinfo.af_inet6.recverr)
  62. return;
  63. skb = alloc_skb(sizeof(struct ipv6hdr), GFP_ATOMIC);
  64. if (!skb)
  65. return;
  66. iph = (struct ipv6hdr*)skb_put(skb, sizeof(struct ipv6hdr));
  67. skb->nh.ipv6h = iph;
  68. memcpy(&iph->daddr, fl->fl6_dst, 16);
  69. serr = SKB_EXT_ERR(skb);
  70. serr->ee.ee_errno = err;
  71. serr->ee.ee_origin = SO_EE_ORIGIN_LOCAL;
  72. serr->ee.ee_type = 0; 
  73. serr->ee.ee_code = 0;
  74. serr->ee.ee_pad = 0;
  75. serr->ee.ee_info = info;
  76. serr->ee.ee_data = 0;
  77. serr->addr_offset = (u8*)&iph->daddr - skb->nh.raw;
  78. serr->port = fl->uli_u.ports.dport;
  79. skb->h.raw = skb->tail;
  80. __skb_pull(skb, skb->tail - skb->data);
  81. if (sock_queue_err_skb(sk, skb))
  82. kfree_skb(skb);
  83. }
  84. /* 
  85.  * Handle MSG_ERRQUEUE
  86.  */
  87. int ipv6_recv_error(struct sock *sk, struct msghdr *msg, int len)
  88. {
  89. struct sock_exterr_skb *serr;
  90. struct sk_buff *skb, *skb2;
  91. struct sockaddr_in6 *sin;
  92. struct {
  93. struct sock_extended_err ee;
  94. struct sockaddr_in6  offender;
  95. } errhdr;
  96. int err;
  97. int copied;
  98. err = -EAGAIN;
  99. skb = skb_dequeue(&sk->error_queue);
  100. if (skb == NULL)
  101. goto out;
  102. copied = skb->len;
  103. if (copied > len) {
  104. msg->msg_flags |= MSG_TRUNC;
  105. copied = len;
  106. }
  107. err = skb_copy_datagram_iovec(skb, 0, msg->msg_iov, copied);
  108. if (err)
  109. goto out_free_skb;
  110. sock_recv_timestamp(msg, sk, skb);
  111. serr = SKB_EXT_ERR(skb);
  112. sin = (struct sockaddr_in6 *)msg->msg_name;
  113. if (sin) {
  114. sin->sin6_family = AF_INET6;
  115. sin->sin6_flowinfo = 0;
  116. sin->sin6_port = serr->port; 
  117. sin->sin6_scope_id = 0;
  118. if (serr->ee.ee_origin == SO_EE_ORIGIN_ICMP6) {
  119. memcpy(&sin->sin6_addr, skb->nh.raw + serr->addr_offset, 16);
  120. if (sk->net_pinfo.af_inet6.sndflow)
  121. sin->sin6_flowinfo = *(u32*)(skb->nh.raw + serr->addr_offset - 24) & IPV6_FLOWINFO_MASK;
  122. if (ipv6_addr_type(&sin->sin6_addr) & IPV6_ADDR_LINKLOCAL) {
  123. struct inet6_skb_parm *opt = (struct inet6_skb_parm *) skb->cb;
  124. sin->sin6_scope_id = opt->iif;
  125. }
  126. } else {
  127. ipv6_addr_set(&sin->sin6_addr, 0, 0,
  128.       __constant_htonl(0xffff),
  129.       *(u32*)(skb->nh.raw + serr->addr_offset));
  130. }
  131. }
  132. memcpy(&errhdr.ee, &serr->ee, sizeof(struct sock_extended_err));
  133. sin = &errhdr.offender;
  134. sin->sin6_family = AF_UNSPEC;
  135. if (serr->ee.ee_origin != SO_EE_ORIGIN_LOCAL) {
  136. sin->sin6_family = AF_INET6;
  137. sin->sin6_flowinfo = 0;
  138. if (serr->ee.ee_origin == SO_EE_ORIGIN_ICMP6) {
  139. memcpy(&sin->sin6_addr, &skb->nh.ipv6h->saddr, 16);
  140. if (sk->net_pinfo.af_inet6.rxopt.all)
  141. datagram_recv_ctl(sk, msg, skb);
  142. if (ipv6_addr_type(&sin->sin6_addr) & IPV6_ADDR_LINKLOCAL) {
  143. struct inet6_skb_parm *opt = (struct inet6_skb_parm *) skb->cb;
  144. sin->sin6_scope_id = opt->iif;
  145. }
  146. } else {
  147. ipv6_addr_set(&sin->sin6_addr, 0, 0,
  148.       __constant_htonl(0xffff),
  149.       skb->nh.iph->saddr);
  150. if (sk->protinfo.af_inet.cmsg_flags)
  151. ip_cmsg_recv(msg, skb);
  152. }
  153. }
  154. put_cmsg(msg, SOL_IPV6, IPV6_RECVERR, sizeof(errhdr), &errhdr);
  155. /* Now we could try to dump offended packet options */
  156. msg->msg_flags |= MSG_ERRQUEUE;
  157. err = copied;
  158. /* Reset and regenerate socket error */
  159. spin_lock_irq(&sk->error_queue.lock);
  160. sk->err = 0;
  161. if ((skb2 = skb_peek(&sk->error_queue)) != NULL) {
  162. sk->err = SKB_EXT_ERR(skb2)->ee.ee_errno;
  163. spin_unlock_irq(&sk->error_queue.lock);
  164. sk->error_report(sk);
  165. } else {
  166. spin_unlock_irq(&sk->error_queue.lock);
  167. }
  168. out_free_skb:
  169. kfree_skb(skb);
  170. out:
  171. return err;
  172. }
  173. int datagram_recv_ctl(struct sock *sk, struct msghdr *msg, struct sk_buff *skb)
  174. {
  175. struct ipv6_pinfo *np = &sk->net_pinfo.af_inet6;
  176. struct inet6_skb_parm *opt = (struct inet6_skb_parm *) skb->cb;
  177. if (np->rxopt.bits.rxinfo) {
  178. struct in6_pktinfo src_info;
  179. src_info.ipi6_ifindex = opt->iif;
  180. ipv6_addr_copy(&src_info.ipi6_addr, &skb->nh.ipv6h->daddr);
  181. put_cmsg(msg, SOL_IPV6, IPV6_PKTINFO, sizeof(src_info), &src_info);
  182. }
  183. if (np->rxopt.bits.rxhlim) {
  184. int hlim = skb->nh.ipv6h->hop_limit;
  185. put_cmsg(msg, SOL_IPV6, IPV6_HOPLIMIT, sizeof(hlim), &hlim);
  186. }
  187. if (np->rxopt.bits.rxflow && (*(u32*)skb->nh.raw & IPV6_FLOWINFO_MASK)) {
  188. u32 flowinfo = *(u32*)skb->nh.raw & IPV6_FLOWINFO_MASK;
  189. put_cmsg(msg, SOL_IPV6, IPV6_FLOWINFO, sizeof(flowinfo), &flowinfo);
  190. }
  191. if (np->rxopt.bits.hopopts && opt->hop) {
  192. u8 *ptr = skb->nh.raw + opt->hop;
  193. put_cmsg(msg, SOL_IPV6, IPV6_HOPOPTS, (ptr[1]+1)<<3, ptr);
  194. }
  195. if (np->rxopt.bits.dstopts && opt->dst0) {
  196. u8 *ptr = skb->nh.raw + opt->dst0;
  197. put_cmsg(msg, SOL_IPV6, IPV6_DSTOPTS, (ptr[1]+1)<<3, ptr);
  198. }
  199. if (np->rxopt.bits.srcrt && opt->srcrt) {
  200. struct ipv6_rt_hdr *rthdr = (struct ipv6_rt_hdr *)(skb->nh.raw + opt->srcrt);
  201. put_cmsg(msg, SOL_IPV6, IPV6_RTHDR, (rthdr->hdrlen+1) << 3, rthdr);
  202. }
  203. if (np->rxopt.bits.authhdr && opt->auth) {
  204. u8 *ptr = skb->nh.raw + opt->auth;
  205. put_cmsg(msg, SOL_IPV6, IPV6_AUTHHDR, (ptr[1]+1)<<2, ptr);
  206. }
  207. if (np->rxopt.bits.dstopts && opt->dst1) {
  208. u8 *ptr = skb->nh.raw + opt->dst1;
  209. put_cmsg(msg, SOL_IPV6, IPV6_DSTOPTS, (ptr[1]+1)<<3, ptr);
  210. }
  211. return 0;
  212. }
  213. int datagram_send_ctl(struct msghdr *msg, struct flowi *fl,
  214.       struct ipv6_txoptions *opt,
  215.       int *hlimit)
  216. {
  217. struct in6_pktinfo *src_info;
  218. struct cmsghdr *cmsg;
  219. struct ipv6_rt_hdr *rthdr;
  220. struct ipv6_opt_hdr *hdr;
  221. int len;
  222. int err = 0;
  223. for (cmsg = CMSG_FIRSTHDR(msg); cmsg; cmsg = CMSG_NXTHDR(msg, cmsg)) {
  224. if (cmsg->cmsg_len < sizeof(struct cmsghdr) ||
  225.     (unsigned long)(((char*)cmsg - (char*)msg->msg_control)
  226.     + cmsg->cmsg_len) > msg->msg_controllen) {
  227. err = -EINVAL;
  228. goto exit_f;
  229. }
  230. if (cmsg->cmsg_level != SOL_IPV6)
  231. continue;
  232. switch (cmsg->cmsg_type) {
  233.   case IPV6_PKTINFO:
  234.   if (cmsg->cmsg_len < CMSG_LEN(sizeof(struct in6_pktinfo))) {
  235. err = -EINVAL;
  236. goto exit_f;
  237. }
  238. src_info = (struct in6_pktinfo *)CMSG_DATA(cmsg);
  239. if (src_info->ipi6_ifindex) {
  240. if (fl->oif && src_info->ipi6_ifindex != fl->oif)
  241. return -EINVAL;
  242. fl->oif = src_info->ipi6_ifindex;
  243. }
  244. if (!ipv6_addr_any(&src_info->ipi6_addr)) {
  245. if (!ipv6_chk_addr(&src_info->ipi6_addr, NULL)) {
  246. err = -EINVAL;
  247. goto exit_f;
  248. }
  249. fl->fl6_src = &src_info->ipi6_addr;
  250. }
  251. break;
  252. case IPV6_FLOWINFO:
  253.                         if (cmsg->cmsg_len < CMSG_LEN(4)) {
  254. err = -EINVAL;
  255. goto exit_f;
  256. }
  257. if (fl->fl6_flowlabel&IPV6_FLOWINFO_MASK) {
  258. if ((fl->fl6_flowlabel^*(u32 *)CMSG_DATA(cmsg))&~IPV6_FLOWINFO_MASK) {
  259. err = -EINVAL;
  260. goto exit_f;
  261. }
  262. }
  263. fl->fl6_flowlabel = IPV6_FLOWINFO_MASK & *(u32 *)CMSG_DATA(cmsg);
  264. break;
  265. case IPV6_HOPOPTS:
  266.                         if (opt->hopopt || cmsg->cmsg_len < CMSG_LEN(sizeof(struct ipv6_opt_hdr))) {
  267. err = -EINVAL;
  268. goto exit_f;
  269. }
  270. hdr = (struct ipv6_opt_hdr *)CMSG_DATA(cmsg);
  271. len = ((hdr->hdrlen + 1) << 3);
  272. if (cmsg->cmsg_len < CMSG_LEN(len)) {
  273. err = -EINVAL;
  274. goto exit_f;
  275. }
  276. if (!capable(CAP_NET_RAW)) {
  277. err = -EPERM;
  278. goto exit_f;
  279. }
  280. opt->opt_nflen += len;
  281. opt->hopopt = hdr;
  282. break;
  283. case IPV6_DSTOPTS:
  284.                         if (cmsg->cmsg_len < CMSG_LEN(sizeof(struct ipv6_opt_hdr))) {
  285. err = -EINVAL;
  286. goto exit_f;
  287. }
  288. hdr = (struct ipv6_opt_hdr *)CMSG_DATA(cmsg);
  289. len = ((hdr->hdrlen + 1) << 3);
  290. if (cmsg->cmsg_len < CMSG_LEN(len)) {
  291. err = -EINVAL;
  292. goto exit_f;
  293. }
  294. if (!capable(CAP_NET_RAW)) {
  295. err = -EPERM;
  296. goto exit_f;
  297. }
  298. if (opt->dst1opt) {
  299. err = -EINVAL;
  300. goto exit_f;
  301. }
  302. opt->opt_flen += len;
  303. opt->dst1opt = hdr;
  304. break;
  305. case IPV6_AUTHHDR:
  306.                         if (cmsg->cmsg_len < CMSG_LEN(sizeof(struct ipv6_opt_hdr))) {
  307. err = -EINVAL;
  308. goto exit_f;
  309. }
  310. hdr = (struct ipv6_opt_hdr *)CMSG_DATA(cmsg);
  311. len = ((hdr->hdrlen + 2) << 2);
  312. if (cmsg->cmsg_len < CMSG_LEN(len)) {
  313. err = -EINVAL;
  314. goto exit_f;
  315. }
  316. if (len & ~7) {
  317. err = -EINVAL;
  318. goto exit_f;
  319. }
  320. opt->opt_flen += len;
  321. opt->auth = hdr;
  322. break;
  323. case IPV6_RTHDR:
  324.                         if (cmsg->cmsg_len < CMSG_LEN(sizeof(struct ipv6_rt_hdr))) {
  325. err = -EINVAL;
  326. goto exit_f;
  327. }
  328. rthdr = (struct ipv6_rt_hdr *)CMSG_DATA(cmsg);
  329. /*
  330.  * TYPE 0
  331.  */
  332. if (rthdr->type) {
  333. err = -EINVAL;
  334. goto exit_f;
  335. }
  336. len = ((rthdr->hdrlen + 1) << 3);
  337.                         if (cmsg->cmsg_len < CMSG_LEN(len)) {
  338. err = -EINVAL;
  339. goto exit_f;
  340. }
  341. /* segments left must also match */
  342. if ((rthdr->hdrlen >> 1) != rthdr->segments_left) {
  343. err = -EINVAL;
  344. goto exit_f;
  345. }
  346. opt->opt_nflen += len;
  347. opt->srcrt = rthdr;
  348. if (opt->dst1opt) {
  349. int dsthdrlen = ((opt->dst1opt->hdrlen+1)<<3);
  350. opt->opt_nflen += dsthdrlen;
  351. opt->dst0opt = opt->dst1opt;
  352. opt->dst1opt = NULL;
  353. opt->opt_flen -= dsthdrlen;
  354. }
  355. break;
  356. case IPV6_HOPLIMIT:
  357. if (cmsg->cmsg_len != CMSG_LEN(sizeof(int))) {
  358. err = -EINVAL;
  359. goto exit_f;
  360. }
  361. *hlimit = *(int *)CMSG_DATA(cmsg);
  362. break;
  363. default:
  364. if (net_ratelimit())
  365. printk(KERN_DEBUG "invalid cmsg type: %dn", cmsg->cmsg_type);
  366. err = -EINVAL;
  367. break;
  368. };
  369. }
  370. exit_f:
  371. return err;
  372. }