1 #include <linux/module.h>
2 #include <linux/version.h>
4 #include <net/protocol.h>
11 MODULE_DESCRIPTION("Swift Transport Protocol");
12 MODULE_AUTHOR("Adrian Bondrescu/Cornel Mercan");
13 MODULE_LICENSE("GPL");
16 struct inet_sock sock;
17 /* swift socket speciffic data */
22 static struct swift_sock * sock_port_map[MAX_SWIFT_PORT];
24 static inline struct swift_sock * swift_sk(struct sock * sock)
26 return (struct swift_sock *)(sock);
29 static inline struct swifthdr * swift_hdr(const struct sk_buff * skb)
31 return (struct swifthdr *) skb_transport_header(skb);
34 static inline uint8_t get_next_free_port(void)
37 for (i = MIN_SWIFT_PORT; i < MAX_SWIFT_PORT; i ++)
38 if (sock_port_map[i] == NULL)
43 static inline void swift_unhash(uint8_t port)
45 sock_port_map[port] = NULL;
48 static inline void swift_hash(uint8_t port, struct swift_sock *ssh)
50 sock_port_map[port] = ssh;
53 static inline struct swift_sock * swift_lookup(uint8_t port)
55 return sock_port_map[port];
58 static int swift_release(struct socket *sock)
60 struct sock *sk = sock->sk;
61 struct swift_sock * ssk = swift_sk(sk);
66 swift_unhash(ssk->src);
68 sock_prot_inuse_add(sock_net(sk), sk->sk_prot, -1);
75 skb_queue_purge(&sk->sk_receive_queue);
77 log_debug("swift_release sock=%p\n", sk);
83 static int swift_bind(struct socket *sock, struct sockaddr *addr, int addr_len)
85 struct sockaddr_swift *swift_addr;
86 struct swift_sock *ssk;
90 if (unlikely(addr_len < sizeof(struct sockaddr_swift))) {
91 log_error("Invalid size for sockaddr\n");
96 swift_addr = (struct sockaddr_swift *) addr;
98 port = swift_addr->dests[0].port;
100 if (unlikely(port == 0 || port >= MAX_SWIFT_PORT)) {
101 log_error("Invalid value for sockaddr port (%u)\n", port);
106 if (unlikely(swift_lookup(port) != NULL)) {
107 log_error("Port %u already in use\n", port);
112 ssk = swift_sk(sock->sk);
113 sock->sk->sk_rcvbuf = 10 * 1024 * 1024;
116 swift_hash(port, ssk);
118 log_debug("Socket %p bound to port %u\n", ssk, port);
126 static int swift_connect(struct socket *sock, struct sockaddr *addr, int addr_len, int flags)
130 struct inet_sock * isk;
131 struct swift_sock * ssk;
133 log_debug("swift_connect\n");
135 if (unlikely(sock == NULL)) {
136 log_error("Sock is NULL\n");
142 if (unlikely(sk == NULL)) {
143 log_error("Sock->sk is NULL\n");
151 if (unlikely(ssk->src != 0)) {
152 log_error("ssk->src is not NULL\n");
158 struct sockaddr_swift * swift_addr = (struct sockaddr_swift *) addr;
160 if (unlikely(addr_len < sizeof(*swift_addr) ||
161 addr_len < swift_addr->count * sizeof(struct swift_dest) ||
162 swift_addr->count <= 0)) {
163 log_error("Invalid size or address family\n");
167 ssk->dst = swift_addr->dests[0].port;
168 if (unlikely(ssk->dst == 0 || ssk->dst >= MAX_SWIFT_PORT)) {
169 log_error("Invalid value for destination port(%u)\n", ssk->dst);
174 isk->inet_daddr = swift_addr->dests[0].addr;
175 log_debug("Received from user space destination port=%u and address=%u\n", ssk->dst, isk->inet_daddr);
177 log_error("Invalid swift_addr (NULL)\n");
182 ssk->src = get_next_free_port();
183 if (unlikely(ssk->src == 0)) {
184 log_error("No free ports\n");
189 swift_hash(ssk->src, ssk);
197 static int swift_sendmsg(struct kiocb *iocb, struct socket *sock, struct msghdr *msg, size_t len)
203 struct sk_buff * skb;
205 struct inet_sock * isk;
206 struct swift_sock * ssk;
207 struct swifthdr * shdr;
210 struct rtable * rt = NULL;
213 struct sockaddr_swift * swift_addr = NULL;
215 if (unlikely(sock == NULL)) {
216 log_error("Sock is NULL\n");
222 if (unlikely(sk == NULL)) {
223 log_error("Sock->sk is NULL\n");
233 sport = get_next_free_port();
234 if (unlikely(sport == 0)) {
235 log_error("No free ports\n");
242 swift_addr = (struct sockaddr_swift *) msg->msg_name;
244 if (unlikely(msg->msg_namelen < sizeof(*swift_addr) ||
245 msg->msg_namelen < swift_addr->count * sizeof(struct swift_dest) ||
246 swift_addr->count <= 0)) {
247 log_error("Invalid size for msg_name\n");
252 dests = swift_addr->count;
255 if (unlikely(!ssk->dst || !isk->inet_daddr)) {
256 log_error("No destination port/address\n");
261 daddr = isk->inet_daddr;
263 log_debug("Got from socket destination port=%u and address=%u\n", dport, daddr);
267 if (msg->msg_iovlen < dests)
268 dests = msg->msg_iovlen;
270 for (i = 0; i < dests; i++) {
271 struct swift_dest *dest = &swift_addr->dests[i];
272 struct iovec *iov = &msg->msg_iov[i];
276 if (unlikely(dport == 0 || dport >= MAX_SWIFT_PORT)) {
277 log_error("Invalid value for destination port(%u)\n", dport);
283 log_debug("Received from user space destination port=%u and address=%u\n", dport, daddr);
286 totlen = len + sizeof(struct swifthdr) + sizeof(struct iphdr);
287 skb = sock_alloc_send_skb(sk, totlen, msg->msg_flags & MSG_DONTWAIT, &err);
288 if (unlikely(!skb)) {
289 log_error("sock_alloc_send_skb failed\n");
292 log_debug("Allocated %u bytes for skb (payload size=%u)\n", totlen, len);
294 skb_reset_network_header(skb);
295 skb_reserve(skb, sizeof(struct iphdr));
296 log_debug("Reseted network header\n");
297 skb_reset_transport_header(skb);
298 skb_put(skb, sizeof(struct swifthdr));
299 log_debug("Reseted transport header\n");
301 shdr = (struct swifthdr *) skb_transport_header(skb);
304 shdr->len = ntohs(len + sizeof(struct swifthdr));
306 payload = skb_put(skb, len);
307 log_debug("payload=%p\n", payload);
309 err = skb_copy_datagram_from_iovec(skb, sizeof(struct swifthdr), iov, 0, len);
311 log_error("skb_copy_datagram_from_iovec failed\n");
314 log_debug("Copied %u bytes into the skb\n", len);
317 rt = (struct rtable *) __sk_dst_check(sk, 0);
320 struct flowi fl = { .fl4_dst = daddr,
321 .proto = sk->sk_protocol,
322 .flags = inet_sk_flowi_flags(sk),
324 err = ip_route_output_flow(sock_net(sk), &rt, &fl, sk, 0);
326 log_error("Route lookup failed\n");
329 #if LINUX_VERSION_CODE < KERNEL_VERSION(2, 6, 36)
330 sk_dst_set(sk, dst_clone(&rt->u.dst));
332 sk_dst_set(sk, dst_clone(&rt->dst));
337 err = ip_queue_xmit(skb);
339 log_debug("Sent %u bytes on wire\n", len);
341 log_error("ip_queue_xmit failed\n");
353 static int swift_recvmsg(struct kiocb *iocb, struct socket *sock, struct msghdr *msg, size_t len, int flags)
356 struct sockaddr_swift *swift_addr;
357 struct sock * sk = sock->sk;
360 struct sockaddr_swift *ret_addr = (struct sockaddr_swift *) msg->msg_name;
362 log_debug("Trying to receive sock=%p sk=%p flags=%d\n", sock, sk, flags);
364 skb = skb_recv_datagram(sk, flags, flags & MSG_DONTWAIT, &err);
365 if (unlikely(!skb)) {
366 log_error("skb_recv_datagram\n");
370 for (i = 0; i < msg->msg_iovlen; i++) {
371 log_debug("Received skb %p\n", skb);
373 swift_addr = (struct sockaddr_swift *) skb->cb;
376 if (copied > msg->msg_iov[i].iov_len) {
377 copied = msg->msg_iov[i].iov_len;
378 msg->msg_flags |= MSG_TRUNC;
381 err = skb_copy_datagram_iovec(skb, 0, &msg->msg_iov[i], copied);
383 log_error("skb_copy_datagram_iovec\n");
387 sock_recv_ts_and_drops(msg, sk, skb);
390 memcpy(&ret_addr->dests[i], &swift_addr->dests[0], sizeof(ret_addr->dests[i]));
395 skb_free_datagram(sk, skb);
397 skb = skb_recv_datagram(sk, flags, 1, &err);
398 if (likely(err == -EAGAIN)) {
399 log_debug("No more skbs in the queue, returning...\n");
405 ret_addr->count = i + 1;
406 msg->msg_namelen = sizeof(struct sockaddr_swift) + (i + 1) * sizeof(struct swift_dest);
412 static int swift_rcv(struct sk_buff *skb)
414 struct swifthdr *shdr;
415 struct swift_sock *ssk;
418 struct sockaddr_swift * swift_addr;
420 int addr_size = sizeof(struct sockaddr_swift) + sizeof(struct swift_dest);
422 if (unlikely(!pskb_may_pull(skb, sizeof(struct swifthdr)))) {
423 log_error("Insufficient space for header\n");
427 shdr = (struct swifthdr *) skb->data;
428 len = ntohs(shdr->len);
430 if (unlikely(skb->len < len)) {
431 log_error("Malformed packet (packet_len=%u, skb_len=%u)\n", len, skb->len);
435 if (unlikely(len < sizeof(struct swifthdr))) {
436 log_error("Malformed packet (packet_len=%u sizeof(swifthdr)=%u\n", len, sizeof(struct swifthdr));
442 if (unlikely(src == 0 || dst == 0 || src >= MAX_SWIFT_PORT || dst >= MAX_SWIFT_PORT)) {
443 log_error("Malformed packet (src=%u, dst=%u)\n", shdr->src, shdr->dst);
447 skb_pull(skb, sizeof(struct swifthdr));
448 len -= sizeof(struct swifthdr);
452 log_debug("Received %u bytes from from port=%u to port=%u\n", len - sizeof(struct swifthdr), src, dst);
454 ssk = swift_lookup(dst);
456 log_error("Swift lookup failed for port %u\n", dst);
460 BUG_ON(addr_size > sizeof(skb->cb));
462 swift_addr = (struct sockaddr_swift *) skb->cb;
463 swift_addr->dests[0].port = shdr->src;
464 swift_addr->dests[0].addr = ip_hdr(skb)->saddr;
466 log_debug("Setting sin_port=%u, sin_addr=%u\n", ntohs(shdr->src), swift_addr->dests[0].addr);
468 err = ip_queue_rcv_skb((struct sock *) &ssk->sock, skb);
470 log_error("ip_queue_rcv_skb failed with %d\n", err);
473 return NET_RX_SUCCESS;
480 static struct proto swift_prot = {
481 .obj_size = sizeof(struct swift_sock),
482 .owner = THIS_MODULE,
486 static const struct proto_ops swift_ops = {
488 .owner = THIS_MODULE,
489 .release = swift_release,
491 .connect = swift_connect,
492 .socketpair = sock_no_socketpair,
493 .accept = sock_no_accept,
494 .getname = sock_no_getname,
495 .poll = datagram_poll,
496 .ioctl = sock_no_ioctl,
497 .listen = sock_no_listen,
498 .shutdown = sock_no_shutdown,
499 .setsockopt = sock_no_setsockopt,
500 .getsockopt = sock_no_getsockopt,
501 .sendmsg = swift_sendmsg,
502 .recvmsg = swift_recvmsg,
503 .mmap = sock_no_mmap,
504 .sendpage = sock_no_sendpage,
507 static const struct net_protocol swift_protocol = {
508 .handler = swift_rcv,
513 static struct inet_protosw swift_protosw = {
515 .protocol = IPPROTO_SWIFT,
521 static int __init swift_init(void)
525 rc = proto_register(&swift_prot, 1);
527 log_error("Error registering swift protocol\n");
531 rc = inet_add_protocol(&swift_protocol, IPPROTO_SWIFT);
533 log_error("Error adding swift protocol\n");
537 inet_register_protosw(&swift_protosw);
538 log_debug("Swift entered\n");
543 proto_unregister(&swift_prot);
549 static void __exit swift_exit(void)
551 inet_unregister_protosw(&swift_protosw);
553 inet_del_protocol(&swift_protocol, IPPROTO_SWIFT);
555 proto_unregister(&swift_prot);
557 log_debug("Swift exited\n");
560 module_init(swift_init);
561 module_exit(swift_exit);