X-Git-Url: http://dolda2000.com/gitweb/?a=blobdiff_plain;f=src%2Fmctap.c;h=9fce9acf6c5d78ba557f6d5a6072a2001cc53490;hb=1fefa40de9ff07accc7dcdde5e5f7a0a346328c3;hp=52dd294add5f65d9a74a553332a7bee7d9264d11;hpb=af3fc4a5b88a7014b43aa092c4e0e5f8fa962ed4;p=mctap.git diff --git a/src/mctap.c b/src/mctap.c index 52dd294..9fce9ac 100644 --- a/src/mctap.c +++ b/src/mctap.c @@ -6,15 +6,24 @@ #include #include #include +#include +#include +#include +#include +#include +#include +#include #include "utils.h" +static int quit = 0; + static void usage(FILE *out) { - fprintf(out, "usage: mctap [-h] MCASTGROUP PORT\n"); + fprintf(out, "usage: mctap [-hdp] [-P PIDFILE] [-D TAPNAME] MCASTGROUP PORT\n"); } -static char *formataddress(struct sockaddr *arg, socklen_t arglen) +static __attribute__ ((unused)) char *formataddress(struct sockaddr *arg, socklen_t arglen) { struct sockaddr_in *ipv4; struct sockaddr_in6 *ipv6; @@ -27,7 +36,7 @@ static char *formataddress(struct sockaddr *arg, socklen_t arglen) switch(arg->sa_family) { case AF_UNIX: - ret = strdup("Unix socket"); + ret = sstrdup("Unix socket"); break; case AF_INET: ipv4 = (struct sockaddr_in *)arg; @@ -74,33 +83,156 @@ static int mkmcastsk4(struct in_addr group, int port) return(fd); } -static void test(int fd) +static __attribute__ ((unused)) void test(int fd) { char buf[65536]; - int ret; - struct sockaddr_storage nm; - socklen_t nmlen; + int i, ret; + struct pollfd pfd; while(1) { - nmlen = sizeof(nm); - ret = recvfrom(fd, buf, sizeof(buf), 0, (struct sockaddr *)&nm, &nmlen); + pfd.fd = fd; + pfd.events = POLLIN; + ret = poll(&pfd, 1, -1); if(ret < 0) { - fprintf(stderr, "mctap: recvfrom: %s\n", strerror(ret)); + fprintf(stderr, "mctap: poll: %s\n", strerror(errno)); exit(1); } - printf("%s %i:\n", formataddress((struct sockaddr *)&nm, nmlen), ret); + 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(!quit) { + 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); +} + +static void sighand(int sig) +{ + switch(sig) { + case SIGINT: + case SIGTERM: + quit = 1; + break; + case SIGHUP: + break; } } int main(int argc, char **argv) { int c; - int sock; + int sock, tap; struct in_addr group; int port; + char *tapname; + char *pidfile; + int daemonize; + struct sockaddr_in dst; + FILE *pidfd; - while((c = getopt(argc, argv, "h")) >= 0) { + tapname = "mctap"; + daemonize = 0; + pidfile = NULL; + while((c = getopt(argc, argv, "hD:dpP:")) >= 0) { switch(c) { + case 'D': + tapname = optarg; + break; + case 'd': + daemonize = 1; + break; + case 'p': + pidfile = (void *)-1; + break; + case 'P': + pidfile = optarg; + break; case 'h': usage(stdout); return(0); @@ -109,6 +241,8 @@ int main(int argc, char **argv) exit(1); } } + if(pidfile == (void *)-1) + pidfile = sprintf2("/var/run/mctap.%s.pid", tapname); if(argc - optind < 2) { usage(stderr); exit(1); @@ -122,7 +256,36 @@ int main(int argc, char **argv) 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); + + pidfd = NULL; + if((pidfile != NULL) && ((pidfd = fopen(pidfile, "w")) == NULL)) { + fprintf(stderr, "mctap: could not create PID file %s: %s\n", pidfile, strerror(errno)); + exit(1); + } + if(daemonize) + daemon(0, 0); + if(pidfd != NULL) { + fprintf(pidfd, "%i\n", getpid()); + fclose(pidfd); + } + + signal(SIGTERM, sighand); + signal(SIGINT, sighand); + signal(SIGHUP, sighand); + + 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)); + syslog(LOG_INFO, "exiting"); - test(sock); + if(pidfile != NULL) + unlink(pidfile); return(0); }