uwsgi/socket.c

#include "uwsgi.h"

int bind_to_unix(char *socket_name, int listen_queue, int chmod_socket, int abstract_socket) {

	int serverfd;
	struct sockaddr_un *s_addr;

	fprintf(stderr, "binding on UNIX socket: %s\n", socket_name);

        // leave 1 byte for abstract namespace (108 linux -> 104 bsd/mac)
        if (strlen(socket_name) > 102) {
        	fprintf(stderr, "invalid socket name\n");
                	exit(1);
        }

        s_addr = malloc(sizeof(struct sockaddr_un));
        if (s_addr == NULL) {
        	perror("malloc()");
                exit(1);
        }

        memset(s_addr, 0, sizeof(struct sockaddr_un)) ;
        serverfd = socket(AF_UNIX, SOCK_STREAM, 0);
        if (serverfd < 0) {
        	perror("socket()");
                exit(1);
        }
        if (abstract_socket == 0) {
        	if (unlink(socket_name) != 0) {
                	perror("unlink()");
                }
        }

        if (abstract_socket ==1) {
        	fprintf(stderr, "setting abstract socket mode (warning: only Linux supports this)\n");
        }

	s_addr->sun_family = AF_UNIX;
	strcpy(s_addr->sun_path+abstract_socket, socket_name);

	if (bind(serverfd, (struct sockaddr *) s_addr, strlen(socket_name)+ abstract_socket + ( (void *)s_addr->sun_path - (void *)s_addr) ) != 0) {
		perror("bind()");
		exit(1);
	}

	if (listen(serverfd, listen_queue) != 0) {
        	perror("listen()");
                exit(1);
	}

	// chmod unix socket for lazy users
	if (chmod_socket == 1 && abstract_socket == 0) {
        	fprintf(stderr, "chmod() socket to 666 for lazy and brave users\n");
                if (chmod(socket_name, S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP|S_IROTH|S_IWOTH) != 0) {
                	perror("chmod()");
                }
	}

	return serverfd;
}

int bind_to_tcp(char *socket_name, int listen_queue, char *tcp_port) {

	int serverfd;
	struct sockaddr_in s_addr;
	int reuse = 1 ;
	
	tcp_port[0] = 0 ;
	memset(&s_addr, 0, sizeof(struct sockaddr_in));

	s_addr.sin_family = AF_INET;
	s_addr.sin_port = htons(atoi(tcp_port+1));


	if (socket_name[0] == 0) {
		s_addr.sin_addr.s_addr = INADDR_ANY;
	}
	else {
		s_addr.sin_addr.s_addr = inet_addr(socket_name);
	}

        serverfd = socket(AF_INET, SOCK_STREAM, 0);
        if (serverfd < 0) {
        	perror("socket()");
                exit(1);
        }

	if (setsockopt(serverfd, SOL_SOCKET, SO_REUSEADDR, (const void *)&reuse , sizeof(int)) < 0) {
		perror("setsockopt()");
		exit(1);
	}


	fprintf(stderr,"binding on TCP port: %d\n", ntohs(s_addr.sin_port));

	if (bind(serverfd, (struct sockaddr *) &s_addr, sizeof(s_addr) ) != 0) {
		perror("bind()");
		exit(1);
	}

	if (listen(serverfd, listen_queue) != 0) {
        	perror("listen()");
                exit(1);
	}

	return serverfd;
}