X-Git-Url: http://dolda2000.com/gitweb/?a=blobdiff_plain;f=src%2Fmctap.c;h=842a9d6d633f0110cffb3650ca0f70586fba14c6;hb=a13b233b0ac23495df52fa80dddd02f24ce71491;hp=061edd558a7a3b39ac0b41b0fe2b189edfca1070;hpb=3415540feb47f1c379767c984620f8a7a3c4e2b1;p=mctap.git diff --git a/src/mctap.c b/src/mctap.c index 061edd5..842a9d6 100644 --- a/src/mctap.c +++ b/src/mctap.c @@ -4,18 +4,211 @@ #include #include #include +#include +#include +#include +#include +#include +#include +#include +#include + +#include "utils.h" static void usage(FILE *out) { - fprintf(out, "usage: mctap [-h]\n"); + fprintf(out, "usage: mctap [-hd] [-D TAPNAME] MCASTGROUP PORT\n"); +} + +static __attribute__ ((unused)) char *formataddress(struct sockaddr *arg, socklen_t arglen) +{ + struct sockaddr_in *ipv4; + struct sockaddr_in6 *ipv6; + static char *ret = NULL; + char buf[1024]; + + if(ret != NULL) + free(ret); + ret = NULL; + switch(arg->sa_family) + { + case AF_UNIX: + ret = sstrdup("Unix socket"); + break; + case AF_INET: + ipv4 = (struct sockaddr_in *)arg; + if(inet_ntop(AF_INET, &ipv4->sin_addr, buf, sizeof(buf)) == NULL) + return(NULL); + ret = sprintf2("%s:%i", buf, (int)ntohs(ipv4->sin_port)); + break; + case AF_INET6: + ipv6 = (struct sockaddr_in6 *)arg; + if(inet_ntop(AF_INET6, &ipv6->sin6_addr, buf, sizeof(buf)) == NULL) + return(NULL); + ret = sprintf2("[%s]:%i", buf, (int)ntohs(ipv6->sin6_port)); + break; + default: + errno = EPFNOSUPPORT; + break; + } + return(ret); +} + +static int mkmcastsk4(struct in_addr group, int port) +{ + int fd; + int soval; + struct sockaddr_in nm; + struct ip_mreqn mreq; + + fd = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP); + soval = 1; + if(setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &soval, sizeof(soval))) + return(-1); + memset(&nm, 0, sizeof(nm)); + nm.sin_family = AF_INET; + nm.sin_port = htons(port); + if(bind(fd, (struct sockaddr *)&nm, sizeof(nm))) + return(-1); + memset(&mreq, 0, sizeof(mreq)); + mreq.imr_multiaddr = group; + if(setsockopt(fd, SOL_IP, IP_ADD_MEMBERSHIP, &mreq, sizeof(mreq))) + return(-1); + soval = 1; + if(setsockopt(fd, SOL_IP, IP_MULTICAST_LOOP, &soval, sizeof(soval))) + return(-1); + return(fd); +} + +static __attribute__ ((unused)) void test(int fd) +{ + char buf[65536]; + int i, ret; + struct pollfd pfd; + + while(1) { + pfd.fd = fd; + pfd.events = POLLIN; + ret = poll(&pfd, 1, -1); + if(ret < 0) { + fprintf(stderr, "mctap: poll: %s\n", strerror(errno)); + exit(1); + } + if(pfd.revents) { + ret = read(fd, buf, sizeof(buf)); + if(ret < 0) { + fprintf(stderr, "mctap: read: %s\n", strerror(errno)); + exit(1); + } + for(i = 0; i < ret; i++) { + printf("%02x ", (unsigned char)buf[i]); + if(i % 20 == 19) + putchar(10); + } + putchar(10); + } + } +} + +static void bridge(int sock, int tap, struct sockaddr *dst, socklen_t dstlen) +{ + char buf[65536]; + int ret; + struct pollfd pfds[2]; + struct tun_pi pi; + + fcntl(sock, F_SETFL, fcntl(sock, F_GETFL) | O_NONBLOCK); + fcntl(tap, F_SETFL, fcntl(tap, F_GETFL) | O_NONBLOCK); + while(1) { + pfds[0].fd = sock; + pfds[0].events = POLLIN; + pfds[1].fd = tap; + pfds[1].events = POLLIN; + ret = poll(pfds, 2, -1); + if(ret < 0) { + if(errno != EINTR) { + syslog(LOG_ERR, "mctap: poll: %s", strerror(errno)); + exit(1); + } + continue; + } + if(pfds[0].revents) { + ret = read(sock, buf, sizeof(buf)); + if(ret < 0) { + if((errno != EINTR) && (errno != EAGAIN)) { + syslog(LOG_ERR, "mctap: mcast packet: %s", strerror(errno)); + exit(1); + } + } else { + if(sizeof(buf) - ret < sizeof(pi)) { + /* Drop */ + } else { + memmove(buf + sizeof(pi), buf, ret); + pi.flags = 0; + pi.proto = 0; + memcpy(buf, &pi, sizeof(pi)); + write(tap, buf, sizeof(pi) + ret); + } + } + } + if(pfds[1].revents) { + ret = read(tap, buf, sizeof(buf)); + if(ret < 0) { + if((errno != EINTR) && (errno != EAGAIN)) { + syslog(LOG_ERR, "mctap: mcast packet: %s", strerror(errno)); + exit(1); + } + } else { + if(ret < sizeof(pi)) { + /* Drop */ + } else { + memcpy(&pi, buf, sizeof(pi)); + if(pi.flags & TUN_PKT_STRIP) { + /* Drop */ + } else { + sendto(sock, buf + sizeof(pi), ret - sizeof(pi), 0, dst, dstlen); + } + } + } + } + } +} + +static int maketap(char *name) +{ + int fd; + struct ifreq rb; + + if((fd = open("/dev/net/tun", O_RDWR)) < 0) + return(-1); + memset(&rb, 0, sizeof(rb)); + rb.ifr_flags = IFF_TAP; + strncpy(rb.ifr_name, name, IFNAMSIZ); + if(ioctl(fd, TUNSETIFF, &rb)) + return(-1); + return(fd); } int main(int argc, char **argv) { int c; + int sock, tap; + struct in_addr group; + int port; + char *tapname; + int daemonize; + struct sockaddr_in dst; - while((c = getopt(argc, argv, "-h")) >= 0) { + tapname = "mctap"; + daemonize = 0; + while((c = getopt(argc, argv, "hD:d")) >= 0) { switch(c) { + case 'D': + tapname = optarg; + break; + case 'd': + daemonize = 1; + break; case 'h': usage(stdout); return(0); @@ -24,6 +217,32 @@ int main(int argc, char **argv) exit(1); } } + if(argc - optind < 2) { + usage(stderr); + exit(1); + } + if(!inet_aton(argv[optind], &group)) { + fprintf(stderr, "mctap: invalid group address: %s\n", argv[optind]); + exit(1); + } + port = atoi(argv[optind + 1]); + if((sock = mkmcastsk4(group, port)) < 0) { + fprintf(stderr, "mctap: could not create multicast socket: %s\n", strerror(errno)); + exit(1); + } + if((tap = maketap(tapname)) < 0) { + fprintf(stderr, "mctap: could not create TAP device: %s\n", strerror(errno)); + exit(1); + } + openlog(sprintf2("mctap-%s", tapname), LOG_PID, LOG_DAEMON); + + if(daemonize) + daemon(0, 0); + memset(&dst, 0, sizeof(dst)); + dst.sin_family = AF_INET; + dst.sin_addr = group; + dst.sin_port = htons(port); + bridge(sock, tap, (struct sockaddr *)&dst, sizeof(dst)); return(0); }