Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 1 | /* |
| 2 | * user-mode-linux networking multicast transport |
| 3 | * Copyright (C) 2001 by Harald Welte <laforge@gnumonks.org> |
| 4 | * |
| 5 | * based on the existing uml-networking code, which is |
| 6 | * Copyright (C) 2001 Lennert Buytenhek (buytenh@gnu.org) and |
| 7 | * James Leu (jleu@mindspring.net). |
| 8 | * Copyright (C) 2001 by various other people who didn't put their name here. |
| 9 | * |
| 10 | * Licensed under the GPL. |
| 11 | * |
| 12 | */ |
| 13 | |
| 14 | #include <errno.h> |
| 15 | #include <unistd.h> |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 16 | #include <sys/socket.h> |
| 17 | #include <sys/un.h> |
| 18 | #include <sys/time.h> |
| 19 | #include <netinet/in.h> |
| 20 | #include "net_user.h" |
| 21 | #include "mcast.h" |
| 22 | #include "kern_util.h" |
| 23 | #include "user_util.h" |
| 24 | #include "user.h" |
| 25 | #include "os.h" |
Paolo 'Blaisorblade' Giarrusso | c13e569 | 2006-10-19 23:28:20 -0700 | [diff] [blame] | 26 | #include "um_malloc.h" |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 27 | |
| 28 | #define MAX_PACKET (ETH_MAX_PACKET + ETH_HEADER_OTHER) |
| 29 | |
| 30 | static struct sockaddr_in *new_addr(char *addr, unsigned short port) |
| 31 | { |
| 32 | struct sockaddr_in *sin; |
| 33 | |
| 34 | sin = um_kmalloc(sizeof(struct sockaddr_in)); |
| 35 | if(sin == NULL){ |
| 36 | printk("new_addr: allocation of sockaddr_in failed\n"); |
Jeff Dike | 56bd194 | 2007-05-06 14:51:02 -0700 | [diff] [blame] | 37 | return NULL; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 38 | } |
| 39 | sin->sin_family = AF_INET; |
| 40 | sin->sin_addr.s_addr = in_aton(addr); |
Jeff Dike | 7c00c31 | 2005-05-20 13:59:09 -0700 | [diff] [blame] | 41 | sin->sin_port = htons(port); |
Jeff Dike | 56bd194 | 2007-05-06 14:51:02 -0700 | [diff] [blame] | 42 | return sin; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 43 | } |
| 44 | |
Jeff Dike | f34d9d2 | 2007-05-06 14:51:04 -0700 | [diff] [blame^] | 45 | static int mcast_user_init(void *data, void *dev) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 46 | { |
| 47 | struct mcast_data *pri = data; |
| 48 | |
| 49 | pri->mcast_addr = new_addr(pri->addr, pri->port); |
| 50 | pri->dev = dev; |
Jeff Dike | f34d9d2 | 2007-05-06 14:51:04 -0700 | [diff] [blame^] | 51 | return 0; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 52 | } |
| 53 | |
Paolo 'Blaisorblade' Giarrusso | 83f4e8a | 2007-03-07 20:41:09 -0800 | [diff] [blame] | 54 | static void mcast_remove(void *data) |
| 55 | { |
| 56 | struct mcast_data *pri = data; |
| 57 | |
| 58 | kfree(pri->mcast_addr); |
| 59 | pri->mcast_addr = NULL; |
| 60 | } |
| 61 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 62 | static int mcast_open(void *data) |
| 63 | { |
| 64 | struct mcast_data *pri = data; |
| 65 | struct sockaddr_in *sin = pri->mcast_addr; |
| 66 | struct ip_mreq mreq; |
Paolo 'Blaisorblade' Giarrusso | c50d2c4 | 2005-11-13 16:07:07 -0800 | [diff] [blame] | 67 | int fd, yes = 1, err = -EINVAL; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 68 | |
| 69 | |
Jeff Dike | 7c00c31 | 2005-05-20 13:59:09 -0700 | [diff] [blame] | 70 | if ((sin->sin_addr.s_addr == 0) || (sin->sin_port == 0)) |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 71 | goto out; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 72 | |
| 73 | fd = socket(AF_INET, SOCK_DGRAM, 0); |
Jeff Dike | 7c00c31 | 2005-05-20 13:59:09 -0700 | [diff] [blame] | 74 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 75 | if (fd < 0){ |
Paolo 'Blaisorblade' Giarrusso | c50d2c4 | 2005-11-13 16:07:07 -0800 | [diff] [blame] | 76 | err = -errno; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 77 | printk("mcast_open : data socket failed, errno = %d\n", |
| 78 | errno); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 79 | goto out; |
| 80 | } |
| 81 | |
| 82 | if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(yes)) < 0) { |
Paolo 'Blaisorblade' Giarrusso | c50d2c4 | 2005-11-13 16:07:07 -0800 | [diff] [blame] | 83 | err = -errno; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 84 | printk("mcast_open: SO_REUSEADDR failed, errno = %d\n", |
| 85 | errno); |
Jeff Dike | 7c00c31 | 2005-05-20 13:59:09 -0700 | [diff] [blame] | 86 | goto out_close; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 87 | } |
| 88 | |
| 89 | /* set ttl according to config */ |
| 90 | if (setsockopt(fd, SOL_IP, IP_MULTICAST_TTL, &pri->ttl, |
| 91 | sizeof(pri->ttl)) < 0) { |
Paolo 'Blaisorblade' Giarrusso | c50d2c4 | 2005-11-13 16:07:07 -0800 | [diff] [blame] | 92 | err = -errno; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 93 | printk("mcast_open: IP_MULTICAST_TTL failed, error = %d\n", |
| 94 | errno); |
Jeff Dike | 7c00c31 | 2005-05-20 13:59:09 -0700 | [diff] [blame] | 95 | goto out_close; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 96 | } |
| 97 | |
| 98 | /* set LOOP, so data does get fed back to local sockets */ |
| 99 | if (setsockopt(fd, SOL_IP, IP_MULTICAST_LOOP, &yes, sizeof(yes)) < 0) { |
Paolo 'Blaisorblade' Giarrusso | c50d2c4 | 2005-11-13 16:07:07 -0800 | [diff] [blame] | 100 | err = -errno; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 101 | printk("mcast_open: IP_MULTICAST_LOOP failed, error = %d\n", |
| 102 | errno); |
Jeff Dike | 7c00c31 | 2005-05-20 13:59:09 -0700 | [diff] [blame] | 103 | goto out_close; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 104 | } |
| 105 | |
| 106 | /* bind socket to mcast address */ |
| 107 | if (bind(fd, (struct sockaddr *) sin, sizeof(*sin)) < 0) { |
Jeff Dike | b4fd310 | 2005-09-16 19:27:49 -0700 | [diff] [blame] | 108 | err = -errno; |
Paolo 'Blaisorblade' Giarrusso | c50d2c4 | 2005-11-13 16:07:07 -0800 | [diff] [blame] | 109 | printk("mcast_open : data bind failed, errno = %d\n", errno); |
Jeff Dike | 7c00c31 | 2005-05-20 13:59:09 -0700 | [diff] [blame] | 110 | goto out_close; |
Jeff Dike | 56bd194 | 2007-05-06 14:51:02 -0700 | [diff] [blame] | 111 | } |
| 112 | |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 113 | /* subscribe to the multicast group */ |
| 114 | mreq.imr_multiaddr.s_addr = sin->sin_addr.s_addr; |
| 115 | mreq.imr_interface.s_addr = 0; |
| 116 | if (setsockopt(fd, SOL_IP, IP_ADD_MEMBERSHIP, |
| 117 | &mreq, sizeof(mreq)) < 0) { |
Paolo 'Blaisorblade' Giarrusso | c50d2c4 | 2005-11-13 16:07:07 -0800 | [diff] [blame] | 118 | err = -errno; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 119 | printk("mcast_open: IP_ADD_MEMBERSHIP failed, error = %d\n", |
| 120 | errno); |
| 121 | printk("There appears not to be a multicast-capable network " |
| 122 | "interface on the host.\n"); |
| 123 | printk("eth0 should be configured in order to use the " |
| 124 | "multicast transport.\n"); |
Paolo 'Blaisorblade' Giarrusso | c50d2c4 | 2005-11-13 16:07:07 -0800 | [diff] [blame] | 125 | goto out_close; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 126 | } |
| 127 | |
Jeff Dike | 7c00c31 | 2005-05-20 13:59:09 -0700 | [diff] [blame] | 128 | return fd; |
| 129 | |
| 130 | out_close: |
Paolo 'Blaisorblade' Giarrusso | c50d2c4 | 2005-11-13 16:07:07 -0800 | [diff] [blame] | 131 | os_close_file(fd); |
Jeff Dike | b4fd310 | 2005-09-16 19:27:49 -0700 | [diff] [blame] | 132 | out: |
Paolo 'Blaisorblade' Giarrusso | c50d2c4 | 2005-11-13 16:07:07 -0800 | [diff] [blame] | 133 | return err; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 134 | } |
| 135 | |
| 136 | static void mcast_close(int fd, void *data) |
| 137 | { |
| 138 | struct ip_mreq mreq; |
| 139 | struct mcast_data *pri = data; |
| 140 | struct sockaddr_in *sin = pri->mcast_addr; |
| 141 | |
| 142 | mreq.imr_multiaddr.s_addr = sin->sin_addr.s_addr; |
| 143 | mreq.imr_interface.s_addr = 0; |
| 144 | if (setsockopt(fd, SOL_IP, IP_DROP_MEMBERSHIP, |
| 145 | &mreq, sizeof(mreq)) < 0) { |
| 146 | printk("mcast_open: IP_DROP_MEMBERSHIP failed, error = %d\n", |
| 147 | errno); |
| 148 | } |
| 149 | |
| 150 | os_close_file(fd); |
| 151 | } |
| 152 | |
| 153 | int mcast_user_write(int fd, void *buf, int len, struct mcast_data *pri) |
| 154 | { |
| 155 | struct sockaddr_in *data_addr = pri->mcast_addr; |
| 156 | |
Jeff Dike | 56bd194 | 2007-05-06 14:51:02 -0700 | [diff] [blame] | 157 | return net_sendto(fd, buf, len, data_addr, sizeof(*data_addr)); |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 158 | } |
| 159 | |
| 160 | static int mcast_set_mtu(int mtu, void *data) |
| 161 | { |
Jeff Dike | 56bd194 | 2007-05-06 14:51:02 -0700 | [diff] [blame] | 162 | return mtu; |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 163 | } |
| 164 | |
Jeff Dike | 5e7672e | 2006-09-27 01:50:33 -0700 | [diff] [blame] | 165 | const struct net_user_info mcast_user_info = { |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 166 | .init = mcast_user_init, |
| 167 | .open = mcast_open, |
| 168 | .close = mcast_close, |
Paolo 'Blaisorblade' Giarrusso | 83f4e8a | 2007-03-07 20:41:09 -0800 | [diff] [blame] | 169 | .remove = mcast_remove, |
Linus Torvalds | 1da177e | 2005-04-16 15:20:36 -0700 | [diff] [blame] | 170 | .set_mtu = mcast_set_mtu, |
| 171 | .add_address = NULL, |
| 172 | .delete_address = NULL, |
| 173 | .max_packet = MAX_PACKET - ETH_HEADER_OTHER |
| 174 | }; |