Ticket #252: main.c

#include <stdio.h>
#include <netinet/in.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <string.h>
#include <sys/socket.h>
#include <string.h>
#include <unistd.h>

#define PORT 51413

/* returns:
 * 0 if both sockets bound (need to set TR_NEED_V4SOCKET)
 * 1 if ipv6 socket didn't bind (die and tell the user to ensure the port
 *     isn't in use)
 * 2 if only ipv6 socket bound (do not need to set TR_NEED_V4SOCKET)
 */

int main(int argc, const char** argv) {
    struct sockaddr_in sock4;
    struct sockaddr_in6 sock6;
    int s;
    int s2;
    memset(&sock6, 0, sizeof(sock6));
    sock6.sin6_family = AF_INET6;
    sock6.sin6_port = htons(PORT);
    memcpy(&sock6.sin6_addr, &in6addr_any, sizeof(struct in6_addr));
    s = socket(AF_INET6, SOCK_STREAM, 0);
    if(bind(s, (struct sockaddr*)&sock6, sizeof(sock6))) {
        close(s);
        return 1;
    }
    memset(&sock4, 0, sizeof(sock4));
    sock4.sin_family = AF_INET;
    sock4.sin_addr.s_addr = INADDR_ANY;
    sock4.sin_port = htons(PORT);
    s2 = socket(AF_INET, SOCK_STREAM, 0);
    if(bind(s, (struct sockaddr*)&sock4, sizeof(sock4))) {
        close(s);
        close(s2);
        return 2;
    }
    close(s);
    close(s2);
    return 0;
}