BAddr.h 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757
  1. /**
  2. * @file BAddr.h
  3. * @author Ambroz Bizjak <ambrop7@gmail.com>
  4. *
  5. * @section LICENSE
  6. *
  7. * Redistribution and use in source and binary forms, with or without
  8. * modification, are permitted provided that the following conditions are met:
  9. * 1. Redistributions of source code must retain the above copyright
  10. * notice, this list of conditions and the following disclaimer.
  11. * 2. Redistributions in binary form must reproduce the above copyright
  12. * notice, this list of conditions and the following disclaimer in the
  13. * documentation and/or other materials provided with the distribution.
  14. * 3. Neither the name of the author nor the
  15. * names of its contributors may be used to endorse or promote products
  16. * derived from this software without specific prior written permission.
  17. *
  18. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
  19. * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  20. * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  21. * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
  22. * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  23. * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  24. * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  25. * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26. * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  27. * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28. *
  29. * @section DESCRIPTION
  30. *
  31. * Network address abstractions.
  32. */
  33. #ifndef BADVPN_SYSTEM_BADDR_H
  34. #define BADVPN_SYSTEM_BADDR_H
  35. #include <stdint.h>
  36. #include <limits.h>
  37. #include <stdlib.h>
  38. #include <string.h>
  39. #include <stdio.h>
  40. #ifdef BADVPN_USE_WINAPI
  41. #include <ws2tcpip.h>
  42. #else
  43. #include <sys/types.h>
  44. #include <sys/socket.h>
  45. #include <netdb.h>
  46. #include <netinet/in.h>
  47. #endif
  48. #include <misc/byteorder.h>
  49. #include <misc/debug.h>
  50. #include <misc/print_macros.h>
  51. #include <misc/read_write_int.h>
  52. #include <misc/compare.h>
  53. #define BADDR_TYPE_NONE 0
  54. #define BADDR_TYPE_IPV4 1
  55. #define BADDR_TYPE_IPV6 2
  56. #ifdef BADVPN_LINUX
  57. #define BADDR_TYPE_PACKET 5
  58. #endif
  59. #define BADDR_MAX_ADDR_LEN 128
  60. #define BIPADDR_MAX_PRINT_LEN 40
  61. #define BADDR_MAX_PRINT_LEN 120
  62. #define BADDR_PACKET_HEADER_TYPE_ETHERNET 1
  63. #define BADDR_PACKET_PACKET_TYPE_HOST 1
  64. #define BADDR_PACKET_PACKET_TYPE_BROADCAST 2
  65. #define BADDR_PACKET_PACKET_TYPE_MULTICAST 3
  66. #define BADDR_PACKET_PACKET_TYPE_OTHERHOST 4
  67. #define BADDR_PACKET_PACKET_TYPE_OUTGOING 5
  68. typedef struct {
  69. int type;
  70. union {
  71. uint32_t ipv4;
  72. uint8_t ipv6[16];
  73. };
  74. } BIPAddr;
  75. static void BIPAddr_InitInvalid (BIPAddr *addr);
  76. static void BIPAddr_InitIPv4 (BIPAddr *addr, uint32_t ip);
  77. static void BIPAddr_InitIPv6 (BIPAddr *addr, uint8_t *ip);
  78. static void BIPAddr_Assert (BIPAddr *addr);
  79. static int BIPAddr_IsInvalid (BIPAddr *addr);
  80. static int BIPAddr_Resolve (BIPAddr *addr, char *str, int noresolve) WARN_UNUSED;
  81. static int BIPAddr_Compare (BIPAddr *addr1, BIPAddr *addr2);
  82. /**
  83. * Converts an IP address to human readable form.
  84. *
  85. * @param addr IP address to convert
  86. * @param out destination buffer. Must be at least BIPADDR_MAX_PRINT_LEN characters long.
  87. */
  88. static void BIPAddr_Print (BIPAddr *addr, char *out);
  89. /**
  90. * Socket address - IP address and transport protocol port number
  91. */
  92. typedef struct {
  93. int type;
  94. union {
  95. struct {
  96. uint32_t ip;
  97. uint16_t port;
  98. } ipv4;
  99. struct {
  100. uint8_t ip[16];
  101. uint16_t port;
  102. } ipv6;
  103. struct {
  104. uint16_t phys_proto;
  105. int interface_index;
  106. int header_type;
  107. int packet_type;
  108. uint8_t phys_addr[8];
  109. } packet;
  110. };
  111. } BAddr;
  112. /**
  113. * Initializes an invalid address.
  114. */
  115. static void BAddr_InitNone (BAddr *addr);
  116. /**
  117. * Initializes an IPv4 address.
  118. *
  119. * @param addr the object
  120. * @param ip IP address in network byte order
  121. * @param port port number in network byte order
  122. */
  123. static void BAddr_InitIPv4 (BAddr *addr, uint32_t ip, uint16_t port);
  124. /**
  125. * Initializes an IPv6 address.
  126. *
  127. * @param addr the object
  128. * @param ip 16-byte IP address in network byte order
  129. * @param port port number in network byte order
  130. */
  131. static void BAddr_InitIPv6 (BAddr *addr, uint8_t *ip, uint16_t port);
  132. static void BAddr_InitFromIpaddrAndPort (BAddr *addr, BIPAddr ipaddr, uint16_t port);
  133. /**
  134. * Initializes a packet socket (data link layer) address.
  135. * Only Ethernet addresses are supported.
  136. *
  137. * @param addr the object
  138. * @param phys_proto identifier for the upper protocol, network byte order (EtherType)
  139. * @param interface_index network interface index
  140. * @param header_type data link layer header type. Must be BADDR_PACKET_HEADER_TYPE_ETHERNET.
  141. * @param packet_type the manner in which packets are sent/received. Must be one of
  142. * BADDR_PACKET_PACKET_TYPE_HOST, BADDR_PACKET_PACKET_TYPE_BROADCAST,
  143. * BADDR_PACKET_PACKET_TYPE_MULTICAST, BADDR_PACKET_PACKET_TYPE_OTHERHOST,
  144. * BADDR_PACKET_PACKET_TYPE_OUTGOING.
  145. * @param phys_addr data link layer address (MAC address)
  146. */
  147. static void BAddr_InitPacket (BAddr *addr, uint16_t phys_proto, int interface_index, int header_type, int packet_type, uint8_t *phys_addr);
  148. /**
  149. * Does nothing.
  150. *
  151. * @param addr the object
  152. */
  153. static void BAddr_Assert (BAddr *addr);
  154. /**
  155. * Determines whether the address is an invalid address.
  156. *
  157. * @param addr the object
  158. * @return 1 if invalid, 0 if invalid
  159. **/
  160. static int BAddr_IsInvalid (BAddr *addr);
  161. /**
  162. * Returns the port number in the address.
  163. *
  164. * @param addr the object
  165. * Must be an IPv4 or IPv6 address.
  166. * @return port number, in network byte order
  167. */
  168. static uint16_t BAddr_GetPort (BAddr *addr);
  169. /**
  170. * Returns the IP address in the address.
  171. *
  172. * @param addr the object
  173. * @param ipaddr IP address will be returned here. If \a addr is not
  174. * an IPv4 or IPv6 address, an invalid address will be
  175. * returned.
  176. */
  177. static void BAddr_GetIPAddr (BAddr *addr, BIPAddr *ipaddr);
  178. /**
  179. * Sets the port number in the address.
  180. *
  181. * @param addr the object
  182. * Must be an IPv4 or IPv6 address.
  183. * @param port port number, in network byte order
  184. */
  185. static void BAddr_SetPort (BAddr *addr, uint16_t port);
  186. /**
  187. * Converts an IP address to human readable form.
  188. *
  189. * @param addr address to convert
  190. * @param out destination buffer. Must be at least BADDR_MAX_PRINT_LEN characters long.
  191. */
  192. static void BAddr_Print (BAddr *addr, char *out);
  193. /**
  194. * Resolves an address string.
  195. * Format is "addr:port" for IPv4, "[addr]:port" for IPv6.
  196. * addr is be a numeric address or a name.
  197. * port is a numeric port number.
  198. *
  199. * @param addr output address
  200. * @param name if not NULL, the name portion of the address will be
  201. * stored here
  202. * @param name_len if name is not NULL, the size of the name buffer
  203. * @param noresolve only accept numeric addresses. Avoids blocking the caller.
  204. * @return 1 on success, 0 on parse error
  205. */
  206. static int BAddr_Parse2 (BAddr *addr, char *str, char *name, int name_len, int noresolve) WARN_UNUSED;
  207. /**
  208. * Resolves an address string.
  209. * IPv4 input format is "a.b.c.d:p", where a.b.c.d is the IP address
  210. * and d is the port number.
  211. * IPv6 input format is "[addr]:p", where addr is an IPv6 addres in
  212. * standard notation and p is the port number.
  213. *
  214. * @param addr output address
  215. * @param name if not NULL, the name portion of the address will be
  216. * stored here
  217. * @param name_len if name is not NULL, the size of the name buffer
  218. * @return 1 on success, 0 on parse error
  219. */
  220. static int BAddr_Parse (BAddr *addr, char *str, char *name, int name_len) WARN_UNUSED;
  221. static int BAddr_Compare (BAddr *addr1, BAddr *addr2);
  222. static int BAddr_CompareOrder (BAddr *addr1, BAddr *addr2);
  223. void BIPAddr_InitInvalid (BIPAddr *addr)
  224. {
  225. addr->type = BADDR_TYPE_NONE;
  226. }
  227. void BIPAddr_InitIPv4 (BIPAddr *addr, uint32_t ip)
  228. {
  229. addr->type = BADDR_TYPE_IPV4;
  230. addr->ipv4 = ip;
  231. }
  232. void BIPAddr_InitIPv6 (BIPAddr *addr, uint8_t *ip)
  233. {
  234. addr->type = BADDR_TYPE_IPV6;
  235. memcpy(addr->ipv6, ip, 16);
  236. }
  237. void BIPAddr_Assert (BIPAddr *addr)
  238. {
  239. switch (addr->type) {
  240. case BADDR_TYPE_NONE:
  241. case BADDR_TYPE_IPV4:
  242. case BADDR_TYPE_IPV6:
  243. return;
  244. default:
  245. ASSERT(0);
  246. }
  247. }
  248. int BIPAddr_IsInvalid (BIPAddr *addr)
  249. {
  250. BIPAddr_Assert(addr);
  251. return (addr->type == BADDR_TYPE_NONE);
  252. }
  253. int BIPAddr_Resolve (BIPAddr *addr, char *str, int noresolve)
  254. {
  255. int len = strlen(str);
  256. char *addr_start;
  257. int addr_len;
  258. // determine address type
  259. if (len >= 1 && str[0] == '[' && str[len - 1] == ']') {
  260. addr->type = BADDR_TYPE_IPV6;
  261. addr_start = str + 1;
  262. addr_len = len - 2;
  263. } else {
  264. addr->type = BADDR_TYPE_IPV4;
  265. addr_start = str;
  266. addr_len = len;
  267. }
  268. // copy
  269. char addr_str[BADDR_MAX_ADDR_LEN + 1];
  270. if (addr_len > BADDR_MAX_ADDR_LEN) {
  271. return 0;
  272. }
  273. memcpy(addr_str, addr_start, addr_len);
  274. addr_str[addr_len] = '\0';
  275. // initialize hints
  276. struct addrinfo hints;
  277. memset(&hints, 0, sizeof(hints));
  278. switch (addr->type) {
  279. case BADDR_TYPE_IPV6:
  280. hints.ai_family = AF_INET6;
  281. break;
  282. case BADDR_TYPE_IPV4:
  283. hints.ai_family = AF_INET;
  284. break;
  285. }
  286. if (noresolve) {
  287. hints.ai_flags |= AI_NUMERICHOST;
  288. }
  289. // call getaddrinfo
  290. struct addrinfo *addrs;
  291. int res;
  292. if ((res = getaddrinfo(addr_str, NULL, &hints, &addrs)) != 0) {
  293. return 0;
  294. }
  295. // set address
  296. switch (addr->type) {
  297. case BADDR_TYPE_IPV6:
  298. memcpy(addr->ipv6, ((struct sockaddr_in6 *)addrs->ai_addr)->sin6_addr.s6_addr, sizeof(addr->ipv6));
  299. break;
  300. case BADDR_TYPE_IPV4:
  301. addr->ipv4 = ((struct sockaddr_in *)addrs->ai_addr)->sin_addr.s_addr;
  302. break;
  303. }
  304. freeaddrinfo(addrs);
  305. return 1;
  306. }
  307. int BIPAddr_Compare (BIPAddr *addr1, BIPAddr *addr2)
  308. {
  309. BIPAddr_Assert(addr1);
  310. BIPAddr_Assert(addr2);
  311. if (addr1->type != addr2->type) {
  312. return 0;
  313. }
  314. switch (addr1->type) {
  315. case BADDR_TYPE_NONE:
  316. return 0;
  317. case BADDR_TYPE_IPV4:
  318. return (addr1->ipv4 == addr2->ipv4);
  319. case BADDR_TYPE_IPV6:
  320. return (!memcmp(addr1->ipv6, addr2->ipv6, sizeof(addr1->ipv6)));
  321. default:
  322. ASSERT(0)
  323. return 0;
  324. }
  325. }
  326. uint16_t BAddr_GetPort (BAddr *addr)
  327. {
  328. BAddr_Assert(addr);
  329. ASSERT(addr->type == BADDR_TYPE_IPV4 || addr->type == BADDR_TYPE_IPV6)
  330. switch (addr->type) {
  331. case BADDR_TYPE_IPV4:
  332. return addr->ipv4.port;
  333. case BADDR_TYPE_IPV6:
  334. return addr->ipv6.port;
  335. default:
  336. ASSERT(0)
  337. return 0;
  338. }
  339. }
  340. void BAddr_GetIPAddr (BAddr *addr, BIPAddr *ipaddr)
  341. {
  342. BAddr_Assert(addr);
  343. switch (addr->type) {
  344. case BADDR_TYPE_IPV4:
  345. BIPAddr_InitIPv4(ipaddr, addr->ipv4.ip);
  346. return;
  347. case BADDR_TYPE_IPV6:
  348. BIPAddr_InitIPv6(ipaddr, addr->ipv6.ip);
  349. return;
  350. default:
  351. BIPAddr_InitInvalid(ipaddr);
  352. }
  353. }
  354. void BAddr_SetPort (BAddr *addr, uint16_t port)
  355. {
  356. BAddr_Assert(addr);
  357. ASSERT(addr->type == BADDR_TYPE_IPV4 || addr->type == BADDR_TYPE_IPV6)
  358. switch (addr->type) {
  359. case BADDR_TYPE_IPV4:
  360. addr->ipv4.port = port;
  361. break;
  362. case BADDR_TYPE_IPV6:
  363. addr->ipv6.port = port;
  364. break;
  365. default:
  366. ASSERT(0);
  367. }
  368. }
  369. void BIPAddr_Print (BIPAddr *addr, char *out)
  370. {
  371. switch (addr->type) {
  372. case BADDR_TYPE_NONE:
  373. sprintf(out, "(none)");
  374. break;
  375. case BADDR_TYPE_IPV4:
  376. sprintf(out, "%"PRIu8".%"PRIu8".%"PRIu8".%"PRIu8,
  377. *((uint8_t *)&addr->ipv4 + 0),
  378. *((uint8_t *)&addr->ipv4 + 1),
  379. *((uint8_t *)&addr->ipv4 + 2),
  380. *((uint8_t *)&addr->ipv4 + 3)
  381. );
  382. break;
  383. case BADDR_TYPE_IPV6: {
  384. const char *ptr = (const char *)addr->ipv6;
  385. sprintf(out,
  386. "%"PRIx16":%"PRIx16":%"PRIx16":%"PRIx16":"
  387. "%"PRIx16":%"PRIx16":%"PRIx16":%"PRIx16,
  388. badvpn_read_be16(ptr + 0),
  389. badvpn_read_be16(ptr + 2),
  390. badvpn_read_be16(ptr + 4),
  391. badvpn_read_be16(ptr + 6),
  392. badvpn_read_be16(ptr + 8),
  393. badvpn_read_be16(ptr + 10),
  394. badvpn_read_be16(ptr + 12),
  395. badvpn_read_be16(ptr + 14)
  396. );
  397. } break;
  398. default:
  399. ASSERT(0);
  400. }
  401. }
  402. void BAddr_InitNone (BAddr *addr)
  403. {
  404. addr->type = BADDR_TYPE_NONE;
  405. }
  406. void BAddr_InitIPv4 (BAddr *addr, uint32_t ip, uint16_t port)
  407. {
  408. addr->type = BADDR_TYPE_IPV4;
  409. addr->ipv4.ip = ip;
  410. addr->ipv4.port = port;
  411. }
  412. void BAddr_InitIPv6 (BAddr *addr, uint8_t *ip, uint16_t port)
  413. {
  414. addr->type = BADDR_TYPE_IPV6;
  415. memcpy(addr->ipv6.ip, ip, 16);
  416. addr->ipv6.port = port;
  417. }
  418. void BAddr_InitFromIpaddrAndPort (BAddr *addr, BIPAddr ipaddr, uint16_t port)
  419. {
  420. BIPAddr_Assert(&ipaddr);
  421. switch (ipaddr.type) {
  422. case BADDR_TYPE_NONE:
  423. BAddr_InitNone(addr);
  424. break;
  425. case BADDR_TYPE_IPV4:
  426. BAddr_InitIPv4(addr, ipaddr.ipv4, port);
  427. break;
  428. case BADDR_TYPE_IPV6:
  429. BAddr_InitIPv6(addr, ipaddr.ipv6, port);
  430. break;
  431. default: ASSERT(0);
  432. }
  433. }
  434. #ifdef BADVPN_LINUX
  435. void BAddr_InitPacket (BAddr *addr, uint16_t phys_proto, int interface_index, int header_type, int packet_type, uint8_t *phys_addr)
  436. {
  437. ASSERT(header_type == BADDR_PACKET_HEADER_TYPE_ETHERNET)
  438. ASSERT(packet_type == BADDR_PACKET_PACKET_TYPE_HOST || packet_type == BADDR_PACKET_PACKET_TYPE_BROADCAST ||
  439. packet_type == BADDR_PACKET_PACKET_TYPE_MULTICAST || packet_type == BADDR_PACKET_PACKET_TYPE_OTHERHOST ||
  440. packet_type == BADDR_PACKET_PACKET_TYPE_OUTGOING)
  441. addr->type = BADDR_TYPE_PACKET;
  442. addr->packet.phys_proto = phys_proto;
  443. addr->packet.interface_index = interface_index;
  444. addr->packet.header_type = header_type;
  445. addr->packet.packet_type = packet_type;
  446. memcpy(addr->packet.phys_addr, phys_addr, 6);
  447. }
  448. #endif
  449. void BAddr_Assert (BAddr *addr)
  450. {
  451. switch (addr->type) {
  452. case BADDR_TYPE_NONE:
  453. case BADDR_TYPE_IPV4:
  454. case BADDR_TYPE_IPV6:
  455. #ifdef BADVPN_LINUX
  456. case BADDR_TYPE_PACKET:
  457. #endif
  458. return;
  459. default:
  460. ASSERT(0);
  461. }
  462. }
  463. int BAddr_IsInvalid (BAddr *addr)
  464. {
  465. BAddr_Assert(addr);
  466. return (addr->type == BADDR_TYPE_NONE);
  467. }
  468. void BAddr_Print (BAddr *addr, char *out)
  469. {
  470. BAddr_Assert(addr);
  471. BIPAddr ipaddr;
  472. switch (addr->type) {
  473. case BADDR_TYPE_NONE:
  474. sprintf(out, "(none)");
  475. break;
  476. case BADDR_TYPE_IPV4:
  477. BIPAddr_InitIPv4(&ipaddr, addr->ipv4.ip);
  478. BIPAddr_Print(&ipaddr, out);
  479. sprintf(out + strlen(out), ":%"PRIu16, ntoh16(addr->ipv4.port));
  480. break;
  481. case BADDR_TYPE_IPV6:
  482. BIPAddr_InitIPv6(&ipaddr, addr->ipv6.ip);
  483. BIPAddr_Print(&ipaddr, out);
  484. sprintf(out + strlen(out), ":%"PRIu16, ntoh16(addr->ipv6.port));
  485. break;
  486. #ifdef BADVPN_LINUX
  487. case BADDR_TYPE_PACKET:
  488. ASSERT(addr->packet.header_type == BADDR_PACKET_HEADER_TYPE_ETHERNET)
  489. sprintf(out, "proto=%"PRIu16",ifindex=%d,htype=eth,ptype=%d,addr=%02"PRIx8":%02"PRIx8":%02"PRIx8":%02"PRIx8":%02"PRIx8":%02"PRIx8,
  490. addr->packet.phys_proto, (int)addr->packet.interface_index, (int)addr->packet.packet_type,
  491. addr->packet.phys_addr[0], addr->packet.phys_addr[1], addr->packet.phys_addr[2],
  492. addr->packet.phys_addr[3], addr->packet.phys_addr[4], addr->packet.phys_addr[5]);
  493. break;
  494. #endif
  495. default:
  496. ASSERT(0);
  497. }
  498. }
  499. int BAddr_Parse2 (BAddr *addr, char *str, char *name, int name_len, int noresolve)
  500. {
  501. int len = strlen(str);
  502. if (len < 1 || len > 1000) {
  503. return 0;
  504. }
  505. int addr_start;
  506. int addr_len;
  507. int port_start;
  508. int port_len;
  509. // leading '[' indicates an IPv6 address
  510. if (str[0] == '[') {
  511. addr->type = BADDR_TYPE_IPV6;
  512. // find ']'
  513. int i=1;
  514. while (i < len && str[i] != ']') i++;
  515. if (i >= len) {
  516. return 0;
  517. }
  518. addr_start = 1;
  519. addr_len = i - addr_start;
  520. // follows ':' and port number
  521. if (i + 1 >= len || str[i + 1] != ':') {
  522. return 0;
  523. }
  524. port_start = i + 2;
  525. port_len = len - port_start;
  526. }
  527. // otherwise it's an IPv4 address
  528. else {
  529. addr->type = BADDR_TYPE_IPV4;
  530. // find ':'
  531. int i=0;
  532. while (i < len && str[i] != ':') i++;
  533. if (i >= len) {
  534. return 0;
  535. }
  536. addr_start = 0;
  537. addr_len = i - addr_start;
  538. port_start = i + 1;
  539. port_len = len - port_start;
  540. }
  541. // copy address and port to zero-terminated buffers
  542. char addr_str[128];
  543. if (addr_len >= sizeof(addr_str)) {
  544. return 0;
  545. }
  546. memcpy(addr_str, str + addr_start, addr_len);
  547. addr_str[addr_len] = '\0';
  548. char port_str[6];
  549. if (port_len >= sizeof(port_str)) {
  550. return 0;
  551. }
  552. memcpy(port_str, str + port_start, port_len);
  553. port_str[port_len] = '\0';
  554. // parse port
  555. char *err;
  556. long int conv_res = strtol(port_str, &err, 10);
  557. if (port_str[0] == '\0' || *err != '\0') {
  558. return 0;
  559. }
  560. if (conv_res < 0 || conv_res > UINT16_MAX) {
  561. return 0;
  562. }
  563. uint16_t port = conv_res;
  564. port = hton16(port);
  565. // initialize hints
  566. struct addrinfo hints;
  567. memset(&hints, 0, sizeof(hints));
  568. switch (addr->type) {
  569. case BADDR_TYPE_IPV6:
  570. hints.ai_family = AF_INET6;
  571. break;
  572. case BADDR_TYPE_IPV4:
  573. hints.ai_family = AF_INET;
  574. break;
  575. }
  576. if (noresolve) {
  577. hints.ai_flags |= AI_NUMERICHOST;
  578. }
  579. // call getaddrinfo
  580. struct addrinfo *addrs;
  581. int res;
  582. if ((res = getaddrinfo(addr_str, NULL, &hints, &addrs)) != 0) {
  583. return 0;
  584. }
  585. // set address
  586. switch (addr->type) {
  587. case BADDR_TYPE_IPV6:
  588. memcpy(addr->ipv6.ip, ((struct sockaddr_in6 *)addrs->ai_addr)->sin6_addr.s6_addr, sizeof(addr->ipv6.ip));
  589. addr->ipv6.port = port;
  590. break;
  591. case BADDR_TYPE_IPV4:
  592. addr->ipv4.ip = ((struct sockaddr_in *)addrs->ai_addr)->sin_addr.s_addr;
  593. addr->ipv4.port = port;
  594. break;
  595. }
  596. freeaddrinfo(addrs);
  597. if (name) {
  598. if (strlen(addr_str) >= name_len) {
  599. return 0;
  600. }
  601. strcpy(name, addr_str);
  602. }
  603. return 1;
  604. }
  605. int BAddr_Parse (BAddr *addr, char *str, char *name, int name_len)
  606. {
  607. return BAddr_Parse2(addr, str, name, name_len, 0);
  608. }
  609. int BAddr_Compare (BAddr *addr1, BAddr *addr2)
  610. {
  611. BAddr_Assert(addr1);
  612. BAddr_Assert(addr2);
  613. if (addr1->type != addr2->type) {
  614. return 0;
  615. }
  616. switch (addr1->type) {
  617. case BADDR_TYPE_IPV4:
  618. return (addr1->ipv4.ip == addr2->ipv4.ip && addr1->ipv4.port == addr2->ipv4.port);
  619. case BADDR_TYPE_IPV6:
  620. return (!memcmp(addr1->ipv6.ip, addr2->ipv6.ip, sizeof(addr1->ipv6.ip)) && addr1->ipv6.port == addr2->ipv6.port);
  621. default:
  622. return 0;
  623. }
  624. }
  625. int BAddr_CompareOrder (BAddr *addr1, BAddr *addr2)
  626. {
  627. BAddr_Assert(addr1);
  628. BAddr_Assert(addr2);
  629. int cmp = B_COMPARE(addr1->type, addr2->type);
  630. if (cmp) {
  631. return cmp;
  632. }
  633. switch (addr1->type) {
  634. case BADDR_TYPE_NONE: {
  635. return 0;
  636. } break;
  637. case BADDR_TYPE_IPV4: {
  638. uint32_t ip1 = ntoh32(addr1->ipv4.ip);
  639. uint32_t ip2 = ntoh32(addr2->ipv4.ip);
  640. cmp = B_COMPARE(ip1, ip2);
  641. if (cmp) {
  642. return cmp;
  643. }
  644. uint16_t port1 = ntoh16(addr1->ipv4.port);
  645. uint16_t port2 = ntoh16(addr2->ipv4.port);
  646. return B_COMPARE(port1, port2);
  647. } break;
  648. case BADDR_TYPE_IPV6: {
  649. cmp = memcmp(addr1->ipv6.ip, addr2->ipv6.ip, sizeof(addr1->ipv6.ip));
  650. if (cmp) {
  651. return B_COMPARE(cmp, 0);
  652. }
  653. uint16_t port1 = ntoh16(addr1->ipv6.port);
  654. uint16_t port2 = ntoh16(addr2->ipv6.port);
  655. return B_COMPARE(port1, port2);
  656. } break;
  657. default: {
  658. return 0;
  659. } break;
  660. }
  661. }
  662. #endif