shim_socket.c 47 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715
  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 OSCAR lab, 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 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 General Public License for more details.
  13. You should have received a copy of the GNU General Public License
  14. along with this program. If not, see <http://www.gnu.org/licenses/>. */
  15. /*
  16. * shim_socket.c
  17. *
  18. * Implementation of system call "socket", "bind", "accept4", "listen",
  19. * "connect", "sendto", "recvfrom", "sendmsg", "recvmsg" and "shutdown" and
  20. * "getsockname", "getpeername".
  21. */
  22. #include <shim_internal.h>
  23. #include <shim_table.h>
  24. #include <shim_handle.h>
  25. #include <shim_fs.h>
  26. #include <shim_utils.h>
  27. #include <shim_checkpoint.h>
  28. #include <pal.h>
  29. #include <pal_error.h>
  30. #include <errno.h>
  31. #include <linux/fcntl.h>
  32. #include <linux/in.h>
  33. #include <linux/in6.h>
  34. #include <asm/socket.h>
  35. /*
  36. * User-settable options (used with setsockopt).
  37. */
  38. #define TCP_NODELAY 1 /* Don't delay send to coalesce packets */
  39. #define TCP_MAXSEG 2 /* Set maximum segment size */
  40. #define TCP_CORK 3 /* Control sending of partial frames */
  41. #define TCP_KEEPIDLE 4 /* Start keeplives after this period */
  42. #define TCP_KEEPINTVL 5 /* Interval between keepalives */
  43. #define TCP_KEEPCNT 6 /* Number of keepalives before death */
  44. #define TCP_SYNCNT 7 /* Number of SYN retransmits */
  45. #define TCP_LINGER2 8 /* Life time of orphaned FIN-WAIT-2 state */
  46. #define TCP_DEFER_ACCEPT 9 /* Wake up listener only when data arrive */
  47. #define TCP_WINDOW_CLAMP 10 /* Bound advertised window */
  48. #define TCP_INFO 11 /* Information about this connection. */
  49. #define TCP_QUICKACK 12 /* Bock/reenable quick ACKs. */
  50. #define TCP_CONGESTION 13 /* Congestion control algorithm. */
  51. #define TCP_MD5SIG 14 /* TCP MD5 Signature (RFC2385) */
  52. #define AF_UNSPEC 0
  53. #define SOCK_URI_SIZE 108
  54. static int rebase_on_lo __attribute_migratable = -1;
  55. static int init_port_rebase (void)
  56. {
  57. if (rebase_on_lo != -1)
  58. return 0;
  59. char cfg[CONFIG_MAX];
  60. int rebase = 0;
  61. if (!root_config ||
  62. get_config(root_config, "net.port.rebase_on_lo", cfg, CONFIG_MAX) <= 0) {
  63. rebase_on_lo = 0;
  64. return 0;
  65. }
  66. for (const char * p = cfg ; *p ; p++) {
  67. if (*p < '0' || *p > '9' || rebase > 32767) {
  68. rebase_on_lo = 0;
  69. return 0;
  70. }
  71. rebase = rebase * 10 + (*p - '0');
  72. }
  73. rebase_on_lo = rebase;
  74. return 0;
  75. }
  76. static int __process_pending_options (struct shim_handle * hdl);
  77. int shim_do_socket (int family, int type, int protocol)
  78. {
  79. struct shim_handle * hdl = get_new_handle();
  80. if (!hdl)
  81. return -ENOMEM;
  82. struct shim_sock_handle * sock = &hdl->info.sock;
  83. hdl->type = TYPE_SOCK;
  84. set_handle_fs(hdl, &socket_builtin_fs);
  85. hdl->flags = type & SOCK_NONBLOCK ? O_NONBLOCK : 0;
  86. hdl->acc_mode = 0;
  87. sock->domain = family;
  88. sock->sock_type = type & ~(SOCK_NONBLOCK|SOCK_CLOEXEC);
  89. sock->protocol = protocol;
  90. int ret = -ENOSYS;
  91. switch (sock->domain) {
  92. case AF_UNIX: //Local communication
  93. case AF_INET: //IPv4 Internet protocols ip(7)
  94. case AF_INET6: //IPv6 Internet protocols
  95. break;
  96. default:
  97. debug("shim_socket: unknown socket domain %d\n",
  98. sock->domain);
  99. goto err;
  100. }
  101. switch (sock->sock_type) {
  102. case SOCK_STREAM: //TCP
  103. break;
  104. case SOCK_DGRAM: //UDP
  105. hdl->acc_mode = MAY_READ|MAY_WRITE;
  106. break;
  107. default:
  108. debug("shim_socket: unknown socket type %d\n",
  109. sock->sock_type);
  110. goto err;
  111. }
  112. sock->sock_state = SOCK_CREATED;
  113. ret = set_new_fd_handle(hdl, type & SOCK_CLOEXEC ? FD_CLOEXEC : 0, NULL);
  114. err:
  115. put_handle(hdl);
  116. return ret;
  117. }
  118. static int unix_create_uri (char * uri, int count, enum shim_sock_state state,
  119. unsigned int pipeid)
  120. {
  121. int bytes = 0;
  122. switch (state) {
  123. case SOCK_CREATED:
  124. case SOCK_BOUNDCONNECTED:
  125. case SOCK_SHUTDOWN:
  126. return -ENOTCONN;
  127. case SOCK_BOUND:
  128. case SOCK_LISTENED:
  129. case SOCK_ACCEPTED:
  130. bytes = snprintf(uri, count, "pipe.srv:%u", pipeid);
  131. break;
  132. case SOCK_CONNECTED:
  133. bytes = snprintf(uri, count, "pipe:%u", pipeid);
  134. break;
  135. default:
  136. return -ENOTCONN;
  137. }
  138. return bytes == count ? -ENAMETOOLONG : bytes;
  139. }
  140. static void inet_rebase_port (bool reverse, int domain, struct addr_inet * addr,
  141. bool local)
  142. {
  143. init_port_rebase();
  144. if (rebase_on_lo) {
  145. if (domain == AF_INET) {
  146. unsigned char * ad = (unsigned char *) &addr->addr.v4.s_addr;
  147. if (!local && memcmp(ad, "\177\0\0\1", 4))
  148. return;
  149. }
  150. if (domain == AF_INET6) {
  151. unsigned short * ad = (void *) &addr->addr.v6.s6_addr;
  152. if (!local && memcmp(ad, "\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\1", 16))
  153. return;
  154. }
  155. }
  156. if (reverse)
  157. addr->port = addr->ext_port - rebase_on_lo;
  158. else
  159. addr->ext_port = addr->port + rebase_on_lo;
  160. }
  161. static int inet_translate_addr (int domain, char * uri, int count,
  162. struct addr_inet * addr)
  163. {
  164. if (domain == AF_INET) {
  165. unsigned char * ad = (unsigned char *) &addr->addr.v4.s_addr;
  166. int bytes = snprintf(uri, count, "%u.%u.%u.%u:%u",
  167. ad[0], ad[1], ad[2], ad[3], addr->ext_port);
  168. return bytes == count ? -ENAMETOOLONG : bytes;
  169. }
  170. if (domain == AF_INET6) {
  171. unsigned short * ad = (void *) &addr->addr.v6.s6_addr;
  172. int bytes = snprintf(uri, count,
  173. "[%04x:%04x:%x:%04x:%04x:%04x:%04x:%04x]:%u",
  174. ad[0], ad[1], ad[2], ad[3],
  175. ad[4], ad[5], ad[6], ad[7], addr->ext_port);
  176. return bytes == count ? -ENAMETOOLONG : bytes;
  177. }
  178. return -EPROTONOSUPPORT;
  179. }
  180. static int inet_create_uri (int domain, char * uri, int count, int sock_type,
  181. enum shim_sock_state state,
  182. struct addr_inet * bind, struct addr_inet * conn)
  183. {
  184. int bytes = 0, ret;
  185. if (sock_type == SOCK_STREAM) {
  186. switch (state) {
  187. case SOCK_CREATED:
  188. case SOCK_SHUTDOWN:
  189. return -ENOTCONN;
  190. case SOCK_BOUND:
  191. case SOCK_LISTENED:
  192. if (count < 9)
  193. return -ENAMETOOLONG;
  194. memcpy(uri, "tcp.srv:", 9);
  195. ret = inet_translate_addr(domain, uri + 8, count - 8, bind);
  196. return ret < 0 ? ret : ret + 8;
  197. case SOCK_BOUNDCONNECTED:
  198. if (count < 5)
  199. return -ENAMETOOLONG;
  200. memcpy(uri, "tcp:", 5);
  201. bytes = 4;
  202. ret = inet_translate_addr(domain, uri + bytes, count - bytes,
  203. bind);
  204. if (ret < 0)
  205. return ret;
  206. uri[bytes + ret] = ':';
  207. bytes += ret + 1;
  208. ret = inet_translate_addr(domain, uri + bytes, count - bytes,
  209. conn);
  210. return ret < 0 ? ret : ret + bytes;
  211. case SOCK_CONNECTED:
  212. case SOCK_ACCEPTED:
  213. if (count < 5)
  214. return -ENAMETOOLONG;
  215. memcpy(uri, "tcp:", 5);
  216. ret = inet_translate_addr(domain, uri + 4, count - 4, conn);
  217. return ret < 0 ? ret : ret + 4;
  218. }
  219. }
  220. if (sock_type == SOCK_DGRAM) {
  221. switch (state) {
  222. case SOCK_CREATED:
  223. case SOCK_SHUTDOWN:
  224. return -ENOTCONN;
  225. case SOCK_LISTENED:
  226. case SOCK_ACCEPTED:
  227. return -EOPNOTSUPP;
  228. case SOCK_BOUNDCONNECTED:
  229. if (count < 9)
  230. return -ENAMETOOLONG;
  231. memcpy(uri, "tcp.srv:", 9);
  232. bytes = 8;
  233. ret = inet_translate_addr(domain, uri + bytes, count - bytes,
  234. bind);
  235. if (ret < 0)
  236. return ret;
  237. uri[bytes + ret] = ':';
  238. bytes += ret + 1;
  239. ret = inet_translate_addr(domain, uri + bytes, count - bytes,
  240. conn);
  241. return ret < 0 ? ret : ret + bytes;
  242. case SOCK_BOUND:
  243. if (count < 9)
  244. return -ENAMETOOLONG;
  245. memcpy(uri, "udp.srv:", 9);
  246. ret = inet_translate_addr(domain, uri + 8, count - 8, bind);
  247. return ret < 0 ? ret : ret + 9;
  248. case SOCK_CONNECTED:
  249. if (count < 5)
  250. return -ENAMETOOLONG;
  251. memcpy(uri, "udp:", 5);
  252. ret = inet_translate_addr(domain, uri + 4, count - 4, conn);
  253. return ret < 0 ? ret : ret + 4;
  254. }
  255. }
  256. return -EPROTONOSUPPORT;
  257. }
  258. static inline void unix_copy_addr (struct sockaddr * saddr,
  259. struct shim_dentry * dent)
  260. {
  261. struct sockaddr_un * un = (struct sockaddr_un *) saddr;
  262. un->sun_family = AF_UNIX;
  263. int size;
  264. const char * path = dentry_get_path(dent, true, &size);
  265. memcpy(un->sun_path, path, size + 1);
  266. }
  267. static bool inet_check_addr (int domain, struct sockaddr * addr, int addrlen)
  268. {
  269. if (domain == AF_INET) {
  270. if (addr->sa_family != AF_INET ||
  271. addrlen != sizeof(struct sockaddr_in))
  272. return false;
  273. return true;
  274. }
  275. if (domain == AF_INET6) {
  276. if (addr->sa_family != AF_INET && addr->sa_family != AF_INET6)
  277. return false;
  278. if (addrlen != ((addr->sa_family == AF_INET) ?
  279. sizeof(struct sockaddr_in) :
  280. sizeof(struct sockaddr_in6)))
  281. return false;
  282. return true;
  283. }
  284. return false;
  285. }
  286. static int inet_copy_addr (int domain, struct sockaddr * saddr,
  287. const struct addr_inet * addr)
  288. {
  289. if (domain == AF_INET) {
  290. struct sockaddr_in * in = (struct sockaddr_in *) saddr;
  291. in->sin_family = AF_INET;
  292. in->sin_port = __htons(addr->port);
  293. in->sin_addr = addr->addr.v4;
  294. return sizeof(struct sockaddr_in);
  295. }
  296. if (domain == AF_INET6) {
  297. struct sockaddr_in6 * in6 = (struct sockaddr_in6 *) saddr;
  298. in6->sin6_family = AF_INET6;
  299. in6->sin6_port = __htons(addr->port);
  300. in6->sin6_addr = addr->addr.v6;
  301. return sizeof(struct sockaddr_in6);
  302. }
  303. return sizeof(struct sockaddr);
  304. }
  305. static void inet_save_addr (int domain, struct addr_inet * addr,
  306. const struct sockaddr * saddr)
  307. {
  308. if (domain == AF_INET) {
  309. const struct sockaddr_in * in = (const struct sockaddr_in *) saddr;
  310. addr->port = __ntohs(in->sin_port);
  311. addr->addr.v4 = in->sin_addr;
  312. return;
  313. }
  314. if (domain == AF_INET6) {
  315. if (saddr->sa_family == AF_INET) {
  316. const struct sockaddr_in * in = (const struct sockaddr_in *) saddr;
  317. addr->port = __ntohs(in->sin_port);
  318. ((uint32_t *) &addr->addr.v6.s6_addr)[0] = 0;
  319. ((uint32_t *) &addr->addr.v6.s6_addr)[1] = 0;
  320. ((uint32_t *) &addr->addr.v6.s6_addr)[2] = 0xffff0000;
  321. ((uint32_t *) &addr->addr.v6.s6_addr)[3] = in->sin_addr.s_addr;
  322. } else {
  323. const struct sockaddr_in6 * in6 = (const struct sockaddr_in6 *) saddr;
  324. addr->port = __ntohs(in6->sin6_port);
  325. addr->addr.v6 = in6->sin6_addr;
  326. }
  327. return;
  328. }
  329. }
  330. static inline bool inet_comp_addr (int domain, const struct addr_inet * addr,
  331. const struct sockaddr * saddr)
  332. {
  333. if (domain == AF_INET) {
  334. const struct sockaddr_in * in = (const struct sockaddr_in *) saddr;
  335. return addr->port == __ntohs(in->sin_port) &&
  336. !memcmp(&addr->addr.v4, &in->sin_addr,
  337. sizeof(struct in_addr));
  338. }
  339. if (domain == AF_INET6) {
  340. const struct sockaddr_in6 * in6 = (const struct sockaddr_in6 *) saddr;
  341. return addr->port == __ntohs(in6->sin6_port) &&
  342. !memcmp(&addr->addr.v6, &in6->sin6_addr,
  343. sizeof(struct in6_addr));
  344. }
  345. return false;
  346. }
  347. static int create_socket_uri (struct shim_handle * hdl)
  348. {
  349. struct shim_sock_handle * sock = &hdl->info.sock;
  350. if (sock->domain == AF_UNIX) {
  351. char uri_buf[32];
  352. int bytes = unix_create_uri(uri_buf, 32, sock->sock_state,
  353. sock->addr.un.pipeid);
  354. if (bytes < 0)
  355. return bytes;
  356. qstrsetstr(&hdl->uri, uri_buf, bytes);
  357. return 0;
  358. }
  359. if (sock->domain == AF_INET || sock->domain == AF_INET6) {
  360. char uri_buf[SOCK_URI_SIZE];
  361. int bytes = inet_create_uri(sock->domain, uri_buf, SOCK_URI_SIZE,
  362. sock->sock_type, sock->sock_state,
  363. &sock->addr.in.bind, &sock->addr.in.conn);
  364. if (bytes < 0)
  365. return bytes;
  366. qstrsetstr(&hdl->uri, uri_buf, bytes);
  367. return 0;
  368. }
  369. return -EPROTONOSUPPORT;
  370. }
  371. int shim_do_bind (int sockfd, struct sockaddr * addr, socklen_t addrlen)
  372. {
  373. struct shim_handle * hdl = get_fd_handle(sockfd, NULL, NULL);
  374. int ret = -EINVAL;
  375. if (!hdl)
  376. return -EBADF;
  377. if (hdl->type != TYPE_SOCK) {
  378. put_handle(hdl);
  379. return -ENOTSOCK;
  380. }
  381. struct shim_sock_handle * sock = &hdl->info.sock;
  382. lock(hdl->lock);
  383. enum shim_sock_state state = sock->sock_state;
  384. if (state != SOCK_CREATED) {
  385. debug("shim_bind: bind on a bound socket\n");
  386. goto out;
  387. }
  388. if (sock->domain == AF_UNIX) {
  389. if (addrlen != sizeof(struct sockaddr_un))
  390. goto out;
  391. struct sockaddr_un * saddr = (struct sockaddr_un *) addr;
  392. char * spath = saddr->sun_path;
  393. struct shim_dentry * dent = NULL;
  394. if ((ret = path_lookupat(NULL, spath, LOOKUP_CREATE, &dent)) < 0)
  395. goto out;
  396. if (dent->state & DENTRY_VALID &&
  397. !(dent->state & DENTRY_NEGATIVE)) {
  398. ret = -EADDRINUSE;
  399. goto out;
  400. }
  401. struct shim_unix_data * data = malloc(sizeof(struct shim_unix_data));
  402. data->pipeid = dent->rel_path.hash >> 32;
  403. sock->addr.un.pipeid = data->pipeid;
  404. sock->addr.un.data = data;
  405. sock->addr.un.dentry = dent;
  406. } else if (sock->domain == AF_INET || sock->domain == AF_INET6) {
  407. if (!inet_check_addr(sock->domain, addr, addrlen))
  408. goto out;
  409. inet_save_addr(sock->domain, &sock->addr.in.bind, addr);
  410. inet_rebase_port(false, sock->domain, &sock->addr.in.bind, true);
  411. }
  412. sock->sock_state = SOCK_BOUND;
  413. if ((ret = create_socket_uri(hdl)) < 0)
  414. goto out;
  415. PAL_HANDLE pal_hdl = DkStreamOpen(qstrgetstr(&hdl->uri),
  416. 0, 0, 0,
  417. hdl->flags & O_NONBLOCK);
  418. if (!pal_hdl) {
  419. ret = (PAL_NATIVE_ERRNO == PAL_ERROR_STREAMEXIST) ? -EADDRINUSE : -PAL_ERRNO;
  420. debug("bind: invalid handle returned\n");
  421. goto out;
  422. }
  423. if (sock->domain == AF_UNIX) {
  424. struct shim_dentry * dent = sock->addr.un.dentry;
  425. dent->state ^= DENTRY_NEGATIVE;
  426. dent->state |= DENTRY_VALID|DENTRY_RECENTLY;
  427. dent->fs = &socket_builtin_fs;
  428. dent->data = sock->addr.un.data;
  429. }
  430. hdl->pal_handle = pal_hdl;
  431. __process_pending_options(hdl);
  432. ret = 0;
  433. out:
  434. if (ret < 0) {
  435. sock->sock_state = state;
  436. sock->error = -ret;
  437. if (sock->domain == AF_UNIX) {
  438. if (sock->addr.un.dentry)
  439. put_dentry(sock->addr.un.dentry);
  440. if (sock->addr.un.data) {
  441. free(sock->addr.un.data);
  442. sock->addr.un.data = NULL;
  443. }
  444. }
  445. }
  446. unlock(hdl->lock);
  447. put_handle(hdl);
  448. return ret;
  449. }
  450. static int inet_parse_addr (int domain, int type, const char * uri,
  451. struct addr_inet * bind,
  452. struct addr_inet * conn)
  453. {
  454. char * ip_str, * port_str, * next_str;
  455. int ip_len = 0;
  456. if (!(next_str = strchr(uri, ':')))
  457. return -EINVAL;
  458. enum { UDP, UDPSRV, TCP, TCPSRV } prefix;
  459. int n = (next_str++) - uri;
  460. if (!memcmp(uri, "udp", n))
  461. prefix = UDP;
  462. else if (!memcmp(uri, "udp.srv", n))
  463. prefix = UDPSRV;
  464. else if (!memcmp(uri, "tcp", n))
  465. prefix = TCP;
  466. else if (!memcmp(uri, "tcp.srv", n))
  467. prefix = TCPSRV;
  468. else
  469. return -EINVAL;
  470. if ((prefix == UDP || prefix == UDPSRV) && type != SOCK_DGRAM)
  471. return -EINVAL;
  472. if ((prefix == TCP || prefix == TCPSRV) && type != SOCK_STREAM)
  473. return -EINVAL;
  474. for (int round = 0 ; (ip_str = next_str) ; round++) {
  475. if (ip_str[0] == '[') {
  476. ip_str++;
  477. if (domain != AF_INET6)
  478. return -EINVAL;
  479. if (!(port_str = strchr(ip_str, ']')))
  480. return -EINVAL;
  481. ip_len = port_str - ip_str;
  482. port_str++;
  483. if (*port_str != ':')
  484. return -EINVAL;
  485. } else {
  486. if (domain != AF_INET)
  487. return -EINVAL;
  488. if (!(port_str = strchr(ip_str, ':')))
  489. return -EINVAL;
  490. ip_len = port_str - ip_str;
  491. }
  492. port_str++;
  493. next_str = strchr(port_str, ':');
  494. if (next_str)
  495. next_str++;
  496. struct addr_inet * addr = round ? conn : bind;
  497. if (domain == AF_INET) {
  498. inet_pton4(ip_str, ip_len, &addr->addr.v4);
  499. addr->ext_port = atoi(port_str);
  500. }
  501. if (domain == AF_INET6) {
  502. inet_pton6(ip_str, ip_len, &addr->addr.v6);
  503. addr->ext_port = atoi(port_str);
  504. }
  505. }
  506. return 0;
  507. }
  508. struct un_conn {
  509. unsigned int pipeid;
  510. unsigned char path_size;
  511. char path[];
  512. } __attribute__((packed));
  513. int shim_do_listen (int sockfd, int backlog)
  514. {
  515. if (backlog < 0)
  516. return -EINVAL;
  517. struct shim_handle * hdl = get_fd_handle(sockfd, NULL, NULL);
  518. if (!hdl)
  519. return -EBADF;
  520. if (hdl->type != TYPE_SOCK) {
  521. put_handle(hdl);
  522. return -ENOTSOCK;
  523. }
  524. struct shim_sock_handle * sock = &hdl->info.sock;
  525. if (sock->sock_type != SOCK_STREAM) {
  526. debug("shim_listen: not a stream socket\n");
  527. put_handle(hdl);
  528. return -EINVAL;
  529. }
  530. lock(hdl->lock);
  531. enum shim_sock_state state = sock->sock_state;
  532. int ret = -EINVAL;
  533. if (state != SOCK_BOUND) {
  534. debug("shim_listen: listen on unbound socket\n");
  535. goto out;
  536. }
  537. hdl->acc_mode = MAY_READ;
  538. sock->sock_state = SOCK_LISTENED;
  539. ret = 0;
  540. out:
  541. if (ret < 0)
  542. sock->sock_state = state;
  543. unlock(hdl->lock);
  544. put_handle(hdl);
  545. return ret;
  546. }
  547. /* Connect with the TCP socket is always in the client.
  548. *
  549. * With UDP, the connection is make to the socket specific for a
  550. * destination. A process with a connected UDP socket can call
  551. * connect again for that socket for one of two reasons: 1. To
  552. * specify a new IP address and port 2. To unconnect the socket.
  553. */
  554. int shim_do_connect (int sockfd, struct sockaddr * addr, int addrlen)
  555. {
  556. struct shim_handle * hdl = get_fd_handle(sockfd, NULL, NULL);
  557. if (!hdl)
  558. return -EBADF;
  559. if (hdl->type != TYPE_SOCK) {
  560. put_handle(hdl);
  561. return -ENOTSOCK;
  562. }
  563. struct shim_sock_handle * sock = &hdl->info.sock;
  564. lock(hdl->lock);
  565. enum shim_sock_state state = sock->sock_state;
  566. int ret = -EINVAL;
  567. if (state == SOCK_CONNECTED) {
  568. if (addr->sa_family == AF_UNSPEC) {
  569. sock->sock_state = SOCK_CREATED;
  570. if (sock->sock_type == SOCK_STREAM && hdl->pal_handle) {
  571. DkStreamDelete(hdl->pal_handle, 0);
  572. DkObjectClose(hdl->pal_handle);
  573. hdl->pal_handle = NULL;
  574. }
  575. debug("shim_connect: reconnect on a stream socket\n");
  576. ret = 0;
  577. goto out;
  578. }
  579. debug("shim_connect: reconnect on a stream socket\n");
  580. goto out;
  581. }
  582. if (state != SOCK_BOUND &&
  583. state != SOCK_CREATED) {
  584. debug("shim_connect: connect on invalid socket\n");
  585. goto out;
  586. }
  587. if (sock->domain == AF_UNIX) {
  588. if (addrlen != sizeof(struct sockaddr_un))
  589. goto out;
  590. struct sockaddr_un * saddr = (struct sockaddr_un *) addr;
  591. char * spath = saddr->sun_path;
  592. struct shim_dentry * dent;
  593. if ((ret = path_lookupat(NULL, spath, LOOKUP_CREATE, &dent)) < 0)
  594. goto out;
  595. struct shim_unix_data * data = dent->data;
  596. if (!(dent->state & DENTRY_VALID) || dent->state & DENTRY_NEGATIVE) {
  597. data = malloc(sizeof(struct shim_unix_data));
  598. data->pipeid = dent->rel_path.hash >> 32;
  599. } else if (dent->fs != &socket_builtin_fs) {
  600. ret = -ECONNREFUSED;
  601. goto out;
  602. }
  603. sock->addr.un.pipeid = data->pipeid;
  604. sock->addr.un.data = data;
  605. sock->addr.un.dentry = dent;
  606. get_dentry(dent);
  607. }
  608. if (state == SOCK_BOUND) {
  609. /* if the socket is bound, the stream needs to be shut and rebound. */
  610. assert(hdl->pal_handle);
  611. DkStreamDelete(hdl->pal_handle, 0);
  612. DkObjectClose(hdl->pal_handle);
  613. hdl->pal_handle = NULL;
  614. }
  615. if (sock->domain != AF_UNIX) {
  616. if (!inet_check_addr(sock->domain, addr, addrlen))
  617. goto out;
  618. inet_save_addr(sock->domain, &sock->addr.in.conn, addr);
  619. inet_rebase_port(false, sock->domain, &sock->addr.in.conn, false);
  620. }
  621. sock->sock_state = (state == SOCK_BOUND) ? SOCK_BOUNDCONNECTED :
  622. SOCK_CONNECTED;
  623. if ((ret = create_socket_uri(hdl)) < 0)
  624. goto out;
  625. PAL_HANDLE pal_hdl = DkStreamOpen(qstrgetstr(&hdl->uri),
  626. 0, 0, 0,
  627. hdl->flags & O_NONBLOCK);
  628. if (!pal_hdl) {
  629. ret = -PAL_ERRNO;
  630. goto out;
  631. }
  632. hdl->pal_handle = pal_hdl;
  633. if (sock->domain == AF_UNIX) {
  634. struct shim_dentry * dent = sock->addr.un.dentry;
  635. lock(dent->lock);
  636. dent->state ^= DENTRY_NEGATIVE;
  637. dent->state |= DENTRY_VALID|DENTRY_RECENTLY;
  638. dent->fs = &socket_builtin_fs;
  639. dent->data = sock->addr.un.data;
  640. unlock(dent->lock);
  641. }
  642. if (sock->domain == AF_INET || sock->domain == AF_INET6) {
  643. char uri[SOCK_URI_SIZE];
  644. if (!DkStreamGetName(pal_hdl, uri, SOCK_URI_SIZE)) {
  645. ret = -PAL_ERRNO;
  646. goto out;
  647. }
  648. if ((ret = inet_parse_addr(sock->domain, sock->sock_type, uri,
  649. &sock->addr.in.bind,
  650. &sock->addr.in.conn)) < 0)
  651. goto out;
  652. inet_rebase_port(true, sock->domain, &sock->addr.in.bind, true);
  653. inet_rebase_port(true, sock->domain, &sock->addr.in.conn, false);
  654. }
  655. hdl->acc_mode = MAY_READ|MAY_WRITE;
  656. __process_pending_options(hdl);
  657. ret = 0;
  658. out:
  659. if (ret < 0) {
  660. sock->sock_state = state;
  661. sock->error = -ret;
  662. if (sock->domain == AF_UNIX) {
  663. if (sock->addr.un.dentry)
  664. put_dentry(sock->addr.un.dentry);
  665. if (sock->addr.un.data) {
  666. free(sock->addr.un.data);
  667. sock->addr.un.data = NULL;
  668. }
  669. }
  670. }
  671. unlock(hdl->lock);
  672. put_handle(hdl);
  673. return ret;
  674. }
  675. int __do_accept (struct shim_handle * hdl, int flags, struct sockaddr * addr,
  676. socklen_t * addrlen)
  677. {
  678. if (hdl->type != TYPE_SOCK)
  679. return -ENOTSOCK;
  680. struct shim_sock_handle * sock = &hdl->info.sock;
  681. int ret;
  682. PAL_HANDLE accepted = NULL;
  683. if (sock->sock_type != SOCK_STREAM) {
  684. debug("shim_listen: not a stream socket\n");
  685. return -EINVAL;
  686. }
  687. lock(hdl->lock);
  688. if (sock->sock_state != SOCK_LISTENED) {
  689. debug("shim_accpet: invalid socket\n");
  690. ret = -EINVAL;
  691. goto out;
  692. }
  693. accepted = DkStreamWaitForClient(hdl->pal_handle);
  694. if (!accepted) {
  695. ret = -PAL_ERRNO;
  696. goto out;
  697. }
  698. if (flags & O_NONBLOCK) {
  699. PAL_STREAM_ATTR attr;
  700. if (!DkStreamAttributesQuerybyHandle(accepted, &attr)) {
  701. ret = -PAL_ERRNO;
  702. goto out;
  703. }
  704. attr.nonblocking = PAL_TRUE;
  705. if (!DkStreamAttributesSetbyHandle(accepted, &attr)) {
  706. ret = -PAL_ERRNO;
  707. goto out;
  708. }
  709. }
  710. struct shim_handle * cli = get_new_handle();
  711. if (!cli) {
  712. ret = -ENOMEM;
  713. goto out;
  714. }
  715. struct shim_sock_handle * cli_sock = &cli->info.sock;
  716. cli->type = TYPE_SOCK;
  717. set_handle_fs(cli, &socket_builtin_fs);
  718. cli->acc_mode = MAY_READ|MAY_WRITE;
  719. cli->flags = O_RDWR|flags;
  720. cli->pal_handle = accepted;
  721. cli_sock->domain = sock->domain;
  722. cli_sock->sock_type = sock->sock_type;
  723. cli_sock->protocol = sock->protocol;
  724. cli_sock->sock_state = SOCK_ACCEPTED;
  725. if (sock->domain == AF_UNIX) {
  726. cli_sock->addr.un.pipeid = sock->addr.un.pipeid;
  727. if (sock->addr.un.dentry) {
  728. get_dentry(sock->addr.un.dentry);
  729. cli_sock->addr.un.dentry = sock->addr.un.dentry;
  730. }
  731. qstrsetstr(&cli->uri, qstrgetstr(&hdl->uri), hdl->uri.len);
  732. if (addr) {
  733. unix_copy_addr(addr, sock->addr.un.dentry);
  734. if (addrlen)
  735. *addrlen = sizeof(struct sockaddr_un);
  736. }
  737. }
  738. if (sock->domain == AF_INET || sock->domain == AF_INET6) {
  739. char uri[SOCK_URI_SIZE];
  740. int uri_len;
  741. if (!(uri_len = DkStreamGetName(accepted, uri, SOCK_URI_SIZE))) {
  742. ret = -PAL_ERRNO;
  743. out_hdl:
  744. DkObjectClose(accepted);
  745. goto out_cli;
  746. }
  747. if ((ret = inet_parse_addr(cli_sock->domain, cli_sock->sock_type, uri,
  748. &cli_sock->addr.in.bind,
  749. &cli_sock->addr.in.conn)) < 0)
  750. goto out_hdl;
  751. qstrsetstr(&cli->uri, uri, uri_len);
  752. inet_rebase_port(true, cli_sock->domain, &cli_sock->addr.in.bind, true);
  753. inet_rebase_port(true, cli_sock->domain, &cli_sock->addr.in.conn, false);
  754. if (addr) {
  755. inet_copy_addr(sock->domain, addr, &sock->addr.in.conn);
  756. if (addrlen)
  757. *addrlen = (sock->domain == AF_INET) ?
  758. sizeof(struct sockaddr_in) :
  759. sizeof(struct sockaddr_in6);
  760. }
  761. }
  762. ret = set_new_fd_handle(cli, flags & O_CLOEXEC ? FD_CLOEXEC : 0, NULL);
  763. out_cli:
  764. put_handle(cli);
  765. out:
  766. if (ret < 0)
  767. sock->error = -ret;
  768. unlock(hdl->lock);
  769. return ret;
  770. }
  771. int shim_do_accept (int fd, struct sockaddr * addr, socklen_t * addrlen)
  772. {
  773. int flags;
  774. struct shim_handle * hdl = get_fd_handle(fd, &flags, NULL);
  775. if (!hdl)
  776. return -EBADF;
  777. int ret = __do_accept(hdl, flags & FD_CLOEXEC ? O_CLOEXEC : 0,
  778. addr, addrlen);
  779. put_handle(hdl);
  780. return ret;
  781. }
  782. int shim_do_accept4 (int fd, struct sockaddr * addr, socklen_t * addrlen,
  783. int flags)
  784. {
  785. struct shim_handle * hdl = get_fd_handle(fd, NULL, NULL);
  786. if (!hdl)
  787. return -EBADF;
  788. int ret = __do_accept(hdl,
  789. (flags & SOCK_CLOEXEC ? FD_CLOEXEC : 0) |
  790. (flags & SOCK_NONBLOCK ? O_NONBLOCK : 0),
  791. addr, addrlen);
  792. put_handle(hdl);
  793. return ret;
  794. }
  795. static ssize_t do_sendmsg (int fd, struct iovec * bufs, int nbufs, int flags,
  796. const struct sockaddr * addr, socklen_t addrlen)
  797. {
  798. struct shim_handle * hdl = get_fd_handle(fd, NULL, NULL);
  799. if (!hdl)
  800. return -EBADF;
  801. int ret = -EINVAL;
  802. if (hdl->type != TYPE_SOCK) {
  803. ret = -ENOTSOCK;
  804. goto out;
  805. }
  806. struct shim_sock_handle * sock = &hdl->info.sock;
  807. lock(hdl->lock);
  808. PAL_HANDLE pal_hdl = hdl->pal_handle;
  809. char * uri = NULL;
  810. /* Data gram sock need not be conneted or bound at all */
  811. if (sock->sock_type == SOCK_STREAM &&
  812. sock->sock_state != SOCK_CONNECTED &&
  813. sock->sock_state != SOCK_BOUNDCONNECTED &&
  814. sock->sock_state != SOCK_ACCEPTED) {
  815. ret = -ENOTCONN;
  816. goto out_locked;
  817. }
  818. if (sock->sock_type == SOCK_DGRAM &&
  819. sock->sock_state == SOCK_SHUTDOWN) {
  820. ret = -ENOTCONN;
  821. goto out_locked;
  822. }
  823. if (!(hdl->acc_mode & MAY_WRITE)) {
  824. ret = -ECONNRESET;
  825. goto out_locked;
  826. }
  827. if (sock->sock_type == SOCK_DGRAM &&
  828. sock->sock_state != SOCK_BOUNDCONNECTED &&
  829. sock->sock_state != SOCK_CONNECTED) {
  830. if (!addr) {
  831. ret = -EDESTADDRREQ;
  832. goto out_locked;
  833. }
  834. if (sock->sock_state == SOCK_CREATED && !pal_hdl) {
  835. pal_hdl = DkStreamOpen("udp:", 0, 0, 0,
  836. hdl->flags & O_NONBLOCK);
  837. if (!pal_hdl) {
  838. ret = -PAL_ERRNO;
  839. goto out_locked;
  840. }
  841. hdl->pal_handle = pal_hdl;
  842. }
  843. if (addr && addr->sa_family != sock->domain) {
  844. ret = -EINVAL;
  845. goto out_locked;
  846. }
  847. uri = __alloca(SOCK_URI_SIZE);
  848. }
  849. unlock(hdl->lock);
  850. if (uri) {
  851. struct addr_inet addr_buf;
  852. inet_save_addr(sock->domain, &addr_buf, addr);
  853. inet_rebase_port(false, sock->domain, &addr_buf, false);
  854. memcpy(uri, "udp:", 5);
  855. if ((ret = inet_translate_addr(sock->domain, uri + 4, SOCK_URI_SIZE - 4,
  856. &addr_buf)) < 0) {
  857. lock(hdl->lock);
  858. goto out_locked;
  859. }
  860. debug("next packet send to %s\n", uri);
  861. }
  862. int bytes = 0;
  863. ret = 0;
  864. for (int i = 0 ; i < nbufs ; i++) {
  865. ret = DkStreamWrite(pal_hdl, 0, bufs[i].iov_len, bufs[i].iov_base,
  866. uri);
  867. if (!ret) {
  868. ret = (PAL_NATIVE_ERRNO == PAL_ERROR_STREAMEXIST) ?
  869. - ECONNABORTED : -PAL_ERRNO;
  870. break;
  871. }
  872. bytes += ret;
  873. }
  874. if (bytes)
  875. ret = bytes;
  876. if (ret < 0) {
  877. lock(hdl->lock);
  878. goto out_locked;
  879. }
  880. goto out;
  881. out_locked:
  882. if (ret < 0)
  883. sock->error = -ret;
  884. unlock(hdl->lock);
  885. out:
  886. put_handle(hdl);
  887. return ret;
  888. }
  889. ssize_t shim_do_sendto (int sockfd, const void * buf, size_t len, int flags,
  890. const struct sockaddr * addr, socklen_t addrlen)
  891. {
  892. struct iovec iovbuf;
  893. iovbuf.iov_base = (void *) buf;
  894. iovbuf.iov_len = len;
  895. return do_sendmsg(sockfd, &iovbuf, 1, flags, addr, addrlen);
  896. }
  897. ssize_t shim_do_sendmsg (int sockfd, struct msghdr * msg, int flags)
  898. {
  899. return do_sendmsg(sockfd, msg->msg_iov, msg->msg_iovlen, flags,
  900. msg->msg_name, msg->msg_namelen);
  901. }
  902. static ssize_t do_recvmsg (int fd, struct iovec * bufs, int nbufs, int flags,
  903. struct sockaddr * addr, socklen_t * addrlen)
  904. {
  905. struct shim_handle * hdl = get_fd_handle(fd, NULL, NULL);
  906. if (!hdl)
  907. return -EBADF;
  908. int ret = 0;
  909. if (hdl->type != TYPE_SOCK) {
  910. ret = -ENOTSOCK;
  911. goto out;
  912. }
  913. struct shim_sock_handle * sock = &hdl->info.sock;
  914. lock(hdl->lock);
  915. PAL_HANDLE pal_hdl = hdl->pal_handle;
  916. char * uri = NULL;
  917. if (sock->sock_type == SOCK_STREAM &&
  918. sock->sock_state != SOCK_CONNECTED &&
  919. sock->sock_state != SOCK_BOUNDCONNECTED &&
  920. sock->sock_state != SOCK_ACCEPTED) {
  921. ret = -ENOTCONN;
  922. goto out_locked;
  923. }
  924. if (!(hdl->acc_mode & MAY_READ)) {
  925. ret = 0;
  926. goto out_locked;
  927. }
  928. if (addr && sock->sock_type == SOCK_DGRAM &&
  929. sock->sock_state != SOCK_CONNECTED &&
  930. sock->sock_state != SOCK_BOUNDCONNECTED) {
  931. if (sock->sock_state == SOCK_CREATED) {
  932. ret = -EINVAL;
  933. goto out_locked;
  934. }
  935. uri = __alloca(SOCK_URI_SIZE);
  936. }
  937. unlock(hdl->lock);
  938. bool address_received = false;
  939. int bytes = 0;
  940. ret = 0;
  941. for (int i = 0 ; i < nbufs ; i++) {
  942. ret = DkStreamRead(pal_hdl, 0, bufs[i].iov_len, bufs[i].iov_base,
  943. uri, uri ? SOCK_URI_SIZE : 0);
  944. if (!ret) {
  945. ret = (PAL_NATIVE_ERRNO == PAL_ERROR_STREAMNOTEXIST) ?
  946. - ECONNABORTED : -PAL_ERRNO;
  947. break;
  948. }
  949. bytes += ret;
  950. if (!addr || !bytes || address_received)
  951. continue;
  952. if (sock->domain == AF_UNIX) {
  953. unix_copy_addr(addr, sock->addr.un.dentry);
  954. *addrlen = sizeof(struct sockaddr_un);
  955. }
  956. if (sock->domain == AF_INET || sock->domain == AF_INET6) {
  957. if (uri) {
  958. struct addr_inet conn;
  959. if ((ret = inet_parse_addr(sock->domain, sock->sock_type, uri,
  960. &conn, NULL)) < 0) {
  961. lock(hdl->lock);
  962. goto out_locked;
  963. }
  964. debug("last packet received from %s\n", uri);
  965. inet_rebase_port(true, sock->domain, &conn, false);
  966. inet_copy_addr(sock->domain, addr, &conn);
  967. } else {
  968. inet_copy_addr(sock->domain, addr, &sock->addr.in.conn);
  969. }
  970. *addrlen = (sock->domain == AF_INET) ?
  971. sizeof(struct sockaddr_in) : sizeof(struct sockaddr_in6);
  972. }
  973. address_received = false;
  974. }
  975. if (bytes)
  976. ret = bytes;
  977. if (ret < 0) {
  978. lock(hdl->lock);
  979. goto out_locked;
  980. }
  981. goto out;
  982. out_locked:
  983. if (ret < 0)
  984. sock->error = -ret;
  985. unlock(hdl->lock);
  986. out:
  987. put_handle(hdl);
  988. return ret;
  989. }
  990. ssize_t shim_do_recvfrom (int sockfd, void * buf, size_t len, int flags,
  991. struct sockaddr * addr, socklen_t * addrlen)
  992. {
  993. struct iovec iovbuf;
  994. iovbuf.iov_base = (void *) buf;
  995. iovbuf.iov_len = len;
  996. return do_recvmsg(sockfd, &iovbuf, 1, flags, addr, addrlen);
  997. }
  998. ssize_t shim_do_recvmsg (int sockfd, struct msghdr * msg, int flags)
  999. {
  1000. return do_recvmsg(sockfd, msg->msg_iov, msg->msg_iovlen, flags,
  1001. msg->msg_name, &msg->msg_namelen);
  1002. }
  1003. #define SHUT_RD 0
  1004. #define SHUT_WR 1
  1005. #define SHUT_RDWR 2
  1006. int shim_do_shutdown (int sockfd, int how)
  1007. {
  1008. struct shim_handle * hdl = get_fd_handle(sockfd, NULL, NULL);
  1009. if (!hdl)
  1010. return -EBADF;
  1011. int ret = 0;
  1012. struct shim_sock_handle * sock = &hdl->info.sock;
  1013. if (hdl->type != TYPE_SOCK) {
  1014. ret = -ENOTSOCK;
  1015. goto out;
  1016. }
  1017. lock(hdl->lock);
  1018. if (sock->sock_state != SOCK_LISTENED &&
  1019. sock->sock_state != SOCK_ACCEPTED &&
  1020. sock->sock_state != SOCK_CONNECTED &&
  1021. sock->sock_state != SOCK_BOUNDCONNECTED) {
  1022. ret = -ENOTCONN;
  1023. goto out_locked;
  1024. }
  1025. switch (how) {
  1026. case SHUT_RD:
  1027. DkStreamDelete(hdl->pal_handle, PAL_DELETE_RD);
  1028. hdl->acc_mode &= ~MAY_READ;
  1029. break;
  1030. case SHUT_WR:
  1031. DkStreamDelete(hdl->pal_handle, PAL_DELETE_WR);
  1032. hdl->acc_mode &= ~MAY_WRITE;
  1033. break;
  1034. case SHUT_RDWR:
  1035. DkStreamDelete(hdl->pal_handle, 0);
  1036. hdl->acc_mode = 0;
  1037. sock->sock_state = SOCK_SHUTDOWN;
  1038. break;
  1039. }
  1040. ret = 0;
  1041. out_locked:
  1042. if (ret < 0)
  1043. sock->error = -ret;
  1044. unlock(hdl->lock);
  1045. out:
  1046. put_handle(hdl);
  1047. return ret;
  1048. }
  1049. int shim_do_getsockname (int sockfd, struct sockaddr * addr, int * addrlen)
  1050. {
  1051. if (!addr || !addrlen)
  1052. return -EFAULT;
  1053. if (*addrlen <= 0)
  1054. return -EINVAL;
  1055. struct shim_handle * hdl = get_fd_handle(sockfd, NULL, NULL);
  1056. if (!hdl)
  1057. return -EBADF;
  1058. int ret = -EINVAL;
  1059. if (hdl->type != TYPE_SOCK) {
  1060. ret = -ENOTSOCK;
  1061. goto out;
  1062. }
  1063. struct shim_sock_handle * sock = &hdl->info.sock;
  1064. lock(hdl->lock);
  1065. struct sockaddr saddr;
  1066. int len = inet_copy_addr(sock->domain, &saddr, &sock->addr.in.bind);
  1067. if (len < *addrlen)
  1068. len = *addrlen;
  1069. memcpy(addr, &saddr, len);
  1070. *addrlen = len;
  1071. ret = 0;
  1072. unlock(hdl->lock);
  1073. out:
  1074. put_handle(hdl);
  1075. return ret;
  1076. }
  1077. int shim_do_getpeername (int sockfd, struct sockaddr * addr, int * addrlen)
  1078. {
  1079. if (!addr || !addrlen)
  1080. return -EFAULT;
  1081. if (*addrlen <= 0)
  1082. return -EINVAL;
  1083. struct shim_handle * hdl = get_fd_handle(sockfd, NULL, NULL);
  1084. if (!hdl)
  1085. return -EBADF;
  1086. int ret = -EINVAL;
  1087. if (hdl->type != TYPE_SOCK) {
  1088. ret = -ENOTSOCK;
  1089. goto out;
  1090. }
  1091. struct shim_sock_handle * sock = &hdl->info.sock;
  1092. lock(hdl->lock);
  1093. /* Data gram sock need not be conneted or bound at all */
  1094. if (sock->sock_type == SOCK_STREAM &&
  1095. sock->sock_state != SOCK_CONNECTED &&
  1096. sock->sock_state != SOCK_BOUNDCONNECTED &&
  1097. sock->sock_state != SOCK_ACCEPTED) {
  1098. ret = -ENOTCONN;
  1099. goto out_locked;
  1100. }
  1101. if (sock->sock_type == SOCK_DGRAM &&
  1102. sock->sock_state != SOCK_CONNECTED &&
  1103. sock->sock_state != SOCK_BOUNDCONNECTED) {
  1104. ret = -ENOTCONN;
  1105. goto out_locked;
  1106. }
  1107. struct sockaddr saddr;
  1108. int len = inet_copy_addr (sock->domain, &saddr, &sock->addr.in.conn);
  1109. if (len < *addrlen)
  1110. len = *addrlen;
  1111. memcpy(addr, &saddr, len);
  1112. *addrlen = len;
  1113. ret = 0;
  1114. out_locked:
  1115. unlock(hdl->lock);
  1116. out:
  1117. put_handle(hdl);
  1118. return ret;
  1119. }
  1120. struct __kernel_linger {
  1121. int l_onoff;
  1122. int l_linger;
  1123. };
  1124. static int __do_setsockopt (struct shim_handle * hdl, int level, int optname,
  1125. char * optval, int optlen, PAL_STREAM_ATTR * attr)
  1126. {
  1127. int intval = *((int *) optval);
  1128. PAL_BOL bolval = intval ? PAL_TRUE : PAL_FALSE;
  1129. if (level == SOL_SOCKET) {
  1130. switch(optname) {
  1131. case SO_ACCEPTCONN:
  1132. case SO_DOMAIN:
  1133. case SO_ERROR:
  1134. case SO_PROTOCOL:
  1135. case SO_TYPE:
  1136. return -EPERM;
  1137. case SO_KEEPALIVE:
  1138. case SO_LINGER:
  1139. case SO_RCVBUF:
  1140. case SO_SNDBUF:
  1141. case SO_RCVTIMEO:
  1142. case SO_SNDTIMEO:
  1143. case SO_REUSEADDR:
  1144. goto query;
  1145. default:
  1146. goto unknown;
  1147. }
  1148. }
  1149. if (level == SOL_TCP) {
  1150. switch(optname) {
  1151. case TCP_CORK:
  1152. case TCP_NODELAY:
  1153. goto query;
  1154. default:
  1155. goto unknown;
  1156. }
  1157. }
  1158. unknown:
  1159. return -ENOPROTOOPT;
  1160. query:
  1161. if (!attr) {
  1162. attr = __alloca(sizeof(PAL_STREAM_ATTR));
  1163. if (!DkStreamAttributesQuerybyHandle(hdl->pal_handle, attr))
  1164. return -PAL_ERRNO;
  1165. }
  1166. if (level == SOL_SOCKET) {
  1167. switch(optname) {
  1168. case SO_KEEPALIVE:
  1169. if (bolval != attr->socket.tcp_keepalive) {
  1170. attr->socket.tcp_keepalive = bolval;
  1171. goto set;
  1172. }
  1173. break;
  1174. case SO_LINGER: {
  1175. struct __kernel_linger * l = (struct __kernel_linger *) optval;
  1176. int linger = l->l_onoff ? l->l_linger : 0;
  1177. if (linger != (int) attr->socket.linger) {
  1178. attr->socket.linger = linger;
  1179. goto set;
  1180. }
  1181. break;
  1182. }
  1183. case SO_RCVBUF:
  1184. if (intval != (int) attr->socket.receivebuf) {
  1185. attr->socket.receivebuf = intval;
  1186. goto set;
  1187. }
  1188. break;
  1189. case SO_SNDBUF:
  1190. if (intval != (int) attr->socket.sendbuf) {
  1191. attr->socket.sendbuf = intval;
  1192. goto set;
  1193. }
  1194. break;
  1195. case SO_RCVTIMEO:
  1196. if (intval != (int) attr->socket.receivetimeout) {
  1197. attr->socket.receivetimeout = intval;
  1198. goto set;
  1199. }
  1200. break;
  1201. case SO_SNDTIMEO:
  1202. if (intval != (int) attr->socket.sendtimeout) {
  1203. attr->socket.sendtimeout = intval;
  1204. goto set;
  1205. }
  1206. break;
  1207. case SO_REUSEADDR:
  1208. break;
  1209. }
  1210. }
  1211. if (level == SOL_TCP) {
  1212. switch(optname) {
  1213. case TCP_CORK:
  1214. if (bolval != attr->socket.tcp_cork) {
  1215. attr->socket.tcp_cork = bolval;
  1216. goto set;
  1217. }
  1218. break;
  1219. case TCP_NODELAY:
  1220. if (bolval != attr->socket.tcp_nodelay) {
  1221. attr->socket.tcp_nodelay = bolval;
  1222. goto set;
  1223. }
  1224. break;
  1225. }
  1226. }
  1227. return 0;
  1228. set:
  1229. if (!DkStreamAttributesSetbyHandle(hdl->pal_handle, attr))
  1230. return -PAL_ERRNO;
  1231. return 0;
  1232. }
  1233. static int __process_pending_options (struct shim_handle * hdl)
  1234. {
  1235. struct shim_sock_handle * sock = &hdl->info.sock;
  1236. if (!sock->pending_options)
  1237. return 0;
  1238. PAL_STREAM_ATTR attr;
  1239. if (!DkStreamAttributesQuerybyHandle(hdl->pal_handle, &attr))
  1240. return -PAL_ERRNO;
  1241. struct shim_sock_option * o = sock->pending_options;
  1242. while (o) {
  1243. PAL_STREAM_ATTR tmp = attr;
  1244. int ret = __do_setsockopt(hdl, o->level, o->optname, o->optval,
  1245. o->optlen, &tmp);
  1246. if (!ret)
  1247. attr = tmp;
  1248. struct shim_sock_option * next = o->next;
  1249. free(o);
  1250. o = next;
  1251. }
  1252. return 0;
  1253. }
  1254. int shim_do_setsockopt (int fd, int level, int optname, char * optval,
  1255. int optlen)
  1256. {
  1257. if (!optval)
  1258. return -EFAULT;
  1259. struct shim_handle * hdl = get_fd_handle(fd, NULL, NULL);
  1260. if (!hdl)
  1261. return -EBADF;
  1262. int ret = 0;
  1263. if (hdl->type != TYPE_SOCK) {
  1264. ret = -ENOTSOCK;
  1265. goto out;
  1266. }
  1267. struct shim_sock_handle * sock = &hdl->info.sock;
  1268. lock(hdl->lock);
  1269. if (!hdl->pal_handle) {
  1270. struct shim_sock_option * o = malloc(sizeof(struct shim_sock_option) +
  1271. optlen);
  1272. if (!o) {
  1273. ret = -ENOMEM;
  1274. goto out_locked;
  1275. }
  1276. struct shim_sock_option ** next = &sock->pending_options;
  1277. while (*next)
  1278. next = &(*next)->next;
  1279. o->next = NULL;
  1280. *next = o;
  1281. o->level = level;
  1282. o->optname = optname;
  1283. o->optlen = optlen;
  1284. memcpy(&o->optval, optval, optlen);
  1285. goto out_locked;
  1286. }
  1287. ret = __do_setsockopt(hdl, level, optname, optval, optlen, NULL);
  1288. out_locked:
  1289. unlock(hdl->lock);
  1290. out:
  1291. put_handle(hdl);
  1292. return ret;
  1293. }
  1294. int shim_do_getsockopt (int fd, int level, int optname, char * optval,
  1295. int * optlen)
  1296. {
  1297. if (!optval || !optlen)
  1298. return -EFAULT;
  1299. struct shim_handle * hdl = get_fd_handle(fd, NULL, NULL);
  1300. if (!hdl)
  1301. return -EBADF;
  1302. int ret = 0;
  1303. if (hdl->type != TYPE_SOCK) {
  1304. ret = -ENOTSOCK;
  1305. goto out;
  1306. }
  1307. struct shim_sock_handle * sock = &hdl->info.sock;
  1308. lock(hdl->lock);
  1309. int * intval = (int *) optval;
  1310. if (level == SOL_SOCKET) {
  1311. switch(optname) {
  1312. case SO_ACCEPTCONN:
  1313. *intval = (sock->sock_state == SOCK_LISTENED) ? 1 : 0;
  1314. goto out;
  1315. case SO_DOMAIN:
  1316. *intval = sock->domain;
  1317. goto out;
  1318. case SO_ERROR:
  1319. *intval = sock->error;
  1320. goto out;
  1321. case SO_PROTOCOL:
  1322. switch(sock->protocol) {
  1323. case SOCK_STREAM:
  1324. *intval = IPPROTO_SCTP;
  1325. break;
  1326. case SOCK_DGRAM:
  1327. *intval = IPPROTO_UDP;
  1328. break;
  1329. default:
  1330. goto unknown;
  1331. }
  1332. goto out;
  1333. case SO_TYPE:
  1334. *intval = sock->protocol;
  1335. goto out;
  1336. case SO_KEEPALIVE:
  1337. case SO_LINGER:
  1338. case SO_RCVBUF:
  1339. case SO_SNDBUF:
  1340. case SO_RCVTIMEO:
  1341. case SO_SNDTIMEO:
  1342. case SO_REUSEADDR:
  1343. goto query;
  1344. default:
  1345. goto unknown;
  1346. }
  1347. }
  1348. if (level == SOL_TCP) {
  1349. switch(optname) {
  1350. case TCP_CORK:
  1351. case TCP_NODELAY:
  1352. goto query;
  1353. default:
  1354. goto unknown;
  1355. }
  1356. }
  1357. unknown:
  1358. ret = -ENOPROTOOPT;
  1359. goto out;
  1360. query:
  1361. {
  1362. PAL_STREAM_ATTR attr;
  1363. if (!DkStreamAttributesQuerybyHandle(hdl->pal_handle, &attr)) {
  1364. ret = -PAL_ERRNO;
  1365. goto out;
  1366. }
  1367. if (level == SOL_SOCKET) {
  1368. switch(optname) {
  1369. case SO_KEEPALIVE:
  1370. *intval = attr.socket.tcp_keepalive ? 1 : 0;
  1371. break;
  1372. case SO_LINGER: {
  1373. struct __kernel_linger * l =
  1374. (struct __kernel_linger *) optval;
  1375. l->l_onoff = attr.socket.linger ? 1 : 0;
  1376. l->l_linger = attr.socket.linger;
  1377. break;
  1378. }
  1379. case SO_RCVBUF:
  1380. *intval = attr.socket.receivebuf;
  1381. break;
  1382. case SO_SNDBUF:
  1383. *intval = attr.socket.sendbuf;
  1384. break;
  1385. case SO_RCVTIMEO:
  1386. *intval = attr.socket.receivetimeout;
  1387. break;
  1388. case SO_SNDTIMEO:
  1389. *intval = attr.socket.sendtimeout;
  1390. break;
  1391. case SO_REUSEADDR:
  1392. *intval = 1;
  1393. break;
  1394. }
  1395. }
  1396. if (level == SOL_TCP) {
  1397. switch(optname) {
  1398. case TCP_CORK:
  1399. *intval = attr.socket.tcp_cork ? 1 : 0;
  1400. break;
  1401. case TCP_NODELAY:
  1402. *intval = attr.socket.tcp_nodelay ? 1 : 0;
  1403. break;
  1404. }
  1405. }
  1406. }
  1407. out:
  1408. unlock(hdl->lock);
  1409. put_handle(hdl);
  1410. return ret;
  1411. }