File:  [ELWIX - Embedded LightWeight unIX -] / libelwix / src / net.c
Revision 1.5: download - view: text, annotated - select for diffs - revision graph
Wed Jun 19 00:11:16 2013 UTC (10 years, 11 months ago) by misho
Branches: MAIN
CVS tags: elwix1_8, elwix1_7, HEAD, ELWIX1_7, ELWIX1_6
version 1.6

    1: /*************************************************************************
    2: * (C) 2013 AITNET ltd - Sofia/Bulgaria - <misho@aitnet.org>
    3: *  by Michael Pounov <misho@elwix.org>
    4: *
    5: * $Author: misho $
    6: * $Id: net.c,v 1.5 2013/06/19 00:11:16 misho Exp $
    7: *
    8: **************************************************************************
    9: The ELWIX and AITNET software is distributed under the following
   10: terms:
   11: 
   12: All of the documentation and software included in the ELWIX and AITNET
   13: Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
   14: 
   15: Copyright 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012, 2013
   16: 	by Michael Pounov <misho@elwix.org>.  All rights reserved.
   17: 
   18: Redistribution and use in source and binary forms, with or without
   19: modification, are permitted provided that the following conditions
   20: are met:
   21: 1. Redistributions of source code must retain the above copyright
   22:    notice, this list of conditions and the following disclaimer.
   23: 2. Redistributions in binary form must reproduce the above copyright
   24:    notice, this list of conditions and the following disclaimer in the
   25:    documentation and/or other materials provided with the distribution.
   26: 3. All advertising materials mentioning features or use of this software
   27:    must display the following acknowledgement:
   28: This product includes software developed by Michael Pounov <misho@elwix.org>
   29: ELWIX - Embedded LightWeight unIX and its contributors.
   30: 4. Neither the name of AITNET nor the names of its contributors
   31:    may be used to endorse or promote products derived from this software
   32:    without specific prior written permission.
   33: 
   34: THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND
   35: ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
   36: IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
   37: ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
   38: FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
   39: DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
   40: OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
   41: HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
   42: LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
   43: OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
   44: SUCH DAMAGE.
   45: */
   46: #include "global.h"
   47: 
   48: 
   49: /*
   50:  * e_ether_ntoa() - Convert ethernet address to string
   51:  *
   52:  * @n = ethernet address structure, like struct ether_addr
   53:  * @a = string
   54:  * @len = string length
   55:  * return: NULL error or !=NULL string a
   56:  */
   57: char *
   58: e_ether_ntoa(const struct e_ether_addr *n, char * __restrict a, int len)
   59: {
   60: 	if (!n || !a)
   61: 		return NULL;
   62: 
   63: 	memset(a, 0, len);
   64: 	if (snprintf(a, len, "%02x:%02x:%02x:%02x:%02x:%02x", 
   65: 			n->ether_addr_octet[0], n->ether_addr_octet[1], 
   66: 			n->ether_addr_octet[2], n->ether_addr_octet[3], 
   67: 			n->ether_addr_octet[4], n->ether_addr_octet[5]) < 17)
   68: 		return NULL;
   69: 
   70: 	return a;
   71: }
   72: 
   73: /*
   74:  * e_ether_aton() - Convert string to ethernet address
   75:  *
   76:  * @a = string
   77:  * @e = ethernet address structure, like struct ether_addr
   78:  * return: NULL error or !=NULL ethernet address structure
   79:  */
   80: struct e_ether_addr *
   81: e_ether_aton(const char *a, struct e_ether_addr *e)
   82: {                       
   83: 	int i;
   84: 	u_int o0, o1, o2, o3, o4, o5;
   85: 
   86: 	if (!a || !e)
   87: 		return NULL;
   88: 
   89: 	i = sscanf(a, "%x:%x:%x:%x:%x:%x", &o0, &o1, &o2, &o3, &o4, &o5);
   90: 	if (i != 6)
   91: 		return NULL;
   92: 
   93: 	e->ether_addr_octet[0] = o0;
   94: 	e->ether_addr_octet[1] = o1;
   95: 	e->ether_addr_octet[2] = o2;
   96: 	e->ether_addr_octet[3] = o3;
   97: 	e->ether_addr_octet[4] = o4;
   98: 	e->ether_addr_octet[5] = o5;
   99: 
  100: 	return e;
  101: }
  102: 
  103: /*
  104:  * e_n2port() - Extract port from network structure
  105:  *
  106:  * @addr = Address
  107:  * return: 0 not supported family type or port number
  108:  */
  109: u_short
  110: e_n2port(sockaddr_t * __restrict addr)
  111: {
  112: 	u_short port = 0;
  113: 
  114: 	if (!addr)
  115: 		return port;
  116: 
  117: 	switch (addr->sa.sa_family) {
  118: 		case AF_INET:
  119: 			return ntohs(addr->sin.sin_port);
  120: 		case AF_INET6:
  121: 			return ntohs(addr->sin6.sin6_port);
  122: 		default:
  123: 			break;
  124: 	}
  125: 
  126: 	return port;
  127: }
  128: 
  129: /*
  130:  * e_n2addr() - Extract address from network structure
  131:  *
  132:  * @addr = Address
  133:  * @val = Value for store string address
  134:  * return: NULL error or !=NULL string address from val
  135:  */
  136: const char *
  137: e_n2addr(sockaddr_t * __restrict addr, ait_val_t * __restrict val)
  138: {
  139: 	char *s, str[INET6_ADDRSTRLEN] = { 0 };
  140: 	const char *ret = NULL;
  141: 
  142: 	if (!addr || !val)
  143: 		return ret;
  144: 
  145: 	AIT_INIT_VAL(val);
  146: 	switch (addr->sa.sa_family) {
  147: 		case AF_INET:
  148: 			if (!inet_ntop(AF_INET, &addr->sin.sin_addr, str, INET_ADDRSTRLEN)) {
  149: 				LOGERR;
  150: 				return ret;
  151: 			} else
  152: 				ret = str;
  153: 			break;
  154: 		case AF_INET6:
  155: 			if (!inet_ntop(AF_INET6, &addr->sin6.sin6_addr, str, INET6_ADDRSTRLEN)) {
  156: 				LOGERR;
  157: 				return ret;
  158: 			} else
  159: 				ret = str;
  160: 			break;
  161: 		case AF_LOCAL:
  162: 			ret = addr->sun.sun_path;
  163: 			break;
  164: 		case AF_LINK:
  165: 			if (!(s = link_ntoa(&addr->sdl))) {
  166: 				LOGERR;
  167: 				return ret;
  168: 			} else
  169: 				ret = s;
  170: 			break;
  171: 		default:
  172: 			elwix_SetErr(EPROTONOSUPPORT, "Unsuported address family %d", 
  173: 					addr->sa.sa_family);
  174: 			return ret;
  175: 	}
  176: 
  177: 	AIT_SET_STR(val, ret);
  178: 	return (const char*) AIT_GET_STR(val);
  179: }
  180: 
  181: /*
  182:  * e_gethostbyname() - Get host and port and make network structure
  183:  *
  184:  * @psHost = Hostname
  185:  * @port = Port
  186:  * @addr = Network address structure
  187:  * return: NULL error or !=NULL network structure
  188:  */
  189: sockaddr_t *
  190: e_gethostbyname(const char *psHost, u_short port, sockaddr_t * __restrict addr)
  191: {
  192: 	struct hostent *host = NULL;
  193: 
  194: 	if (!psHost || !addr)
  195: 		return NULL;
  196: 
  197: 	if (*psHost != '/') {
  198: 		/* resolver */
  199: 		if (!addr->sa.sa_family)
  200: 			host = gethostbyname(psHost);
  201: 		else
  202: 			host = gethostbyname2(psHost, addr->sa.sa_family);
  203: 		if (!host) {
  204: 			elwix_SetErr(EINVAL, "Resolver #%d - %s", h_errno, hstrerror(h_errno));
  205: 			return NULL;
  206: 		} else {
  207: 			memset(addr, 0, sizeof(sockaddr_t));
  208: 			addr->sa.sa_family = host->h_addrtype;
  209: 		}
  210: 	} else {
  211: 		memset(addr, 0, sizeof(sockaddr_t));
  212: 		addr->sa.sa_family = AF_LOCAL;
  213: 	}
  214: 		
  215: 
  216: 	switch (addr->sa.sa_family) {
  217: 		case AF_INET:
  218: 			addr->sin.sin_len = sizeof(struct sockaddr_in);
  219: 			addr->sin.sin_family = AF_INET;
  220: 			addr->sin.sin_port = htons(port);
  221: 			memcpy(&addr->sin.sin_addr, host->h_addr, sizeof addr->sin.sin_addr);
  222: 			return addr;
  223: 		case AF_INET6:
  224: 			addr->sin6.sin6_len = sizeof(struct sockaddr_in6);
  225: 			addr->sin6.sin6_family = AF_INET6;
  226: 			addr->sin6.sin6_port = htons(port);
  227: 			memcpy(&addr->sin6.sin6_addr, host->h_addr, sizeof addr->sin6.sin6_addr);
  228: 			return addr;
  229: 		case AF_LOCAL:
  230: 			addr->sun.sun_len = sizeof(struct sockaddr_un);
  231: 			addr->sun.sun_family = AF_LOCAL;
  232: 			memset(addr->sun.sun_path, 0, sizeof addr->sun.sun_path);
  233: 			snprintf(addr->sun.sun_path, sizeof addr->sun.sun_path, "%s-%hu", psHost, port);
  234: 			return addr;
  235: 		default:
  236: 			elwix_SetErr(EPROTONOSUPPORT, "Unsuported address family %d", addr->sa.sa_family);
  237: 			break;
  238: 	}
  239: 
  240: 	return NULL;
  241: }
  242: 
  243: /*
  244:  * e_addrcmp() - Compare network addresses
  245:  *
  246:  * @a = 1st address
  247:  * @b = 2nd address
  248:  * @p = compare and ports, if family is AF_INET or AF_INET6
  249:  * return: 0 is equal or !=0 is different
  250:  */
  251: int
  252: e_addrcmp(sockaddr_t * __restrict a, sockaddr_t * __restrict b, int p)
  253: {
  254: 	if (a && b && a->sa.sa_family == b->sa.sa_family)
  255: 		switch (a->sa.sa_family) {
  256: 			case AF_LOCAL:
  257: 				return strcmp(a->sun.sun_path, b->sun.sun_path);
  258: 			case AF_INET:
  259: 				if (p && (a->sin.sin_port - b->sin.sin_port))
  260: 					return (int) !!(a->sin.sin_port - b->sin.sin_port);
  261: 				else
  262: 					return memcmp(&a->sin.sin_addr, &b->sin.sin_addr, 
  263: 							sizeof a->sin.sin_addr);
  264: 			case AF_INET6:
  265: 				if (p && (a->sin6.sin6_port - b->sin6.sin6_port))
  266: 					return (int) !!(a->sin6.sin6_port - b->sin6.sin6_port);
  267: 				else
  268: 					return memcmp(&a->sin6.sin6_addr, &b->sin6.sin6_addr, 
  269: 							sizeof a->sin6.sin6_addr);
  270: 			case AF_LINK:
  271: 				return memcmp(&a->sdl.sdl_data, &b->sdl.sdl_data, 
  272: 						sizeof a->sdl.sdl_data);
  273: 		}
  274: 
  275: 	return (int) !!(a - b);
  276: }
  277: 
  278: /*
  279:  * e_usleep() - usleep() replacement for ELWIX
  280:  *
  281:  * @usec = microseconds for sleep
  282:  * return: -1 interrupted by signal or 0 ok
  283:  */
  284: int
  285: e_usleep(u_int usec)
  286: {
  287: 	struct timeval tv = { (time_t) (usec / 1000000), (long) (usec % 1000000) };
  288: 
  289: 	return select(0, NULL, NULL, NULL, &tv);
  290: }
  291: 
  292: /*
  293:  * e_innet() - Test address match in network
  294:  *
  295:  * @net = network
  296:  * @addr = address
  297:  * return: -1 error, 0 match or 1 not match
  298:  */
  299: int
  300: e_innet(netaddr_t * __restrict net, inaddr_t * __restrict addr)
  301: {
  302: 	register int i;
  303: 	int ret = 0;
  304: 
  305: 	if (!net || !addr)
  306: 		return -1;
  307: 
  308: 	switch (net->addr.sa.sa_family) {
  309: 		case AF_INET:
  310: 			for (i = 0; i < sizeof(struct in_addr); i++) {
  311: 				ret = ((caddr_t) &net->addr.sin.sin_addr.s_addr)[i] & 
  312: 					net->mask.in.s4_addr[i];
  313: 				ret -= addr->in.s4_addr[i] & net->mask.in.s4_addr[i];
  314: 				if (ret)
  315: 					break;
  316: 			}
  317: 			break;
  318: 		case AF_INET6:
  319: 			for (i = 0; i < sizeof(struct in6_addr); i++) {
  320: 				ret = net->addr.sin6.sin6_addr.s6_addr[i] & 
  321: 					net->mask.in6.s6_addr[i];
  322: 				ret -= addr->in6.s6_addr[i] & net->mask.in6.s6_addr[i];
  323: 				if (ret)
  324: 					break;
  325: 			}
  326: 			break;
  327: 		default:
  328: 			return -1;
  329: 	}
  330: 
  331: 	return !!ret;
  332: }
  333: 
  334: /*
  335:  * e_getnet() - Get network from string
  336:  *
  337:  * @net = Network string (format: <net[/cidr]>)
  338:  * return: NULL error or !=NULL network should be e_free()
  339:  */
  340: netaddr_t *
  341: e_getnet(const char *net)
  342: {
  343: 	netaddr_t *n;
  344: 	char *str, *wrk;
  345: 	struct hostent *host;
  346: 
  347: 	n = e_malloc(sizeof(netaddr_t));
  348: 	if (!n) {
  349: 		LOGERR;
  350: 		return NULL;
  351: 	} else
  352: 		memset(n, 0, sizeof(netaddr_t));
  353: 	str = e_strdup(net);
  354: 	if (!str) {
  355: 		LOGERR;
  356: 		e_free(n);
  357: 		return NULL;
  358: 	}
  359: 	wrk = strchr(str, '/');
  360: 	if (wrk)
  361: 		*wrk++ = 0;
  362: 
  363: 	host = gethostbyname2(str, strchr(str, ':') ? AF_INET6 : AF_INET);
  364: 	if (!host) {
  365: 		elwix_SetErr(EINVAL, "Resolver #%d - %s", h_errno, hstrerror(h_errno));
  366: 		e_free(str);
  367: 		e_free(n);
  368: 		return NULL;
  369: 	}
  370: 	switch (host->h_addrtype) {
  371: 		case AF_INET:
  372: 			n->addr.sin.sin_len = sizeof(struct sockaddr_in);
  373: 			n->addr.sin.sin_family = host->h_addrtype;
  374: 			memcpy(&n->addr.sin.sin_addr, host->h_addr, sizeof n->addr.sin.sin_addr);
  375: 			if (wrk)
  376: 				n->mask.in.s_addr = E_CIDRMASK(strtol(wrk, NULL, 10));
  377: 			else
  378: 				n->mask.in.s_addr = 0xFFFFFFFF;
  379: 			break;
  380: 		case AF_INET6:
  381: 			n->addr.sin6.sin6_len = sizeof(struct sockaddr_in6);
  382: 			n->addr.sin6.sin6_family = host->h_addrtype;
  383: 			memcpy(&n->addr.sin6.sin6_addr, host->h_addr, sizeof n->addr.sin6.sin6_addr);
  384: 			/* TODO: should support ipv6 mask */
  385: 			break;
  386: 		default:
  387: 			elwix_SetErr(EINVAL, "Unsupported family #%d", host->h_addrtype);
  388: 			e_free(str);
  389: 			e_free(n);
  390: 			return NULL;
  391: 	}
  392: 
  393: 	e_free(str);
  394: 	return n;
  395: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>