db_sockets.c 42 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480
  1. /* -*- mode:c; c-file-style:"k&r"; c-basic-offset: 4; tab-width:4; indent-tabs-mode:nil; mode:auto-fill; fill-column:78; -*- */
  2. /* vim: set ts=4 sw=4 et tw=78 fo=cqt wm=0: */
  3. /* Copyright (C) 2014 Stony Brook University
  4. This file is part of Graphene Library OS.
  5. Graphene Library OS is free software: you can redistribute it and/or
  6. modify it under the terms of the GNU Lesser General Public License
  7. as published by the Free Software Foundation, either version 3 of the
  8. License, or (at your option) any later version.
  9. Graphene Library OS is distributed in the hope that it will be useful,
  10. but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. GNU Lesser General Public License for more details.
  13. You should have received a copy of the GNU Lesser General Public License
  14. along with this program. If not, see <http://www.gnu.org/licenses/>. */
  15. /*
  16. * db_socket.c
  17. *
  18. * This file contains operands for streams with URIs that start with
  19. * "tcp:", "tcp.srv:", "udp:", "udp.srv:".
  20. */
  21. #include "pal_defs.h"
  22. #include "pal_freebsd_defs.h"
  23. #include "pal.h"
  24. #include "pal_internal.h"
  25. #include "pal_freebsd.h"
  26. #include "pal_debug.h"
  27. #include "pal_security.h"
  28. #include "pal_error.h"
  29. #include "api.h"
  30. #include <sys/types.h>
  31. #include <poll.h>
  32. typedef __kernel_pid_t pid_t;
  33. #include <fcntl.h>
  34. #include <sys/socket.h>
  35. #include <netinet/in.h>
  36. #include <arpa/inet.h>
  37. #include <netinet/tcp.h>
  38. #include <errno.h>
  39. #include <sys/filio.h>
  40. /* FreeBSD doesn't have addr_check_any. */
  41. #undef ALLOW_BIND_ANY
  42. #define ALLOW_BIND_ANY 1
  43. /* 96 bytes is the minimal size of buffer to store a IPv4/IPv6
  44. address */
  45. #define PAL_SOCKADDR_SIZE 96
  46. typedef uint16_t __be16;
  47. #define SOL_TCP 6
  48. #define TCP_CORK TCP_NOPUSH
  49. static size_t addr_size(const struct sockaddr* addr) {
  50. switch (addr->sa_family) {
  51. case AF_INET:
  52. return sizeof(struct sockaddr_in);
  53. case AF_INET6:
  54. return sizeof(struct sockaddr_in6);
  55. default:
  56. return 0;
  57. }
  58. }
  59. /* parsing the string of uri, and fill in the socket address structure.
  60. the latest pointer of uri, length of socket address are returned. */
  61. static int inet_parse_uri (char ** uri, struct sockaddr * addr, int * addrlen)
  62. {
  63. char * tmp = *uri, * end;
  64. char * addr_str = NULL, * port_str;
  65. int af;
  66. void * addr_buf;
  67. int addr_len;
  68. __be16 * port_buf;
  69. int slen;
  70. if (tmp[0] == '[') {
  71. /* for IPv6, the address will be in the form of
  72. "[xx:xx:xx:xx:xx:xx:xx:xx]:port". */
  73. struct sockaddr_in6 * addr_in6 = (struct sockaddr_in6 *) addr;
  74. slen = sizeof(struct sockaddr_in6);
  75. memset(addr, 0, slen);
  76. end = strchr(tmp + 1, ']');
  77. if (!end || *(end + 1) != ':')
  78. goto inval;
  79. addr_str = tmp + 1;
  80. addr_len = end - tmp + 1;
  81. port_str = end + 2;
  82. for (end = port_str ; *end >= '0' && *end <= '9' ; end++);
  83. addr_in6->sin6_family = af = AF_INET6;
  84. addr_buf = &addr_in6->sin6_addr.s6_addr;
  85. port_buf = &addr_in6->sin6_port;
  86. } else {
  87. /* for IP, the address will be in the form of "x.x.x.x:port". */
  88. struct sockaddr_in * addr_in = (struct sockaddr_in *) addr;
  89. slen = sizeof(struct sockaddr_in);
  90. memset(addr, 0, slen);
  91. end = strchr(tmp, ':');
  92. if (!end)
  93. goto inval;
  94. addr_str = tmp;
  95. addr_len = end - tmp;
  96. port_str = end + 1;
  97. for (end = port_str ; *end >= '0' && *end <= '9' ; end++);
  98. addr_in->sin_family = af = AF_INET;
  99. addr_buf = &addr_in->sin_addr.s_addr;
  100. port_buf = &addr_in->sin_port;
  101. }
  102. if (af == AF_INET) {
  103. if (inet_pton4(addr_str, addr_len, addr_buf) < 0)
  104. goto inval;
  105. } else {
  106. if (inet_pton6(addr_str, addr_len, addr_buf) < 0)
  107. goto inval;
  108. }
  109. *port_buf = __htons(atoi(port_str));
  110. *uri = *end ? end + 1 : NULL;
  111. if (addrlen)
  112. *addrlen = slen;
  113. return 0;
  114. inval:
  115. return -PAL_ERROR_INVAL;
  116. }
  117. /* create the string of uri from the given socket address */
  118. static int inet_create_uri (char * uri, int count, struct sockaddr * addr,
  119. int addrlen)
  120. {
  121. int len = 0;
  122. if (addr->sa_family == AF_INET) {
  123. if (addrlen != sizeof(struct sockaddr_in))
  124. return PAL_ERROR_INVAL;
  125. struct sockaddr_in * addr_in = (struct sockaddr_in *) addr;
  126. char * addr = (char *) &addr_in->sin_addr.s_addr;
  127. /* for IP, the address will be in the form of "x.x.x.x:port". */
  128. len = snprintf(uri, count, "%u.%u.%u.%u:%u",
  129. (unsigned char) addr[0],
  130. (unsigned char) addr[1],
  131. (unsigned char) addr[2],
  132. (unsigned char) addr[3],
  133. __ntohs(addr_in->sin_port));
  134. } else if (addr->sa_family == AF_INET6) {
  135. if (addrlen != sizeof(struct sockaddr_in6))
  136. return PAL_ERROR_INVAL;
  137. struct sockaddr_in6 * addr_in6 = (struct sockaddr_in6 *) addr;
  138. short * addr = (short *) &addr_in6->sin6_addr.s6_addr;
  139. /* for IPv6, the address will be in the form of
  140. "[xx:xx:xx:xx:xx:xx:xx:xx]:port". */
  141. len = snprintf(uri, count, "[%x:%x:%x:%x:%x:%x:%x:%x]:%u",
  142. addr[0], addr[1], addr[2], addr[3],
  143. addr[4], addr[5], addr[6], addr[7],
  144. __ntohs(addr_in6->sin6_port));
  145. } else {
  146. return -PAL_ERROR_INVAL;
  147. }
  148. return len;
  149. }
  150. /* parse the uri for a socket stream. The uri might have both binding
  151. address and connecting address, or connecting address only. The form
  152. of uri will be either "bind-addr:bind-port:connect-addr:connect-port"
  153. or "addr:port". */
  154. static int socket_parse_uri (char * uri,
  155. struct sockaddr ** bind_addr, int * bind_addrlen,
  156. struct sockaddr ** dest_addr, int * dest_addrlen)
  157. {
  158. int ret;
  159. if (!bind_addr && !dest_addr)
  160. return 0;
  161. if (!uri || !(*uri)) {
  162. if (bind_addr)
  163. *bind_addr = NULL;
  164. if (dest_addr)
  165. *dest_addr = NULL;
  166. return 0;
  167. }
  168. /* at least parse uri once */
  169. if ((ret = inet_parse_uri(&uri, bind_addr ? *bind_addr : *dest_addr,
  170. bind_addr ? bind_addrlen : dest_addrlen)) < 0)
  171. return ret;
  172. if (!(bind_addr && dest_addr))
  173. return 0;
  174. /* if you reach here, it can only be connection address */
  175. if (!uri || (ret = inet_parse_uri(&uri, *dest_addr, dest_addrlen)) < 0) {
  176. *dest_addr = *bind_addr;
  177. *dest_addrlen = *bind_addrlen;
  178. *bind_addr = NULL;
  179. *bind_addrlen = 0;
  180. }
  181. return 0;
  182. }
  183. /* fill in the PAL handle based on the file descriptors and address given. */
  184. static inline
  185. PAL_HANDLE socket_create_handle (int type, int fd, int options,
  186. struct sockaddr * bind_addr, int bind_addrlen,
  187. struct sockaddr * dest_addr, int dest_addrlen)
  188. {
  189. PAL_HANDLE hdl = malloc(HANDLE_SIZE(sock) + (bind_addr ? bind_addrlen : 0) +
  190. (dest_addr ? dest_addrlen : 0));
  191. if (!hdl)
  192. return NULL;
  193. options = HOST_SOCKET_OPTIONS(options);
  194. memset(hdl, 0, sizeof(union pal_handle));
  195. PAL_GET_TYPE(hdl) = type;
  196. hdl->hdr.flags |= RFD(0)|(type != pal_type_tcpsrv ? WFD(0) : 0);
  197. hdl->sock.fd = fd;
  198. void * addr = (void *) hdl + HANDLE_SIZE(sock);
  199. if (bind_addr) {
  200. hdl->sock.bind = addr;
  201. memcpy(addr, bind_addr, bind_addrlen);
  202. addr += bind_addrlen;
  203. } else {
  204. hdl->sock.bind = NULL;
  205. }
  206. if (dest_addr) {
  207. hdl->sock.conn = addr;
  208. memcpy(addr, dest_addr, dest_addrlen);
  209. addr += dest_addrlen;
  210. } else {
  211. hdl->sock.conn = NULL;
  212. }
  213. hdl->sock.nonblocking = (options & SOCK_NONBLOCK) ?
  214. PAL_TRUE : PAL_FALSE;
  215. hdl->sock.linger = 0;
  216. if (type == pal_type_tcpsrv) {
  217. hdl->sock.receivebuf = 0;
  218. hdl->sock.sendbuf = 0;
  219. } else {
  220. int ret, val, len = sizeof(int);
  221. ret = INLINE_SYSCALL(getsockopt, 5, fd, SOL_SOCKET, SO_RCVBUF,
  222. &val, &len);
  223. hdl->sock.receivebuf = IS_ERR(ret) ? 0 : val;
  224. ret = INLINE_SYSCALL(getsockopt, 5, fd, SOL_SOCKET, SO_SNDBUF,
  225. &val, &len);
  226. hdl->sock.sendbuf = IS_ERR(ret) ? 0 : val;
  227. }
  228. hdl->sock.receivetimeout = 0;
  229. hdl->sock.sendtimeout = 0;
  230. hdl->sock.tcp_cork = PAL_FALSE;
  231. hdl->sock.tcp_keepalive = PAL_FALSE;
  232. hdl->sock.tcp_nodelay = PAL_FALSE;
  233. return hdl;
  234. }
  235. #if ALLOW_BIND_ANY == 0
  236. static bool check_zero (void * mem, size_t size)
  237. {
  238. void * p = mem, * q = mem + size;
  239. while (p < q) {
  240. if (p <= q - sizeof(long)) {
  241. if (*(long *) p)
  242. return false;
  243. p += sizeof(long);
  244. } else if (p <= q - sizeof(int)) {
  245. if (*(int *) p)
  246. return false;
  247. p += sizeof(int);
  248. } else if (p <= q - sizeof(short)) {
  249. if (*(short *) p)
  250. return false;
  251. p += sizeof(short);
  252. } else {
  253. if (*(char *) p)
  254. return false;
  255. p++;
  256. }
  257. }
  258. return true;
  259. }
  260. /* check if an address is "Any" */
  261. static bool check_any_addr (struct sockaddr * addr)
  262. {
  263. if (addr->sa_family == AF_INET) {
  264. struct sockaddr_in * addr_in =
  265. (struct sockaddr_in *) addr;
  266. return addr_in->sin_port == 0 &&
  267. check_zero(&addr_in->sin_addr,
  268. sizeof(addr_in->sin_addr));
  269. } else if (addr->sa_family == AF_INET6) {
  270. struct sockaddr_in6 * addr_in6 =
  271. (struct sockaddr_in6 *) addr;
  272. return addr_in6->sin6_port == 0 &&
  273. check_zero(&addr_in6->sin6_addr,
  274. sizeof(addr_in6->sin6_addr));
  275. }
  276. return false;
  277. }
  278. #endif
  279. /* listen on a tcp socket */
  280. static int tcp_listen (PAL_HANDLE * handle, char * uri, int options)
  281. {
  282. struct sockaddr buffer, * bind_addr = &buffer;
  283. int bind_addrlen;
  284. int ret, fd = -1;
  285. if ((ret = socket_parse_uri(uri, &bind_addr, &bind_addrlen,
  286. NULL, NULL)) < 0)
  287. return ret;
  288. #if ALLOW_BIND_ANY == 0
  289. /* the socket need to have a binding address, a null address or an
  290. any address is not allowed */
  291. if (addr_check_any(bind_addr))
  292. return -PAL_ERROR_INVAL;
  293. #endif
  294. options = HOST_SOCKET_OPTIONS(options);
  295. /* the socket need to have a binding address, a null address or an
  296. any address is not allowed */
  297. if (!bind_addr == 0)
  298. return -PAL_ERROR_INVAL;
  299. fd = INLINE_SYSCALL(socket, 3, bind_addr->sa_family,
  300. SOCK_STREAM|SOCK_CLOEXEC|options, 0);
  301. if (IS_ERR(fd))
  302. return -PAL_ERROR_DENIED;
  303. /* must set the socket to be reuseable */
  304. int reuseaddr = 1;
  305. INLINE_SYSCALL(setsockopt, 5, fd, SOL_SOCKET, SO_REUSEADDR, &reuseaddr,
  306. sizeof(int)); // maybe SO_REUSEPORT
  307. ret = INLINE_SYSCALL(bind, 3, fd, bind_addr, bind_addrlen);
  308. if (IS_ERR(ret)) {
  309. switch(ERRNO(ret)) {
  310. case EINVAL:
  311. ret = -PAL_ERROR_INVAL;
  312. goto failed;
  313. case EADDRINUSE:
  314. ret = -PAL_ERROR_STREAMEXIST;
  315. goto failed;
  316. default:
  317. ret = -PAL_ERROR_DENIED;
  318. goto failed;
  319. }
  320. }
  321. ret = INLINE_SYSCALL(listen, 2, fd, DEFAULT_BACKLOG);
  322. if (IS_ERR(ret))
  323. return -PAL_ERROR_DENIED;
  324. *handle = socket_create_handle(pal_type_tcpsrv, fd, options,
  325. bind_addr, bind_addrlen, NULL, 0);
  326. if (!(*handle)) {
  327. ret = -PAL_ERROR_NOMEM;
  328. goto failed;
  329. }
  330. return 0;
  331. failed:
  332. INLINE_SYSCALL(close, 1, fd);
  333. return ret;
  334. }
  335. /* accept a tcp connection */
  336. static int tcp_accept (PAL_HANDLE handle, PAL_HANDLE * client)
  337. {
  338. if (!IS_HANDLE_TYPE(handle, tcpsrv) ||
  339. !handle->sock.bind || handle->sock.conn)
  340. return -PAL_ERROR_NOTSERVER;
  341. if (handle->sock.fd == PAL_IDX_POISON)
  342. return -PAL_ERROR_BADHANDLE;
  343. struct sockaddr * bind_addr = (struct sockaddr *) handle->sock.bind;
  344. int bind_addrlen = addr_size(bind_addr);
  345. struct sockaddr buffer;
  346. socklen_t addrlen = sizeof(struct sockaddr);
  347. int ret = 0;
  348. int newfd = INLINE_SYSCALL(accept4, 4, handle->sock.fd, &buffer,
  349. &addrlen, SOCK_CLOEXEC);
  350. if (IS_ERR(newfd))
  351. switch(ERRNO(newfd)) {
  352. case EWOULDBLOCK:
  353. return -PAL_ERROR_TRYAGAIN;
  354. case ECONNABORTED:
  355. return -PAL_ERROR_STREAMNOTEXIST;
  356. default:
  357. return unix_to_pal_error(ERRNO(newfd));
  358. }
  359. struct sockaddr * dest_addr = &buffer;
  360. int dest_addrlen = addrlen;
  361. *client = socket_create_handle(pal_type_tcp, newfd, 0,
  362. bind_addr, bind_addrlen,
  363. dest_addr, dest_addrlen);
  364. if (!(*client)) {
  365. ret = -PAL_ERROR_NOMEM;
  366. goto failed;
  367. }
  368. return 0;
  369. failed:
  370. INLINE_SYSCALL(close, 1, newfd);
  371. return ret;
  372. }
  373. /* connect on a tcp socket */
  374. static int tcp_connect (PAL_HANDLE * handle, char * uri, int options)
  375. {
  376. struct sockaddr buffer[3];
  377. struct sockaddr * bind_addr = buffer, * dest_addr = buffer + 1;
  378. int bind_addrlen, dest_addrlen;
  379. int ret, fd = -1;
  380. options = HOST_SOCKET_OPTIONS(options);
  381. /* accepting two kind of different uri:
  382. dest-ip:dest-port or bind-ip:bind-port:dest-ip:dest-port */
  383. if ((ret = socket_parse_uri(uri, &bind_addr, &bind_addrlen,
  384. &dest_addr, &dest_addrlen)) < 0)
  385. return ret;
  386. if (!dest_addr)
  387. return -PAL_ERROR_INVAL;
  388. if (bind_addr && bind_addr->sa_family != dest_addr->sa_family)
  389. return -PAL_ERROR_INVAL;
  390. #if ALLOW_BIND_ANY == 0
  391. /* the socket need to have a binding address, a null address or an
  392. any address is not allowed */
  393. if (bind_addr && addr_check_any(bind_addr))
  394. return -PAL_ERROR_INVAL;
  395. #endif
  396. fd = INLINE_SYSCALL(socket, 3, dest_addr->sa_family,
  397. SOCK_STREAM|SOCK_CLOEXEC|options, 0);
  398. if (IS_ERR(fd))
  399. return -PAL_ERROR_DENIED;
  400. if (bind_addr) {
  401. if (IS_ERR(ret)) {
  402. INLINE_SYSCALL(close, 1, fd);
  403. switch (ERRNO(ret)) {
  404. case EADDRINUSE:
  405. ret = -PAL_ERROR_STREAMEXIST;
  406. goto failed;
  407. case EADDRNOTAVAIL:
  408. ret = -PAL_ERROR_ADDRNOTEXIST;
  409. goto failed;
  410. default:
  411. ret = unix_to_pal_error(ERRNO(ret));
  412. goto failed;
  413. }
  414. }
  415. }
  416. ret = INLINE_SYSCALL(connect, 3, fd, dest_addr, dest_addrlen);
  417. if (IS_ERR(ret)) {
  418. ret = unix_to_pal_error(ERRNO(ret));
  419. goto failed;
  420. }
  421. if (!bind_addr) {
  422. /* save some space to get socket address */
  423. bind_addr = buffer + 2;
  424. bind_addrlen = sizeof(struct sockaddr);
  425. /* call getsockname to get socket address */
  426. if ((ret = INLINE_SYSCALL(getsockname, 3, fd,
  427. bind_addr, &bind_addrlen)) < 0)
  428. bind_addr = NULL;
  429. }
  430. *handle = socket_create_handle(pal_type_tcp, fd, options,
  431. bind_addr, bind_addrlen,
  432. dest_addr, dest_addrlen);
  433. if (!(*handle)) {
  434. ret = -PAL_ERROR_NOMEM;
  435. goto failed;
  436. }
  437. return 0;
  438. failed:
  439. INLINE_SYSCALL(close, 1, fd);
  440. return ret;
  441. }
  442. /* 'open' operation of tcp stream */
  443. static int tcp_open (PAL_HANDLE *handle, const char * type, const char * uri,
  444. int access, int share, int create, int options)
  445. {
  446. int uri_len = strlen(uri) + 1;
  447. if (uri_len > PAL_SOCKADDR_SIZE)
  448. return -PAL_ERROR_TOOLONG;
  449. char uri_buf[PAL_SOCKADDR_SIZE];
  450. memcpy(uri_buf, uri, uri_len);
  451. if (strcmp_static(type, "tcp.srv"))
  452. return tcp_listen(handle, uri_buf, options);
  453. if (strcmp_static(type, "tcp"))
  454. return tcp_connect(handle, uri_buf, options);
  455. return -PAL_ERROR_NOTSUPPORT;
  456. }
  457. /* 'read' operation of tcp stream */
  458. static int tcp_read (PAL_HANDLE handle, int offset, int len, void * buf)
  459. {
  460. if (!IS_HANDLE_TYPE(handle, tcp) || !handle->sock.conn)
  461. return -PAL_ERROR_NOTCONNECTION;
  462. if (handle->sock.fd == PAL_IDX_POISON)
  463. return -PAL_ERROR_ENDOFSTREAM;
  464. struct msghdr hdr;
  465. struct iovec iov;
  466. iov.iov_base = buf;
  467. iov.iov_len = len;
  468. hdr.msg_name = NULL;
  469. hdr.msg_namelen = 0;
  470. hdr.msg_iov = &iov;
  471. hdr.msg_iovlen = 1;
  472. hdr.msg_control = NULL;
  473. hdr.msg_controllen = 0;
  474. hdr.msg_flags = 0;
  475. int bytes = INLINE_SYSCALL(recvmsg, 3, handle->sock.fd, &hdr, 0);
  476. if (IS_ERR(bytes))
  477. switch (ERRNO(bytes)) {
  478. case EWOULDBLOCK:
  479. return -PAL_ERROR_TRYAGAIN;
  480. default:
  481. return unix_to_pal_error(ERRNO(bytes));
  482. }
  483. if (!bytes)
  484. return -PAL_ERROR_ENDOFSTREAM;
  485. return bytes;
  486. }
  487. /* write' operation of tcp stream */
  488. static int tcp_write (PAL_HANDLE handle, int offset, int len, const void * buf)
  489. {
  490. if (!IS_HANDLE_TYPE(handle, tcp) || !handle->sock.conn)
  491. return -PAL_ERROR_NOTCONNECTION;
  492. if (handle->sock.fd == PAL_IDX_POISON)
  493. return -PAL_ERROR_CONNFAILED;
  494. struct msghdr hdr;
  495. struct iovec iov;
  496. iov.iov_base = (void *) buf;
  497. iov.iov_len = len;
  498. hdr.msg_name = NULL;
  499. hdr.msg_namelen = 0;
  500. hdr.msg_iov = &iov;
  501. hdr.msg_iovlen = 1;
  502. hdr.msg_control = NULL;
  503. hdr.msg_controllen = 0;
  504. hdr.msg_flags = 0;
  505. int bytes = INLINE_SYSCALL(sendmsg, 3, handle->sock.fd, &hdr, MSG_NOSIGNAL);
  506. if (IS_ERR(bytes))
  507. switch(ERRNO(bytes)) {
  508. case ECONNRESET:
  509. case EPIPE:
  510. return -PAL_ERROR_CONNFAILED;
  511. case EWOULDBLOCK:
  512. handle->hdr.flags &= ~WRITABLE(0);
  513. return -PAL_ERROR_TRYAGAIN;
  514. default:
  515. return unix_to_pal_error(ERRNO(bytes));
  516. }
  517. if (bytes == len)
  518. handle->hdr.flags |= WRITABLE(0);
  519. else
  520. handle->hdr.flags &= ~WRITABLE(0);
  521. return bytes;
  522. }
  523. /* used by 'open' operation of tcp stream for bound socket */
  524. static int udp_bind (PAL_HANDLE * handle, char * uri, int options)
  525. {
  526. struct sockaddr buffer, * bind_addr = &buffer;
  527. int bind_addrlen;
  528. int ret = 0, fd = -1;
  529. if ((ret = socket_parse_uri(uri, &bind_addr, &bind_addrlen,
  530. NULL, NULL)) < 0)
  531. return ret;
  532. assert(bind_addr);
  533. assert(bind_addrlen == addr_size(bind_addr));
  534. #if ALLOW_BIND_ANY == 0
  535. /* the socket need to have a binding address, a null address or an
  536. any address is not allowed */
  537. if (addr_check_any(bind_addr))
  538. return -PAL_ERROR_INVAL;
  539. #endif
  540. options = HOST_SOCKET_OPTIONS(options);
  541. fd = INLINE_SYSCALL(socket, 3, bind_addr->sa_family,
  542. SOCK_DGRAM|SOCK_CLOEXEC|options, 0);
  543. if (IS_ERR(fd))
  544. return -PAL_ERROR_DENIED;
  545. ret = INLINE_SYSCALL(bind, 3, fd, bind_addr, bind_addrlen);
  546. if (IS_ERR(ret)) {
  547. switch (ERRNO(ret)) {
  548. case EADDRINUSE:
  549. ret = -PAL_ERROR_STREAMEXIST;
  550. goto failed;
  551. case EADDRNOTAVAIL:
  552. ret = -PAL_ERROR_ADDRNOTEXIST;
  553. goto failed;
  554. default:
  555. ret = unix_to_pal_error(ERRNO(ret));
  556. goto failed;
  557. }
  558. }
  559. *handle = socket_create_handle(pal_type_udpsrv, fd, options,
  560. bind_addr, bind_addrlen, NULL, 0);
  561. if (!(*handle)) {
  562. ret = -ENOMEM;
  563. goto failed;
  564. }
  565. return 0;
  566. failed:
  567. INLINE_SYSCALL(close, 1, fd);
  568. return ret;
  569. }
  570. /* used by 'open' operation of tcp stream for connected socket */
  571. static int udp_connect (PAL_HANDLE * handle, char * uri, int options)
  572. {
  573. struct sockaddr buffer[2];
  574. struct sockaddr * bind_addr = buffer, * dest_addr = buffer + 1;
  575. int bind_addrlen, dest_addrlen;
  576. int ret, fd = -1;
  577. if ((ret = socket_parse_uri(uri, &bind_addr, &bind_addrlen,
  578. &dest_addr, &dest_addrlen)) < 0)
  579. return ret;
  580. #if ALLOW_BIND_ANY == 0
  581. /* the socket need to have a binding address, a null address or an
  582. any address is not allowed */
  583. if (bind_addr && addr_check_any(bind_addr))
  584. return -PAL_ERROR_INVAL;
  585. #endif
  586. options = HOST_SOCKET_OPTIONS(options);
  587. fd = INLINE_SYSCALL(socket, 3, dest_addr ? dest_addr->sa_family : AF_INET,
  588. SOCK_DGRAM|SOCK_CLOEXEC|options, 0);
  589. if (IS_ERR(fd))
  590. return -PAL_ERROR_DENIED;
  591. if (bind_addr) {
  592. ret = INLINE_SYSCALL(bind, 3, fd, bind_addr, bind_addrlen);
  593. if (IS_ERR(ret)) {
  594. switch (ERRNO(ret)) {
  595. case EADDRINUSE:
  596. ret = -PAL_ERROR_STREAMEXIST;
  597. goto failed;
  598. case EADDRNOTAVAIL:
  599. ret = -PAL_ERROR_ADDRNOTEXIST;
  600. goto failed;
  601. default:
  602. ret = unix_to_pal_error(ERRNO(ret));
  603. goto failed;
  604. }
  605. }
  606. }
  607. *handle = socket_create_handle(dest_addr ? pal_type_udp : pal_type_udpsrv,
  608. fd, options,
  609. bind_addr, bind_addrlen,
  610. dest_addr, dest_addrlen);
  611. if (!(*handle)) {
  612. ret = -ENOMEM;
  613. goto failed;
  614. }
  615. return 0;
  616. failed:
  617. INLINE_SYSCALL(close, 1, fd);
  618. return ret;
  619. }
  620. static int udp_open (PAL_HANDLE *hdl, const char * type, const char * uri,
  621. int access, int share, int create, int options)
  622. {
  623. char buf[PAL_SOCKADDR_SIZE];
  624. int len = strlen(uri);
  625. if (len >= PAL_SOCKADDR_SIZE)
  626. return -PAL_ERROR_TOOLONG;
  627. memcpy(buf, uri, len + 1);
  628. options &= PAL_OPTION_MASK;
  629. if (!strcmp_static(type, "udp.srv"))
  630. return udp_bind(hdl, buf, options);
  631. if (!strcmp_static(type, "udp"))
  632. return udp_connect(hdl, buf, options);
  633. return -PAL_ERROR_NOTSUPPORT;
  634. }
  635. static int udp_receive (PAL_HANDLE handle, int offset, int len, void * buf)
  636. {
  637. if (!IS_HANDLE_TYPE(handle, udp))
  638. return -PAL_ERROR_NOTCONNECTION;
  639. if (handle->sock.fd == PAL_IDX_POISON)
  640. return -PAL_ERROR_BADHANDLE;
  641. struct msghdr hdr;
  642. struct iovec iov;
  643. iov.iov_base = buf;
  644. iov.iov_len = len;
  645. hdr.msg_name = NULL;
  646. hdr.msg_namelen = 0;
  647. hdr.msg_iov = &iov;
  648. hdr.msg_iovlen = 1;
  649. hdr.msg_control = NULL;
  650. hdr.msg_controllen = 0;
  651. hdr.msg_flags = 0;
  652. int bytes = INLINE_SYSCALL(recvmsg, 3, handle->sock.fd, &hdr, 0);
  653. if (IS_ERR(bytes))
  654. switch(ERRNO(bytes)) {
  655. case EWOULDBLOCK:
  656. return -PAL_ERROR_TRYAGAIN;
  657. case EINTR:
  658. return -PAL_ERROR_INTERRUPTED;
  659. default:
  660. return unix_to_pal_error(ERRNO(bytes));
  661. }
  662. return bytes;
  663. }
  664. static int udp_receivebyaddr (PAL_HANDLE handle, int offset, int len,
  665. void * buf, char * addr, int addrlen)
  666. {
  667. if (!IS_HANDLE_TYPE(handle, udpsrv))
  668. return -PAL_ERROR_NOTCONNECTION;
  669. if (handle->sock.fd == PAL_IDX_POISON)
  670. return -PAL_ERROR_BADHANDLE;
  671. struct sockaddr conn_addr;
  672. socklen_t conn_addrlen = sizeof(struct sockaddr);
  673. struct msghdr hdr;
  674. struct iovec iov;
  675. iov.iov_base = buf;
  676. iov.iov_len = len;
  677. hdr.msg_name = &conn_addr;
  678. hdr.msg_namelen = conn_addrlen;
  679. hdr.msg_iov = &iov;
  680. hdr.msg_iovlen = 1;
  681. hdr.msg_control = NULL;
  682. hdr.msg_controllen = 0;
  683. hdr.msg_flags = 0;
  684. int bytes = INLINE_SYSCALL(recvmsg, 3, handle->sock.fd, &hdr, 0);
  685. if (IS_ERR(bytes))
  686. switch(ERRNO(bytes)) {
  687. case EWOULDBLOCK:
  688. return -PAL_ERROR_TRYAGAIN;
  689. case EINTR:
  690. return -PAL_ERROR_INTERRUPTED;
  691. case ECONNREFUSED:
  692. return -PAL_ERROR_STREAMNOTEXIST;
  693. default:
  694. return unix_to_pal_error(ERRNO(bytes));
  695. }
  696. char * tmp = strcpy_static(addr, "udp:", addrlen);
  697. if (!tmp)
  698. return -PAL_ERROR_OVERFLOW;
  699. inet_create_uri(tmp, addr + addrlen - tmp, &conn_addr, hdr.msg_namelen);
  700. return bytes;
  701. }
  702. static int udp_send (PAL_HANDLE handle, int offset, int len, const void * buf)
  703. {
  704. if (!IS_HANDLE_TYPE(handle, udp))
  705. return -PAL_ERROR_NOTCONNECTION;
  706. if (handle->sock.fd == PAL_IDX_POISON)
  707. return -PAL_ERROR_BADHANDLE;
  708. struct msghdr hdr;
  709. struct iovec iov;
  710. iov.iov_base = (void *) buf;
  711. iov.iov_len = len;
  712. hdr.msg_name = handle->sock.conn;
  713. hdr.msg_namelen = addr_size(handle->sock.conn);
  714. hdr.msg_iov = &iov;
  715. hdr.msg_iovlen = 1;
  716. hdr.msg_control = NULL;
  717. hdr.msg_controllen = 0;
  718. hdr.msg_flags = 0;
  719. int bytes = INLINE_SYSCALL(sendmsg, 3, handle->sock.fd, &hdr, MSG_NOSIGNAL);
  720. if (IS_ERR(bytes))
  721. switch(ERRNO(bytes)) {
  722. case EAGAIN:
  723. handle->hdr.flags &= ~WRITABLE(0);
  724. return -PAL_ERROR_TRYAGAIN;
  725. case ECONNRESET:
  726. case EPIPE:
  727. return -PAL_ERROR_CONNFAILED;
  728. default:
  729. return unix_to_pal_error(ERRNO(bytes));
  730. }
  731. if (bytes == len)
  732. handle->hdr.flags |= WRITABLE(0);
  733. else
  734. handle->hdr.flags &= ~WRITABLE(0);
  735. return bytes;
  736. }
  737. static int udp_sendbyaddr (PAL_HANDLE handle, int offset, int len,
  738. const void * buf, const char * addr, int addrlen)
  739. {
  740. if (!IS_HANDLE_TYPE(handle, udpsrv))
  741. return -PAL_ERROR_NOTCONNECTION;
  742. if (handle->sock.fd == PAL_IDX_POISON)
  743. return -PAL_ERROR_BADHANDLE;
  744. if (strpartcmp_static(addr, "udp:"))
  745. return -PAL_ERROR_INVAL;
  746. addr += static_strlen("udp:");
  747. addrlen -= static_strlen("udp:");
  748. char * addrbuf = __alloca(addrlen + 1);
  749. memcpy(addrbuf, addr, addrlen + 1);
  750. struct sockaddr conn_addr;
  751. int conn_addrlen;
  752. int ret = inet_parse_uri(&addrbuf, &conn_addr, &conn_addrlen);
  753. if (ret < 0)
  754. return ret;
  755. struct msghdr hdr;
  756. struct iovec iov;
  757. iov.iov_base = (void *) buf;
  758. iov.iov_len = len;
  759. hdr.msg_name = &conn_addr;
  760. hdr.msg_namelen = conn_addrlen;
  761. hdr.msg_iov = &iov;
  762. hdr.msg_iovlen = 1;
  763. hdr.msg_control = NULL;
  764. hdr.msg_controllen = 0;
  765. hdr.msg_flags = 0;
  766. int bytes = INLINE_SYSCALL(sendmsg, 3, handle->sock.fd, &hdr, MSG_NOSIGNAL);
  767. if (IS_ERR(bytes))
  768. switch(ERRNO(bytes)) {
  769. case ECONNRESET:
  770. case EPIPE:
  771. return -PAL_ERROR_CONNFAILED;
  772. case EAGAIN:
  773. handle->hdr.flags &= ~WRITABLE(0);
  774. default:
  775. return unix_to_pal_error(ERRNO(bytes));
  776. }
  777. if (bytes == len)
  778. handle->hdr.flags |= WRITABLE(0);
  779. else
  780. handle->hdr.flags &= ~WRITABLE(0);
  781. return bytes;
  782. }
  783. static int socket_delete (PAL_HANDLE handle, int access)
  784. {
  785. if (handle->sock.fd == PAL_IDX_POISON)
  786. return 0;
  787. if (!IS_HANDLE_TYPE(handle, tcp) && access)
  788. return -PAL_ERROR_INVAL;
  789. if (IS_HANDLE_TYPE(handle, tcp) || IS_HANDLE_TYPE(handle, tcpsrv)) {
  790. int shutdown;
  791. switch (access) {
  792. case 0:
  793. shutdown = SHUT_RDWR;
  794. break;
  795. case PAL_DELETE_RD:
  796. shutdown = SHUT_RD;
  797. break;
  798. case PAL_DELETE_WR:
  799. shutdown = SHUT_WR;
  800. break;
  801. default:
  802. return -PAL_ERROR_INVAL;
  803. }
  804. INLINE_SYSCALL(shutdown, 2, handle->sock.fd, shutdown);
  805. }
  806. return 0;
  807. }
  808. static int socket_close (PAL_HANDLE handle)
  809. {
  810. if (handle->sock.fd != PAL_IDX_POISON) {
  811. INLINE_SYSCALL(close, 1, handle->sock.fd);
  812. handle->sock.fd = PAL_IDX_POISON;
  813. }
  814. if (handle->sock.bind)
  815. handle->sock.bind = NULL;
  816. if (handle->sock.conn)
  817. handle->sock.conn = NULL;
  818. return 0;
  819. }
  820. static int socket_attrquerybyhdl (PAL_HANDLE handle, PAL_STREAM_ATTR * attr)
  821. {
  822. if (handle->sock.fd == PAL_IDX_POISON)
  823. return -PAL_ERROR_BADHANDLE;
  824. int fd = handle->sock.fd, ret, val;
  825. if (handle->sock.conn) {
  826. /* try use ioctl FIONEAD to get the size of socket */
  827. ret = INLINE_SYSCALL(ioctl, 3, fd, FIONREAD, &val);
  828. if (IS_ERR(ret))
  829. return unix_to_pal_error(ERRNO(ret));
  830. attr->pending_size = val;
  831. attr->readable = !!attr->pending_size > 0;
  832. } else {
  833. attr->readable = !attr->disconnected;
  834. }
  835. attr->handle_type = handle->hdr.type;
  836. attr->disconnected = handle->hdr.flags & ERROR(0);
  837. attr->nonblocking = handle->sock.nonblocking;
  838. attr->writable = handle->hdr.flags & WRITABLE(0);
  839. attr->socket.linger = handle->sock.linger;
  840. attr->socket.receivebuf = handle->sock.receivebuf;
  841. attr->socket.sendbuf = handle->sock.sendbuf;
  842. attr->socket.receivetimeout = handle->sock.receivetimeout;
  843. attr->socket.sendtimeout = handle->sock.sendtimeout;
  844. attr->socket.tcp_cork = handle->sock.tcp_cork;
  845. attr->socket.tcp_keepalive = handle->sock.tcp_keepalive;
  846. attr->socket.tcp_nodelay = handle->sock.tcp_nodelay;
  847. return 0;
  848. }
  849. static int socket_attrsetbyhdl (PAL_HANDLE handle, PAL_STREAM_ATTR * attr)
  850. {
  851. if (handle->sock.fd == PAL_IDX_POISON)
  852. return -PAL_ERROR_BADHANDLE;
  853. int fd = handle->sock.fd, ret, val;
  854. if (attr->nonblocking != handle->sock.nonblocking) {
  855. ret = INLINE_SYSCALL(fcntl, 3, fd, F_SETFL,
  856. attr->nonblocking ? O_NONBLOCK : 0);
  857. if (IS_ERR(ret))
  858. return unix_to_pal_error(ERRNO(ret));
  859. handle->sock.nonblocking = attr->nonblocking;
  860. }
  861. if (IS_HANDLE_TYPE(handle, tcpsrv)) {
  862. struct __kernel_linger {
  863. int l_onoff;
  864. int l_linger;
  865. };
  866. if (attr->socket.linger != handle->sock.linger) {
  867. struct __kernel_linger l;
  868. l.l_onoff = attr->socket.linger ? 1 : 0;
  869. l.l_linger = attr->socket.linger;
  870. ret = INLINE_SYSCALL(setsockopt, 5, fd, SOL_SOCKET, SO_LINGER,
  871. &l, sizeof(struct __kernel_linger));
  872. if (IS_ERR(ret))
  873. return unix_to_pal_error(ERRNO(ret));
  874. handle->sock.linger = attr->socket.linger;
  875. }
  876. if (attr->socket.receivebuf != handle->sock.receivebuf) {
  877. int val = attr->socket.receivebuf;
  878. ret = INLINE_SYSCALL(setsockopt, 5, fd, SOL_SOCKET, SO_RCVBUF,
  879. &val, sizeof(int));
  880. if (IS_ERR(ret))
  881. return unix_to_pal_error(ERRNO(ret));
  882. handle->sock.receivebuf = attr->socket.receivebuf;
  883. }
  884. if (attr->socket.sendbuf != handle->sock.sendbuf) {
  885. int val = attr->socket.sendbuf;
  886. ret = INLINE_SYSCALL(setsockopt, 5, fd, SOL_SOCKET, SO_SNDBUF,
  887. &val, sizeof(int));
  888. if (IS_ERR(ret))
  889. return unix_to_pal_error(ERRNO(ret));
  890. handle->sock.sendbuf = attr->socket.sendbuf;
  891. }
  892. if (attr->socket.receivetimeout != handle->sock.receivetimeout) {
  893. int val = attr->socket.receivetimeout;
  894. ret = INLINE_SYSCALL(setsockopt, 5, fd, SOL_SOCKET, SO_RCVTIMEO,
  895. &val, sizeof(int));
  896. if (IS_ERR(ret))
  897. return unix_to_pal_error(ERRNO(ret));
  898. handle->sock.receivetimeout = attr->socket.receivetimeout;
  899. }
  900. if (attr->socket.sendtimeout != handle->sock.sendtimeout) {
  901. int val = attr->socket.sendtimeout;
  902. ret = INLINE_SYSCALL(setsockopt, 5, fd, SOL_SOCKET, SO_SNDTIMEO,
  903. &val, sizeof(int));
  904. if (IS_ERR(ret))
  905. return unix_to_pal_error(ERRNO(ret));
  906. handle->sock.sendtimeout = attr->socket.sendtimeout;
  907. }
  908. }
  909. if (IS_HANDLE_TYPE(handle, tcp) || IS_HANDLE_TYPE(handle, tcpsrv)) {
  910. if (attr->socket.tcp_cork != handle->sock.tcp_cork) {
  911. val = attr->socket.tcp_cork ? 1 : 0;
  912. ret = INLINE_SYSCALL(setsockopt, 5, fd, SOL_TCP, TCP_CORK,
  913. &val, sizeof(int));
  914. if (IS_ERR(ret))
  915. return unix_to_pal_error(ERRNO(ret));
  916. handle->sock.tcp_cork = attr->socket.tcp_cork;
  917. }
  918. if (attr->socket.tcp_keepalive != handle->sock.tcp_keepalive) {
  919. val = attr->socket.tcp_keepalive ? 1 : 0;
  920. ret = INLINE_SYSCALL(setsockopt, 5, fd, SOL_SOCKET, SO_KEEPALIVE,
  921. &val, sizeof(int));
  922. if (IS_ERR(ret))
  923. return unix_to_pal_error(ERRNO(ret));
  924. handle->sock.tcp_keepalive = attr->socket.tcp_keepalive;
  925. }
  926. if (attr->socket.tcp_nodelay != handle->sock.tcp_nodelay) {
  927. val = attr->socket.tcp_nodelay ? 1 : 0;
  928. ret = INLINE_SYSCALL(setsockopt, 5, fd, SOL_TCP, TCP_NODELAY,
  929. &val, sizeof(int));
  930. if (IS_ERR(ret))
  931. return unix_to_pal_error(ERRNO(ret));
  932. handle->sock.tcp_nodelay = attr->socket.tcp_nodelay;
  933. }
  934. }
  935. return 0;
  936. }
  937. static int socket_getname (PAL_HANDLE handle, char * buffer, int count)
  938. {
  939. int old_count = count;
  940. int ret;
  941. const char * prefix = NULL;
  942. int prefix_len = 0;
  943. struct sockaddr * bind_addr = NULL, * dest_addr = NULL;
  944. switch (PAL_GET_TYPE(handle)) {
  945. case pal_type_tcpsrv:
  946. prefix_len = 7;
  947. prefix = "tcp.srv";
  948. bind_addr = handle->sock.bind;
  949. break;
  950. case pal_type_tcp:
  951. prefix_len = 3;
  952. prefix = "tcp";
  953. bind_addr = handle->sock.bind;
  954. dest_addr = handle->sock.conn;
  955. break;
  956. case pal_type_udpsrv:
  957. prefix_len = 7;
  958. prefix = "udp.srv";
  959. bind_addr = handle->sock.bind;
  960. break;
  961. case pal_type_udp:
  962. prefix_len = 3;
  963. prefix = "udp";
  964. bind_addr = handle->sock.bind;
  965. dest_addr = handle->sock.conn;
  966. break;
  967. default:
  968. return -PAL_ERROR_INVAL;
  969. }
  970. if (prefix_len >= count)
  971. return -PAL_ERROR_OVERFLOW;
  972. memcpy(buffer, prefix, prefix_len + 1);
  973. buffer += prefix_len;
  974. count -= prefix_len;
  975. for (int i = 0 ; i < 2 ; i++) {
  976. struct sockaddr * addr = i ? dest_addr : bind_addr;
  977. if (addr) {
  978. if (count <= 1)
  979. return -PAL_ERROR_OVERFLOW;
  980. buffer[0] = ':';
  981. buffer[1] = 0;
  982. buffer++;
  983. count--;
  984. if ((ret = inet_create_uri(buffer, count, addr,
  985. addr_size(addr))) < 0)
  986. return ret;
  987. buffer += ret;
  988. count -= ret;
  989. }
  990. }
  991. return old_count - count;
  992. }
  993. struct handle_ops tcp_ops = {
  994. .getname = &socket_getname,
  995. .open = &tcp_open,
  996. .waitforclient = &tcp_accept,
  997. .read = &tcp_read,
  998. .write = &tcp_write,
  999. .delete = &socket_delete,
  1000. .close = &socket_close,
  1001. .attrquerybyhdl = &socket_attrquerybyhdl,
  1002. .attrsetbyhdl = &socket_attrsetbyhdl,
  1003. };
  1004. struct handle_ops udp_ops = {
  1005. .getname = &socket_getname,
  1006. .open = &udp_open,
  1007. .read = &udp_receive,
  1008. .write = &udp_send,
  1009. .delete = &socket_delete,
  1010. .close = &socket_close,
  1011. .attrquerybyhdl = &socket_attrquerybyhdl,
  1012. .attrsetbyhdl = &socket_attrsetbyhdl,
  1013. };
  1014. struct handle_ops udpsrv_ops = {
  1015. .getname = &socket_getname,
  1016. .open = &udp_open,
  1017. .readbyaddr = &udp_receivebyaddr,
  1018. .writebyaddr = &udp_sendbyaddr,
  1019. .delete = &socket_delete,
  1020. .close = &socket_close,
  1021. .attrquerybyhdl = &socket_attrquerybyhdl,
  1022. .attrsetbyhdl = &socket_attrsetbyhdl,
  1023. };
  1024. static int mcast_s (PAL_HANDLE handle, int port)
  1025. {
  1026. handle->mcast.srv = PAL_IDX_POISON;
  1027. int ret = 0;
  1028. int fd = INLINE_SYSCALL(socket, 3, AF_INET, SOCK_DGRAM, 0);
  1029. if (IS_ERR(fd))
  1030. return -PAL_ERROR_DENIED;
  1031. struct in_addr local;
  1032. local.s_addr = INADDR_ANY;
  1033. ret = INLINE_SYSCALL(setsockopt, 5, fd, IPPROTO_IP, IP_MULTICAST_IF,
  1034. &local, sizeof(local));
  1035. if (IS_ERR(ret))
  1036. return -PAL_ERROR_DENIED;
  1037. handle->hdr.flags |= WFD(1)|WRITABLE(1);
  1038. handle->mcast.srv = fd;
  1039. return 0;
  1040. }
  1041. static int mcast_c (PAL_HANDLE handle, int port)
  1042. {
  1043. handle->mcast.cli = PAL_IDX_POISON;
  1044. int ret = 0;
  1045. int fd = INLINE_SYSCALL(socket, 3, AF_INET, SOCK_DGRAM, 0);
  1046. if (IS_ERR(fd))
  1047. return -PAL_ERROR_DENIED;
  1048. int reuse = 1;
  1049. INLINE_SYSCALL(setsockopt, 5, fd, SOL_SOCKET, SO_REUSEPORT,
  1050. &reuse, sizeof(reuse));
  1051. struct sockaddr_in addr;
  1052. addr.sin_family = AF_INET;
  1053. addr.sin_addr.s_addr = htonl(INADDR_ANY);
  1054. addr.sin_port = htons(port);
  1055. ret = INLINE_SYSCALL(bind, 3, fd, &addr, sizeof(addr));
  1056. if (IS_ERR(ret))
  1057. return -PAL_ERROR_DENIED;
  1058. struct in_addr local;
  1059. local.s_addr = INADDR_ANY;
  1060. ret = INLINE_SYSCALL(setsockopt, 5, fd, IPPROTO_IP, IP_MULTICAST_IF,
  1061. &local, sizeof(local));
  1062. if (IS_ERR(ret))
  1063. return -PAL_ERROR_DENIED;
  1064. struct ip_mreq group;
  1065. inet_pton4(GRAPHENE_MCAST_GROUP, sizeof(GRAPHENE_MCAST_GROUP) - 1,
  1066. &group.imr_multiaddr.s_addr);
  1067. group.imr_interface.s_addr = htonl(INADDR_ANY);
  1068. ret = INLINE_SYSCALL(setsockopt, 5, fd, IPPROTO_IP, IP_ADD_MEMBERSHIP,
  1069. &group, sizeof(group));
  1070. if (IS_ERR(ret))
  1071. return -PAL_ERROR_DENIED;
  1072. handle->mcast.cli = fd;
  1073. handle->mcast.nonblocking = PAL_FALSE;
  1074. return 0;
  1075. }
  1076. PAL_HANDLE _DkBroadcastStreamOpen (void)
  1077. {
  1078. if (!pal_sec.mcast_port) {
  1079. unsigned short mcast_port;
  1080. _DkFastRandomBitsRead(&mcast_port, sizeof(unsigned short));
  1081. if (mcast_port < 1024)
  1082. mcast_port += 1024;
  1083. pal_sec.mcast_port = mcast_port > 1024 ? mcast_port : mcast_port + 1204;
  1084. }
  1085. PAL_HANDLE hdl = malloc(HANDLE_SIZE(mcast));
  1086. SET_HANDLE_TYPE(hdl, mcast);
  1087. mcast_s(hdl, pal_sec.mcast_port);
  1088. mcast_c(hdl, pal_sec.mcast_port);
  1089. hdl->mcast.port = pal_sec.mcast_port;
  1090. return hdl;
  1091. }
  1092. static int mcast_send (PAL_HANDLE handle, int offset, int size,
  1093. const void * buf)
  1094. {
  1095. if (handle->mcast.srv == PAL_IDX_POISON)
  1096. return -PAL_ERROR_BADHANDLE;
  1097. struct sockaddr_in addr;
  1098. addr.sin_family = AF_INET;
  1099. inet_pton4(GRAPHENE_MCAST_GROUP, sizeof(GRAPHENE_MCAST_GROUP) - 1,
  1100. &addr.sin_addr.s_addr);
  1101. addr.sin_port = htons(handle->mcast.port);
  1102. struct msghdr hdr;
  1103. struct iovec iov;
  1104. iov.iov_base = (void *) buf;
  1105. iov.iov_len = size;
  1106. hdr.msg_name = &addr;
  1107. hdr.msg_namelen = sizeof(addr);
  1108. hdr.msg_iov = &iov;
  1109. hdr.msg_iovlen = 1;
  1110. hdr.msg_control = NULL;
  1111. hdr.msg_controllen = 0;
  1112. hdr.msg_flags = 0;
  1113. int bytes = INLINE_SYSCALL(sendmsg, 3, handle->mcast.srv, &hdr,
  1114. MSG_NOSIGNAL);
  1115. if (IS_ERR(bytes))
  1116. switch(ERRNO(bytes)) {
  1117. case ECONNRESET:
  1118. case EPIPE:
  1119. return -PAL_ERROR_CONNFAILED;
  1120. case EAGAIN:
  1121. handle->hdr.flags &= ~WRITABLE(1);
  1122. /* fallthrough */
  1123. default:
  1124. return unix_to_pal_error(ERRNO(bytes));
  1125. }
  1126. if (bytes == size)
  1127. handle->hdr.flags |= WRITABLE(1);
  1128. else
  1129. handle->hdr.flags &= ~WRITABLE(1);
  1130. return bytes;
  1131. }
  1132. static int mcast_receive (PAL_HANDLE handle, int offset, int size, void * buf)
  1133. {
  1134. if (handle->mcast.cli == PAL_IDX_POISON)
  1135. return -PAL_ERROR_BADHANDLE;
  1136. struct msghdr hdr;
  1137. struct iovec iov;
  1138. iov.iov_base = buf;
  1139. iov.iov_len = size;
  1140. hdr.msg_name = NULL;
  1141. hdr.msg_namelen = 0;
  1142. hdr.msg_iov = &iov;
  1143. hdr.msg_iovlen = 1;
  1144. hdr.msg_control = NULL;
  1145. hdr.msg_controllen = 0;
  1146. hdr.msg_flags = 0;
  1147. int bytes = INLINE_SYSCALL(recvmsg, 3, handle->mcast.cli, &hdr, 0);
  1148. if (IS_ERR(bytes))
  1149. return -PAL_ERROR_DENIED;
  1150. return bytes;
  1151. }
  1152. static int mcast_close (PAL_HANDLE handle)
  1153. {
  1154. if (handle->mcast.srv != PAL_IDX_POISON) {
  1155. INLINE_SYSCALL(close, 1, handle->mcast.srv);
  1156. handle->mcast.srv = PAL_IDX_POISON;
  1157. }
  1158. if (handle->mcast.cli != PAL_IDX_POISON) {
  1159. INLINE_SYSCALL(close, 1, handle->mcast.cli);
  1160. handle->mcast.cli = PAL_IDX_POISON;
  1161. }
  1162. return 0;
  1163. }
  1164. static int mcast_attrquerybyhdl (PAL_HANDLE handle, PAL_STREAM_ATTR * attr)
  1165. {
  1166. int ret, val;
  1167. if (handle->mcast.cli == PAL_IDX_POISON)
  1168. return -PAL_ERROR_BADHANDLE;
  1169. ret = INLINE_SYSCALL(ioctl, 3, handle->mcast.cli, FIONREAD, &val);
  1170. if (IS_ERR(ret))
  1171. return unix_to_pal_error(ERRNO(ret));
  1172. attr->handle_type = pal_type_mcast;
  1173. attr->disconnected = handle->hdr.flags & (ERROR(0)|ERROR(1));
  1174. attr->nonblocking = handle->mcast.nonblocking;
  1175. attr->readable = !!val;
  1176. attr->writable = handle->hdr.flags & WRITABLE(1);
  1177. attr->runnable = PAL_FALSE;
  1178. attr->pending_size = val;
  1179. return 0;
  1180. }
  1181. static int mcast_attrsetbyhdl (PAL_HANDLE handle, PAL_STREAM_ATTR * attr)
  1182. {
  1183. if (handle->mcast.cli == PAL_IDX_POISON)
  1184. return -PAL_ERROR_BADHANDLE;
  1185. int ret;
  1186. PAL_BOL * nonblocking = &handle->mcast.nonblocking;
  1187. if (attr->nonblocking != *nonblocking) {
  1188. ret = INLINE_SYSCALL(fcntl, 3, handle->mcast.cli, F_SETFL,
  1189. *nonblocking ? O_NONBLOCK : 0);
  1190. if (IS_ERR(ret))
  1191. return unix_to_pal_error(ERRNO(ret));
  1192. *nonblocking = attr->nonblocking;
  1193. }
  1194. return 0;
  1195. }
  1196. struct handle_ops mcast_ops = {
  1197. .write = &mcast_send,
  1198. .read = &mcast_receive,
  1199. .close = &mcast_close,
  1200. .attrquerybyhdl = &mcast_attrquerybyhdl,
  1201. .attrsetbyhdl = &mcast_attrsetbyhdl,
  1202. };