src/bsp/lk/tools/network.c

/*
 * Copyright (c) 2014 Brian Swetland
 *
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files
 * (the "Software"), to deal in the Software without restriction,
 * including without limitation the rights to use, copy, modify, merge,
 * publish, distribute, sublicense, and/or sell copies of the Software,
 * and to permit persons to whom the Software is furnished to do so,
 * subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
 * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "network.h"

in_addr_t lookup_hostname(const char *hostname) {
	int err;
	struct addrinfo *info, *temp;
	in_addr_t addr = 0;

	struct addrinfo hints;
	memset(&hints, 0, sizeof(hints));
	hints.ai_family = AF_INET;
	hints.ai_socktype = SOCK_STREAM;

	err = getaddrinfo(hostname, NULL, &hints, &info);
	if (err < 0) {
		printf("getaddrinfo() returns %d, error '%s'\n", err, gai_strerror(err));
		return 0;
	}

	for (temp = info; temp; temp = temp->ai_next) {
//		printf("flags 0x%x, family %d, socktype %d, protocol %d, addrlen %d\n",
//			temp->ai_flags, temp->ai_family, temp->ai_socktype, temp->ai_protocol, temp->ai_addrlen);

		if (temp->ai_family == AF_INET && temp->ai_protocol == IPPROTO_TCP) {
			struct sockaddr_in *sa = (struct sockaddr_in *)temp->ai_addr;
//			printf("port %d, addr 0x%x\n", sa->sin_port, sa->sin_addr.s_addr);
			addr = sa->sin_addr.s_addr;
		}
	}

	freeaddrinfo(info);

	return addr;
}

static int inet_listen(in_addr_t addr, int type, unsigned port, int shared) {
	struct sockaddr_in sa;
	int s;
	if ((s = socket(AF_INET, type, 0)) < 0) {
		return -1;
	}
	if (shared) {
		int opt = 1;
    		setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
    		//setsockopt(s, SOL_SOCKET, SO_REUSEPORT, &opt, sizeof(opt));
	}
	memset(&sa, 0, sizeof(sa));
	sa.sin_family = AF_INET;
	sa.sin_port = htons(port);
	sa.sin_addr.s_addr = addr;
	if (bind(s, (struct sockaddr *) &sa, sizeof(sa)) < 0) {
		close(s);
		return -1;
	}
	return s;
}

static int inet_connect(in_addr_t addr, int type, unsigned port) {
	struct sockaddr_in sa;
	int s;
	if ((s = socket(AF_INET, type, 0)) < 0) {
		return -1;
	}
	if (addr == 0xFFFFFFFF) {
		int opt = 1;
		setsockopt(s, SOL_SOCKET, SO_BROADCAST, &opt, sizeof(opt));
	}
	memset(&sa, 0, sizeof(sa));
	sa.sin_family = AF_INET;
	sa.sin_port = htons(port);
	sa.sin_addr.s_addr = addr;
	if (connect(s, (struct sockaddr *) &sa, sizeof(sa)) < 0) {
		close(s);
		return -1;
	}
	return s;
}

int udp_listen(in_addr_t addr, unsigned port, int shared) {
	return inet_listen(addr, SOCK_DGRAM, port, shared);
}

int udp_connect(in_addr_t addr, unsigned port) {
	return inet_connect(addr, SOCK_DGRAM, port);
}

int tcp_listen(in_addr_t addr, unsigned port) {
	return inet_listen(addr, SOCK_STREAM, port, 0);
}

int tcp_connect(in_addr_t addr, unsigned port) {
	return inet_connect(addr, SOCK_STREAM, port);
}

// vim: noexpandtab