sockets.c 135 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182
  1. /**
  2. * @file
  3. * Sockets BSD-Like API module
  4. */
  5. /*
  6. * Copyright (c) 2001-2004 Swedish Institute of Computer Science.
  7. * All rights reserved.
  8. *
  9. * Redistribution and use in source and binary forms, with or without modification,
  10. * are permitted provided that the following conditions are met:
  11. *
  12. * 1. Redistributions of source code must retain the above copyright notice,
  13. * this list of conditions and the following disclaimer.
  14. * 2. Redistributions in binary form must reproduce the above copyright notice,
  15. * this list of conditions and the following disclaimer in the documentation
  16. * and/or other materials provided with the distribution.
  17. * 3. The name of the author may not be used to endorse or promote products
  18. * derived from this software without specific prior written permission.
  19. *
  20. * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
  21. * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
  22. * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT
  23. * SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  24. * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
  25. * OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  26. * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  27. * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
  28. * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
  29. * OF SUCH DAMAGE.
  30. *
  31. * This file is part of the lwIP TCP/IP stack.
  32. *
  33. * Author: Adam Dunkels <adam@sics.se>
  34. *
  35. * Improved by Marc Boucher <marc@mbsi.ca> and David Haas <dhaas@alum.rpi.edu>
  36. *
  37. */
  38. #include "lwip/opt.h"
  39. #if LWIP_SOCKET /* don't build if not configured for use in lwipopts.h */
  40. #include "lwip/sockets.h"
  41. #include "lwip/priv/sockets_priv.h"
  42. #include "lwip/api.h"
  43. #include "lwip/igmp.h"
  44. #include "lwip/inet.h"
  45. #include "lwip/tcp.h"
  46. #include "lwip/raw.h"
  47. #include "lwip/udp.h"
  48. #include "lwip/memp.h"
  49. #include "lwip/pbuf.h"
  50. #include "lwip/netif.h"
  51. #include "lwip/priv/tcpip_priv.h"
  52. #include "lwip/mld6.h"
  53. #if LWIP_CHECKSUM_ON_COPY
  54. #include "lwip/inet_chksum.h"
  55. #endif
  56. #if LWIP_COMPAT_SOCKETS == 2 && LWIP_POSIX_SOCKETS_IO_NAMES
  57. #include <stdarg.h>
  58. #endif
  59. #include <string.h>
  60. #ifdef LWIP_HOOK_FILENAME
  61. #include LWIP_HOOK_FILENAME
  62. #endif
  63. /* If the netconn API is not required publicly, then we include the necessary
  64. files here to get the implementation */
  65. #if !LWIP_NETCONN
  66. #undef LWIP_NETCONN
  67. #define LWIP_NETCONN 1
  68. #include "api_msg.c"
  69. #include "api_lib.c"
  70. #include "netbuf.c"
  71. #undef LWIP_NETCONN
  72. #define LWIP_NETCONN 0
  73. #endif
  74. #define API_SELECT_CB_VAR_REF(name) API_VAR_REF(name)
  75. #define API_SELECT_CB_VAR_DECLARE(name) API_VAR_DECLARE(struct lwip_select_cb, name)
  76. #define API_SELECT_CB_VAR_ALLOC(name, retblock) API_VAR_ALLOC_EXT(struct lwip_select_cb, MEMP_SELECT_CB, name, retblock)
  77. #define API_SELECT_CB_VAR_FREE(name) API_VAR_FREE(MEMP_SELECT_CB, name)
  78. #if LWIP_IPV4
  79. #define IP4ADDR_PORT_TO_SOCKADDR(sin, ipaddr, port) do { \
  80. (sin)->sin_len = sizeof(struct sockaddr_in); \
  81. (sin)->sin_family = AF_INET; \
  82. (sin)->sin_port = lwip_htons((port)); \
  83. inet_addr_from_ip4addr(&(sin)->sin_addr, ipaddr); \
  84. memset((sin)->sin_zero, 0, SIN_ZERO_LEN); }while(0)
  85. #define SOCKADDR4_TO_IP4ADDR_PORT(sin, ipaddr, port) do { \
  86. inet_addr_to_ip4addr(ip_2_ip4(ipaddr), &((sin)->sin_addr)); \
  87. (port) = lwip_ntohs((sin)->sin_port); }while(0)
  88. #endif /* LWIP_IPV4 */
  89. #if LWIP_IPV6
  90. #define IP6ADDR_PORT_TO_SOCKADDR(sin6, ipaddr, port) do { \
  91. (sin6)->sin6_len = sizeof(struct sockaddr_in6); \
  92. (sin6)->sin6_family = AF_INET6; \
  93. (sin6)->sin6_port = lwip_htons((port)); \
  94. (sin6)->sin6_flowinfo = 0; \
  95. inet6_addr_from_ip6addr(&(sin6)->sin6_addr, ipaddr); \
  96. (sin6)->sin6_scope_id = ip6_addr_zone(ipaddr); }while(0)
  97. #define SOCKADDR6_TO_IP6ADDR_PORT(sin6, ipaddr, port) do { \
  98. inet6_addr_to_ip6addr(ip_2_ip6(ipaddr), &((sin6)->sin6_addr)); \
  99. if (ip6_addr_has_scope(ip_2_ip6(ipaddr), IP6_UNKNOWN)) { \
  100. ip6_addr_set_zone(ip_2_ip6(ipaddr), (u8_t)((sin6)->sin6_scope_id)); \
  101. } \
  102. (port) = lwip_ntohs((sin6)->sin6_port); }while(0)
  103. #endif /* LWIP_IPV6 */
  104. #if LWIP_IPV4 && LWIP_IPV6
  105. static void sockaddr_to_ipaddr_port(const struct sockaddr *sockaddr, ip_addr_t *ipaddr, u16_t *port);
  106. #define IS_SOCK_ADDR_LEN_VALID(namelen) (((namelen) == sizeof(struct sockaddr_in)) || \
  107. ((namelen) == sizeof(struct sockaddr_in6)))
  108. #define IS_SOCK_ADDR_TYPE_VALID(name) (((name)->sa_family == AF_INET) || \
  109. ((name)->sa_family == AF_INET6))
  110. #define SOCK_ADDR_TYPE_MATCH(name, sock) \
  111. ((((name)->sa_family == AF_INET) && !(NETCONNTYPE_ISIPV6((sock)->conn->type))) || \
  112. (((name)->sa_family == AF_INET6) && (NETCONNTYPE_ISIPV6((sock)->conn->type))))
  113. #define IPADDR_PORT_TO_SOCKADDR(sockaddr, ipaddr, port) do { \
  114. if (IP_IS_ANY_TYPE_VAL(*ipaddr) || IP_IS_V6_VAL(*ipaddr)) { \
  115. IP6ADDR_PORT_TO_SOCKADDR((struct sockaddr_in6*)(void*)(sockaddr), ip_2_ip6(ipaddr), port); \
  116. } else { \
  117. IP4ADDR_PORT_TO_SOCKADDR((struct sockaddr_in*)(void*)(sockaddr), ip_2_ip4(ipaddr), port); \
  118. } } while(0)
  119. #define SOCKADDR_TO_IPADDR_PORT(sockaddr, ipaddr, port) sockaddr_to_ipaddr_port(sockaddr, ipaddr, &(port))
  120. #define DOMAIN_TO_NETCONN_TYPE(domain, type) (((domain) == AF_INET) ? \
  121. (type) : (enum netconn_type)((type) | NETCONN_TYPE_IPV6))
  122. #elif LWIP_IPV6 /* LWIP_IPV4 && LWIP_IPV6 */
  123. #define IS_SOCK_ADDR_LEN_VALID(namelen) ((namelen) == sizeof(struct sockaddr_in6))
  124. #define IS_SOCK_ADDR_TYPE_VALID(name) ((name)->sa_family == AF_INET6)
  125. #define SOCK_ADDR_TYPE_MATCH(name, sock) 1
  126. #define IPADDR_PORT_TO_SOCKADDR(sockaddr, ipaddr, port) \
  127. IP6ADDR_PORT_TO_SOCKADDR((struct sockaddr_in6*)(void*)(sockaddr), ip_2_ip6(ipaddr), port)
  128. #define SOCKADDR_TO_IPADDR_PORT(sockaddr, ipaddr, port) \
  129. SOCKADDR6_TO_IP6ADDR_PORT((const struct sockaddr_in6*)(const void*)(sockaddr), ipaddr, port)
  130. #define DOMAIN_TO_NETCONN_TYPE(domain, netconn_type) (netconn_type)
  131. #else /*-> LWIP_IPV4: LWIP_IPV4 && LWIP_IPV6 */
  132. #define IS_SOCK_ADDR_LEN_VALID(namelen) ((namelen) == sizeof(struct sockaddr_in))
  133. #define IS_SOCK_ADDR_TYPE_VALID(name) ((name)->sa_family == AF_INET)
  134. #define SOCK_ADDR_TYPE_MATCH(name, sock) 1
  135. #define IPADDR_PORT_TO_SOCKADDR(sockaddr, ipaddr, port) \
  136. IP4ADDR_PORT_TO_SOCKADDR((struct sockaddr_in*)(void*)(sockaddr), ip_2_ip4(ipaddr), port)
  137. #define SOCKADDR_TO_IPADDR_PORT(sockaddr, ipaddr, port) \
  138. SOCKADDR4_TO_IP4ADDR_PORT((const struct sockaddr_in*)(const void*)(sockaddr), ipaddr, port)
  139. #define DOMAIN_TO_NETCONN_TYPE(domain, netconn_type) (netconn_type)
  140. #endif /* LWIP_IPV6 */
  141. #define IS_SOCK_ADDR_TYPE_VALID_OR_UNSPEC(name) (((name)->sa_family == AF_UNSPEC) || \
  142. IS_SOCK_ADDR_TYPE_VALID(name))
  143. #define SOCK_ADDR_TYPE_MATCH_OR_UNSPEC(name, sock) (((name)->sa_family == AF_UNSPEC) || \
  144. SOCK_ADDR_TYPE_MATCH(name, sock))
  145. #define IS_SOCK_ADDR_ALIGNED(name) ((((mem_ptr_t)(name)) % 4) == 0)
  146. #define LWIP_SOCKOPT_CHECK_OPTLEN(sock, optlen, opttype) do { if ((optlen) < sizeof(opttype)) { done_socket(sock); return EINVAL; }}while(0)
  147. #define LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, optlen, opttype) do { \
  148. LWIP_SOCKOPT_CHECK_OPTLEN(sock, optlen, opttype); \
  149. if ((sock)->conn == NULL) { done_socket(sock); return EINVAL; } }while(0)
  150. #define LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, optlen, opttype) do { \
  151. LWIP_SOCKOPT_CHECK_OPTLEN(sock, optlen, opttype); \
  152. if (((sock)->conn == NULL) || ((sock)->conn->pcb.tcp == NULL)) { done_socket(sock); return EINVAL; } }while(0)
  153. #define LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, opttype, netconntype) do { \
  154. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, optlen, opttype); \
  155. if (NETCONNTYPE_GROUP(netconn_type((sock)->conn)) != netconntype) { done_socket(sock); return ENOPROTOOPT; } }while(0)
  156. #define LWIP_SETGETSOCKOPT_DATA_VAR_REF(name) API_VAR_REF(name)
  157. #define LWIP_SETGETSOCKOPT_DATA_VAR_DECLARE(name) API_VAR_DECLARE(struct lwip_setgetsockopt_data, name)
  158. #define LWIP_SETGETSOCKOPT_DATA_VAR_FREE(name) API_VAR_FREE(MEMP_SOCKET_SETGETSOCKOPT_DATA, name)
  159. #if LWIP_MPU_COMPATIBLE
  160. #define LWIP_SETGETSOCKOPT_DATA_VAR_ALLOC(name, sock) do { \
  161. name = (struct lwip_setgetsockopt_data *)memp_malloc(MEMP_SOCKET_SETGETSOCKOPT_DATA); \
  162. if (name == NULL) { \
  163. sock_set_errno(sock, ENOMEM); \
  164. done_socket(sock); \
  165. return -1; \
  166. } }while(0)
  167. #else /* LWIP_MPU_COMPATIBLE */
  168. #define LWIP_SETGETSOCKOPT_DATA_VAR_ALLOC(name, sock)
  169. #endif /* LWIP_MPU_COMPATIBLE */
  170. #if LWIP_SO_SNDRCVTIMEO_NONSTANDARD
  171. #define LWIP_SO_SNDRCVTIMEO_OPTTYPE int
  172. #define LWIP_SO_SNDRCVTIMEO_SET(optval, val) (*(int *)(optval) = (val))
  173. #define LWIP_SO_SNDRCVTIMEO_GET_MS(optval) ((long)*(const int*)(optval))
  174. #else
  175. #define LWIP_SO_SNDRCVTIMEO_OPTTYPE struct timeval
  176. #define LWIP_SO_SNDRCVTIMEO_SET(optval, val) do { \
  177. u32_t loc = (val); \
  178. ((struct timeval *)(optval))->tv_sec = (long)((loc) / 1000U); \
  179. ((struct timeval *)(optval))->tv_usec = (long)(((loc) % 1000U) * 1000U); }while(0)
  180. #define LWIP_SO_SNDRCVTIMEO_GET_MS(optval) ((((const struct timeval *)(optval))->tv_sec * 1000) + (((const struct timeval *)(optval))->tv_usec / 1000))
  181. #endif
  182. /** A struct sockaddr replacement that has the same alignment as sockaddr_in/
  183. * sockaddr_in6 if instantiated.
  184. */
  185. union sockaddr_aligned {
  186. struct sockaddr sa;
  187. #if LWIP_IPV6
  188. struct sockaddr_in6 sin6;
  189. #endif /* LWIP_IPV6 */
  190. #if LWIP_IPV4
  191. struct sockaddr_in sin;
  192. #endif /* LWIP_IPV4 */
  193. };
  194. /* Define the number of IPv4 multicast memberships, default is one per socket */
  195. #ifndef LWIP_SOCKET_MAX_MEMBERSHIPS
  196. #define LWIP_SOCKET_MAX_MEMBERSHIPS NUM_SOCKETS
  197. #endif
  198. #if LWIP_IGMP
  199. /* This is to keep track of IP_ADD_MEMBERSHIP calls to drop the membership when
  200. a socket is closed */
  201. struct lwip_socket_multicast_pair {
  202. /** the socket */
  203. struct lwip_sock *sock;
  204. /** the interface address */
  205. ip4_addr_t if_addr;
  206. /** the group address */
  207. ip4_addr_t multi_addr;
  208. };
  209. static struct lwip_socket_multicast_pair socket_ipv4_multicast_memberships[LWIP_SOCKET_MAX_MEMBERSHIPS];
  210. static int lwip_socket_register_membership(int s, const ip4_addr_t *if_addr, const ip4_addr_t *multi_addr);
  211. static void lwip_socket_unregister_membership(int s, const ip4_addr_t *if_addr, const ip4_addr_t *multi_addr);
  212. static void lwip_socket_drop_registered_memberships(int s);
  213. #endif /* LWIP_IGMP */
  214. #if LWIP_IPV6_MLD
  215. /* This is to keep track of IP_JOIN_GROUP calls to drop the membership when
  216. a socket is closed */
  217. struct lwip_socket_multicast_mld6_pair {
  218. /** the socket */
  219. struct lwip_sock *sock;
  220. /** the interface index */
  221. u8_t if_idx;
  222. /** the group address */
  223. ip6_addr_t multi_addr;
  224. };
  225. static struct lwip_socket_multicast_mld6_pair socket_ipv6_multicast_memberships[LWIP_SOCKET_MAX_MEMBERSHIPS];
  226. static int lwip_socket_register_mld6_membership(int s, unsigned int if_idx, const ip6_addr_t *multi_addr);
  227. static void lwip_socket_unregister_mld6_membership(int s, unsigned int if_idx, const ip6_addr_t *multi_addr);
  228. static void lwip_socket_drop_registered_mld6_memberships(int s);
  229. #endif /* LWIP_IPV6_MLD */
  230. /** The global array of available sockets */
  231. static struct lwip_sock sockets[NUM_SOCKETS];
  232. #if LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL
  233. #if LWIP_TCPIP_CORE_LOCKING
  234. /* protect the select_cb_list using core lock */
  235. #define LWIP_SOCKET_SELECT_DECL_PROTECT(lev)
  236. #define LWIP_SOCKET_SELECT_PROTECT(lev) LOCK_TCPIP_CORE()
  237. #define LWIP_SOCKET_SELECT_UNPROTECT(lev) UNLOCK_TCPIP_CORE()
  238. #else /* LWIP_TCPIP_CORE_LOCKING */
  239. /* protect the select_cb_list using SYS_LIGHTWEIGHT_PROT */
  240. #define LWIP_SOCKET_SELECT_DECL_PROTECT(lev) SYS_ARCH_DECL_PROTECT(lev)
  241. #define LWIP_SOCKET_SELECT_PROTECT(lev) SYS_ARCH_PROTECT(lev)
  242. #define LWIP_SOCKET_SELECT_UNPROTECT(lev) SYS_ARCH_UNPROTECT(lev)
  243. /** This counter is increased from lwip_select when the list is changed
  244. and checked in select_check_waiters to see if it has changed. */
  245. static volatile int select_cb_ctr;
  246. #endif /* LWIP_TCPIP_CORE_LOCKING */
  247. /** The global list of tasks waiting for select */
  248. static struct lwip_select_cb *select_cb_list;
  249. #endif /* LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL */
  250. #define sock_set_errno(sk, e) do { \
  251. const int sockerr = (e); \
  252. set_errno(sockerr); \
  253. } while (0)
  254. /* Forward declaration of some functions */
  255. #if LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL
  256. static void event_callback(struct netconn *conn, enum netconn_evt evt, u16_t len);
  257. #define DEFAULT_SOCKET_EVENTCB event_callback
  258. static void select_check_waiters(int s, int has_recvevent, int has_sendevent, int has_errevent);
  259. #else
  260. #define DEFAULT_SOCKET_EVENTCB NULL
  261. #endif
  262. #if !LWIP_TCPIP_CORE_LOCKING
  263. static void lwip_getsockopt_callback(void *arg);
  264. static void lwip_setsockopt_callback(void *arg);
  265. #endif
  266. static int lwip_getsockopt_impl(int s, int level, int optname, void *optval, socklen_t *optlen);
  267. static int lwip_setsockopt_impl(int s, int level, int optname, const void *optval, socklen_t optlen);
  268. static int free_socket_locked(struct lwip_sock *sock, int is_tcp, struct netconn **conn,
  269. union lwip_sock_lastdata *lastdata);
  270. static void free_socket_free_elements(int is_tcp, struct netconn *conn, union lwip_sock_lastdata *lastdata);
  271. #if LWIP_IPV4 && LWIP_IPV6
  272. static void
  273. sockaddr_to_ipaddr_port(const struct sockaddr *sockaddr, ip_addr_t *ipaddr, u16_t *port)
  274. {
  275. if ((sockaddr->sa_family) == AF_INET6) {
  276. SOCKADDR6_TO_IP6ADDR_PORT((const struct sockaddr_in6 *)(const void *)(sockaddr), ipaddr, *port);
  277. ipaddr->type = IPADDR_TYPE_V6;
  278. } else {
  279. SOCKADDR4_TO_IP4ADDR_PORT((const struct sockaddr_in *)(const void *)(sockaddr), ipaddr, *port);
  280. ipaddr->type = IPADDR_TYPE_V4;
  281. }
  282. }
  283. #endif /* LWIP_IPV4 && LWIP_IPV6 */
  284. /** LWIP_NETCONN_SEM_PER_THREAD==1: initialize thread-local semaphore */
  285. void
  286. lwip_socket_thread_init(void)
  287. {
  288. netconn_thread_init();
  289. }
  290. /** LWIP_NETCONN_SEM_PER_THREAD==1: destroy thread-local semaphore */
  291. void
  292. lwip_socket_thread_cleanup(void)
  293. {
  294. netconn_thread_cleanup();
  295. }
  296. #if LWIP_NETCONN_FULLDUPLEX
  297. /* Thread-safe increment of sock->fd_used, with overflow check */
  298. static int
  299. sock_inc_used(struct lwip_sock *sock)
  300. {
  301. int ret;
  302. SYS_ARCH_DECL_PROTECT(lev);
  303. LWIP_ASSERT("sock != NULL", sock != NULL);
  304. SYS_ARCH_PROTECT(lev);
  305. if (sock->fd_free_pending) {
  306. /* prevent new usage of this socket if free is pending */
  307. ret = 0;
  308. } else {
  309. ++sock->fd_used;
  310. ret = 1;
  311. LWIP_ASSERT("sock->fd_used != 0", sock->fd_used != 0);
  312. }
  313. SYS_ARCH_UNPROTECT(lev);
  314. return ret;
  315. }
  316. /* Like sock_inc_used(), but called under SYS_ARCH_PROTECT lock. */
  317. static int
  318. sock_inc_used_locked(struct lwip_sock *sock)
  319. {
  320. LWIP_ASSERT("sock != NULL", sock != NULL);
  321. if (sock->fd_free_pending) {
  322. LWIP_ASSERT("sock->fd_used != 0", sock->fd_used != 0);
  323. return 0;
  324. }
  325. ++sock->fd_used;
  326. LWIP_ASSERT("sock->fd_used != 0", sock->fd_used != 0);
  327. return 1;
  328. }
  329. /* In full-duplex mode,sock->fd_used != 0 prevents a socket descriptor from being
  330. * released (and possibly reused) when used from more than one thread
  331. * (e.g. read-while-write or close-while-write, etc)
  332. * This function is called at the end of functions using (try)get_socket*().
  333. */
  334. static void
  335. done_socket(struct lwip_sock *sock)
  336. {
  337. int freed = 0;
  338. int is_tcp = 0;
  339. struct netconn *conn = NULL;
  340. union lwip_sock_lastdata lastdata;
  341. SYS_ARCH_DECL_PROTECT(lev);
  342. LWIP_ASSERT("sock != NULL", sock != NULL);
  343. SYS_ARCH_PROTECT(lev);
  344. LWIP_ASSERT("sock->fd_used > 0", sock->fd_used > 0);
  345. if (--sock->fd_used == 0) {
  346. if (sock->fd_free_pending) {
  347. /* free the socket */
  348. sock->fd_used = 1;
  349. is_tcp = sock->fd_free_pending & LWIP_SOCK_FD_FREE_TCP;
  350. freed = free_socket_locked(sock, is_tcp, &conn, &lastdata);
  351. }
  352. }
  353. SYS_ARCH_UNPROTECT(lev);
  354. if (freed) {
  355. free_socket_free_elements(is_tcp, conn, &lastdata);
  356. }
  357. }
  358. #else /* LWIP_NETCONN_FULLDUPLEX */
  359. #define sock_inc_used(sock) 1
  360. #define sock_inc_used_locked(sock) 1
  361. #define done_socket(sock)
  362. #endif /* LWIP_NETCONN_FULLDUPLEX */
  363. /* Translate a socket 'int' into a pointer (only fails if the index is invalid) */
  364. static struct lwip_sock *
  365. tryget_socket_unconn_nouse(int fd)
  366. {
  367. int s = fd - LWIP_SOCKET_OFFSET;
  368. if ((s < 0) || (s >= NUM_SOCKETS)) {
  369. LWIP_DEBUGF(SOCKETS_DEBUG, ("tryget_socket_unconn(%d): invalid\n", fd));
  370. return NULL;
  371. }
  372. return &sockets[s];
  373. }
  374. struct lwip_sock *
  375. lwip_socket_dbg_get_socket(int fd)
  376. {
  377. return tryget_socket_unconn_nouse(fd);
  378. }
  379. /* Translate a socket 'int' into a pointer (only fails if the index is invalid) */
  380. static struct lwip_sock *
  381. tryget_socket_unconn(int fd)
  382. {
  383. struct lwip_sock *ret = tryget_socket_unconn_nouse(fd);
  384. if (ret != NULL) {
  385. if (!sock_inc_used(ret)) {
  386. return NULL;
  387. }
  388. }
  389. return ret;
  390. }
  391. /* Like tryget_socket_unconn(), but called under SYS_ARCH_PROTECT lock. */
  392. static struct lwip_sock *
  393. tryget_socket_unconn_locked(int fd)
  394. {
  395. struct lwip_sock *ret = tryget_socket_unconn_nouse(fd);
  396. if (ret != NULL) {
  397. if (!sock_inc_used_locked(ret)) {
  398. return NULL;
  399. }
  400. }
  401. return ret;
  402. }
  403. /**
  404. * Same as get_socket but doesn't set errno
  405. *
  406. * @param fd externally used socket index
  407. * @return struct lwip_sock for the socket or NULL if not found
  408. */
  409. static struct lwip_sock *
  410. tryget_socket(int fd)
  411. {
  412. struct lwip_sock *sock = tryget_socket_unconn(fd);
  413. if (sock != NULL) {
  414. if (sock->conn) {
  415. return sock;
  416. }
  417. done_socket(sock);
  418. }
  419. return NULL;
  420. }
  421. /**
  422. * Same as tryget_socket but a global routine.
  423. *
  424. * @param fd externally used socket index
  425. * @return struct lwip_sock for the socket or NULL if not found
  426. */
  427. struct lwip_sock *
  428. lwip_tryget_socket(int fd)
  429. {
  430. struct lwip_sock *sock = tryget_socket_unconn(fd);
  431. if (sock != NULL) {
  432. if (sock->conn) {
  433. return sock;
  434. }
  435. done_socket(sock);
  436. }
  437. return NULL;
  438. }
  439. /**
  440. * Map a externally used socket index to the internal socket representation.
  441. *
  442. * @param fd externally used socket index
  443. * @return struct lwip_sock for the socket or NULL if not found
  444. */
  445. static struct lwip_sock *
  446. get_socket(int fd)
  447. {
  448. struct lwip_sock *sock = tryget_socket(fd);
  449. if (!sock) {
  450. if ((fd < LWIP_SOCKET_OFFSET) || (fd >= (LWIP_SOCKET_OFFSET + NUM_SOCKETS))) {
  451. LWIP_DEBUGF(SOCKETS_DEBUG, ("get_socket(%d): invalid\n", fd));
  452. }
  453. set_errno(EBADF);
  454. return NULL;
  455. }
  456. return sock;
  457. }
  458. /**
  459. * Allocate a new socket for a given netconn.
  460. *
  461. * @param newconn the netconn for which to allocate a socket
  462. * @param accepted 1 if socket has been created by accept(),
  463. * 0 if socket has been created by socket()
  464. * @return the index of the new socket; -1 on error
  465. */
  466. static int
  467. alloc_socket(struct netconn *newconn, int accepted)
  468. {
  469. int i;
  470. SYS_ARCH_DECL_PROTECT(lev);
  471. LWIP_UNUSED_ARG(accepted);
  472. /* allocate a new socket identifier */
  473. for (i = 0; i < NUM_SOCKETS; ++i) {
  474. /* Protect socket array */
  475. SYS_ARCH_PROTECT(lev);
  476. if (!sockets[i].conn) {
  477. #if LWIP_NETCONN_FULLDUPLEX
  478. if (sockets[i].fd_used) {
  479. SYS_ARCH_UNPROTECT(lev);
  480. continue;
  481. }
  482. sockets[i].fd_used = 1;
  483. sockets[i].fd_free_pending = 0;
  484. #endif
  485. sockets[i].conn = newconn;
  486. /* The socket is not yet known to anyone, so no need to protect
  487. after having marked it as used. */
  488. SYS_ARCH_UNPROTECT(lev);
  489. sockets[i].lastdata.pbuf = NULL;
  490. #if LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL
  491. LWIP_ASSERT("sockets[i].select_waiting == 0", sockets[i].select_waiting == 0);
  492. sockets[i].rcvevent = 0;
  493. /* TCP sendbuf is empty, but the socket is not yet writable until connected
  494. * (unless it has been created by accept()). */
  495. sockets[i].sendevent = (NETCONNTYPE_GROUP(newconn->type) == NETCONN_TCP ? (accepted != 0) : 1);
  496. sockets[i].errevent = 0;
  497. #endif /* LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL */
  498. #ifdef SAL_USING_POSIX
  499. rt_wqueue_init(&sockets[i].wait_head);
  500. #endif
  501. return i + LWIP_SOCKET_OFFSET;
  502. }
  503. SYS_ARCH_UNPROTECT(lev);
  504. }
  505. return -1;
  506. }
  507. /** Free a socket (under lock)
  508. *
  509. * @param sock the socket to free
  510. * @param is_tcp != 0 for TCP sockets, used to free lastdata
  511. * @param conn the socekt's netconn is stored here, must be freed externally
  512. * @param lastdata lastdata is stored here, must be freed externally
  513. */
  514. static int
  515. free_socket_locked(struct lwip_sock *sock, int is_tcp, struct netconn **conn,
  516. union lwip_sock_lastdata *lastdata)
  517. {
  518. #if LWIP_NETCONN_FULLDUPLEX
  519. LWIP_ASSERT("sock->fd_used > 0", sock->fd_used > 0);
  520. sock->fd_used--;
  521. if (sock->fd_used > 0) {
  522. sock->fd_free_pending = LWIP_SOCK_FD_FREE_FREE | (is_tcp ? LWIP_SOCK_FD_FREE_TCP : 0);
  523. return 0;
  524. }
  525. #else /* LWIP_NETCONN_FULLDUPLEX */
  526. LWIP_UNUSED_ARG(is_tcp);
  527. #endif /* LWIP_NETCONN_FULLDUPLEX */
  528. *lastdata = sock->lastdata;
  529. sock->lastdata.pbuf = NULL;
  530. *conn = sock->conn;
  531. sock->conn = NULL;
  532. return 1;
  533. }
  534. /** Free a socket's leftover members.
  535. */
  536. static void
  537. free_socket_free_elements(int is_tcp, struct netconn *conn, union lwip_sock_lastdata *lastdata)
  538. {
  539. if (lastdata->pbuf != NULL) {
  540. if (is_tcp) {
  541. pbuf_free(lastdata->pbuf);
  542. } else {
  543. netbuf_delete(lastdata->netbuf);
  544. }
  545. }
  546. if (conn != NULL) {
  547. /* netconn_prepare_delete() has already been called, here we only free the conn */
  548. netconn_delete(conn);
  549. }
  550. }
  551. /** Free a socket. The socket's netconn must have been
  552. * delete before!
  553. *
  554. * @param sock the socket to free
  555. * @param is_tcp != 0 for TCP sockets, used to free lastdata
  556. */
  557. static void
  558. free_socket(struct lwip_sock *sock, int is_tcp)
  559. {
  560. int freed;
  561. struct netconn *conn;
  562. union lwip_sock_lastdata lastdata;
  563. SYS_ARCH_DECL_PROTECT(lev);
  564. /* Protect socket array */
  565. SYS_ARCH_PROTECT(lev);
  566. freed = free_socket_locked(sock, is_tcp, &conn, &lastdata);
  567. SYS_ARCH_UNPROTECT(lev);
  568. /* don't use 'sock' after this line, as another task might have allocated it */
  569. if (freed) {
  570. free_socket_free_elements(is_tcp, conn, &lastdata);
  571. }
  572. }
  573. /* Below this, the well-known socket functions are implemented.
  574. * Use google.com or opengroup.org to get a good description :-)
  575. *
  576. * Exceptions are documented!
  577. */
  578. int
  579. lwip_accept(int s, struct sockaddr *addr, socklen_t *addrlen)
  580. {
  581. struct lwip_sock *sock, *nsock;
  582. struct netconn *newconn;
  583. ip_addr_t naddr;
  584. u16_t port = 0;
  585. int newsock;
  586. err_t err;
  587. int recvevent;
  588. SYS_ARCH_DECL_PROTECT(lev);
  589. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_accept(%d)...\n", s));
  590. sock = get_socket(s);
  591. if (!sock) {
  592. return -1;
  593. }
  594. /* wait for a new connection */
  595. err = netconn_accept(sock->conn, &newconn);
  596. if (err != ERR_OK) {
  597. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_accept(%d): netconn_acept failed, err=%d\n", s, err));
  598. if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) != NETCONN_TCP) {
  599. sock_set_errno(sock, EOPNOTSUPP);
  600. } else if (err == ERR_CLSD) {
  601. sock_set_errno(sock, EINVAL);
  602. } else {
  603. sock_set_errno(sock, err_to_errno(err));
  604. }
  605. done_socket(sock);
  606. return -1;
  607. }
  608. LWIP_ASSERT("newconn != NULL", newconn != NULL);
  609. newsock = alloc_socket(newconn, 1);
  610. if (newsock == -1) {
  611. netconn_delete(newconn);
  612. sock_set_errno(sock, ENFILE);
  613. done_socket(sock);
  614. return -1;
  615. }
  616. LWIP_ASSERT("invalid socket index", (newsock >= LWIP_SOCKET_OFFSET) && (newsock < NUM_SOCKETS + LWIP_SOCKET_OFFSET));
  617. nsock = &sockets[newsock - LWIP_SOCKET_OFFSET];
  618. /* See event_callback: If data comes in right away after an accept, even
  619. * though the server task might not have created a new socket yet.
  620. * In that case, newconn->socket is counted down (newconn->socket--),
  621. * so nsock->rcvevent is >= 1 here!
  622. */
  623. SYS_ARCH_PROTECT(lev);
  624. recvevent = (s16_t)(-1 - newconn->socket);
  625. newconn->socket = newsock;
  626. SYS_ARCH_UNPROTECT(lev);
  627. if (newconn->callback) {
  628. LOCK_TCPIP_CORE();
  629. while (recvevent > 0) {
  630. recvevent--;
  631. newconn->callback(newconn, NETCONN_EVT_RCVPLUS, 0);
  632. }
  633. UNLOCK_TCPIP_CORE();
  634. }
  635. /* Note that POSIX only requires us to check addr is non-NULL. addrlen must
  636. * not be NULL if addr is valid.
  637. */
  638. if ((addr != NULL) && (addrlen != NULL)) {
  639. union sockaddr_aligned tempaddr;
  640. /* get the IP address and port of the remote host */
  641. err = netconn_peer(newconn, &naddr, &port);
  642. if (err != ERR_OK) {
  643. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_accept(%d): netconn_peer failed, err=%d\n", s, err));
  644. netconn_delete(newconn);
  645. free_socket(nsock, 1);
  646. sock_set_errno(sock, err_to_errno(err));
  647. done_socket(sock);
  648. return -1;
  649. }
  650. IPADDR_PORT_TO_SOCKADDR(&tempaddr, &naddr, port);
  651. if (*addrlen > tempaddr.sa.sa_len) {
  652. *addrlen = tempaddr.sa.sa_len;
  653. }
  654. MEMCPY(addr, &tempaddr, *addrlen);
  655. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_accept(%d) returning new sock=%d addr=", s, newsock));
  656. ip_addr_debug_print_val(SOCKETS_DEBUG, naddr);
  657. LWIP_DEBUGF(SOCKETS_DEBUG, (" port=%"U16_F"\n", port));
  658. } else {
  659. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_accept(%d) returning new sock=%d", s, newsock));
  660. }
  661. sock_set_errno(sock, 0);
  662. done_socket(sock);
  663. done_socket(nsock);
  664. return newsock;
  665. }
  666. int
  667. lwip_bind(int s, const struct sockaddr *name, socklen_t namelen)
  668. {
  669. struct lwip_sock *sock;
  670. ip_addr_t local_addr;
  671. u16_t local_port;
  672. err_t err;
  673. sock = get_socket(s);
  674. if (!sock) {
  675. return -1;
  676. }
  677. if (!SOCK_ADDR_TYPE_MATCH(name, sock)) {
  678. /* sockaddr does not match socket type (IPv4/IPv6) */
  679. sock_set_errno(sock, err_to_errno(ERR_VAL));
  680. done_socket(sock);
  681. return -1;
  682. }
  683. /* check size, family and alignment of 'name' */
  684. LWIP_ERROR("lwip_bind: invalid address", (IS_SOCK_ADDR_LEN_VALID(namelen) &&
  685. IS_SOCK_ADDR_TYPE_VALID(name) && IS_SOCK_ADDR_ALIGNED(name)),
  686. sock_set_errno(sock, err_to_errno(ERR_ARG)); done_socket(sock); return -1;);
  687. LWIP_UNUSED_ARG(namelen);
  688. SOCKADDR_TO_IPADDR_PORT(name, &local_addr, local_port);
  689. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_bind(%d, addr=", s));
  690. ip_addr_debug_print_val(SOCKETS_DEBUG, local_addr);
  691. LWIP_DEBUGF(SOCKETS_DEBUG, (" port=%"U16_F")\n", local_port));
  692. #if LWIP_IPV4 && LWIP_IPV6
  693. /* Dual-stack: Unmap IPv4 mapped IPv6 addresses */
  694. if (IP_IS_V6_VAL(local_addr) && ip6_addr_isipv4mappedipv6(ip_2_ip6(&local_addr))) {
  695. unmap_ipv4_mapped_ipv6(ip_2_ip4(&local_addr), ip_2_ip6(&local_addr));
  696. IP_SET_TYPE_VAL(local_addr, IPADDR_TYPE_V4);
  697. }
  698. #endif /* LWIP_IPV4 && LWIP_IPV6 */
  699. err = netconn_bind(sock->conn, &local_addr, local_port);
  700. if (err != ERR_OK) {
  701. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_bind(%d) failed, err=%d\n", s, err));
  702. sock_set_errno(sock, err_to_errno(err));
  703. done_socket(sock);
  704. return -1;
  705. }
  706. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_bind(%d) succeeded\n", s));
  707. sock_set_errno(sock, 0);
  708. done_socket(sock);
  709. return 0;
  710. }
  711. int
  712. lwip_close(int s)
  713. {
  714. struct lwip_sock *sock;
  715. int is_tcp = 0;
  716. err_t err;
  717. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_close(%d)\n", s));
  718. sock = get_socket(s);
  719. if (!sock) {
  720. return -1;
  721. }
  722. if (sock->conn != NULL) {
  723. is_tcp = NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_TCP;
  724. } else {
  725. LWIP_ASSERT("sock->lastdata == NULL", sock->lastdata.pbuf == NULL);
  726. }
  727. #if LWIP_IGMP
  728. /* drop all possibly joined IGMP memberships */
  729. lwip_socket_drop_registered_memberships(s);
  730. #endif /* LWIP_IGMP */
  731. #if LWIP_IPV6_MLD
  732. /* drop all possibly joined MLD6 memberships */
  733. lwip_socket_drop_registered_mld6_memberships(s);
  734. #endif /* LWIP_IPV6_MLD */
  735. err = netconn_prepare_delete(sock->conn);
  736. if (err != ERR_OK) {
  737. sock_set_errno(sock, err_to_errno(err));
  738. done_socket(sock);
  739. return -1;
  740. }
  741. free_socket(sock, is_tcp);
  742. set_errno(0);
  743. return 0;
  744. }
  745. int
  746. lwip_connect(int s, const struct sockaddr *name, socklen_t namelen)
  747. {
  748. struct lwip_sock *sock;
  749. err_t err;
  750. sock = get_socket(s);
  751. if (!sock) {
  752. return -1;
  753. }
  754. if (!SOCK_ADDR_TYPE_MATCH_OR_UNSPEC(name, sock)) {
  755. /* sockaddr does not match socket type (IPv4/IPv6) */
  756. sock_set_errno(sock, err_to_errno(ERR_VAL));
  757. done_socket(sock);
  758. return -1;
  759. }
  760. LWIP_UNUSED_ARG(namelen);
  761. if (name->sa_family == AF_UNSPEC) {
  762. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_connect(%d, AF_UNSPEC)\n", s));
  763. err = netconn_disconnect(sock->conn);
  764. } else {
  765. ip_addr_t remote_addr;
  766. u16_t remote_port;
  767. /* check size, family and alignment of 'name' */
  768. LWIP_ERROR("lwip_connect: invalid address", IS_SOCK_ADDR_LEN_VALID(namelen) &&
  769. IS_SOCK_ADDR_TYPE_VALID_OR_UNSPEC(name) && IS_SOCK_ADDR_ALIGNED(name),
  770. sock_set_errno(sock, err_to_errno(ERR_ARG)); done_socket(sock); return -1;);
  771. SOCKADDR_TO_IPADDR_PORT(name, &remote_addr, remote_port);
  772. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_connect(%d, addr=", s));
  773. ip_addr_debug_print_val(SOCKETS_DEBUG, remote_addr);
  774. LWIP_DEBUGF(SOCKETS_DEBUG, (" port=%"U16_F")\n", remote_port));
  775. #if LWIP_IPV4 && LWIP_IPV6
  776. /* Dual-stack: Unmap IPv4 mapped IPv6 addresses */
  777. if (IP_IS_V6_VAL(remote_addr) && ip6_addr_isipv4mappedipv6(ip_2_ip6(&remote_addr))) {
  778. unmap_ipv4_mapped_ipv6(ip_2_ip4(&remote_addr), ip_2_ip6(&remote_addr));
  779. IP_SET_TYPE_VAL(remote_addr, IPADDR_TYPE_V4);
  780. }
  781. #endif /* LWIP_IPV4 && LWIP_IPV6 */
  782. err = netconn_connect(sock->conn, &remote_addr, remote_port);
  783. }
  784. if (err != ERR_OK) {
  785. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_connect(%d) failed, err=%d\n", s, err));
  786. sock_set_errno(sock, err_to_errno(err));
  787. done_socket(sock);
  788. return -1;
  789. }
  790. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_connect(%d) succeeded\n", s));
  791. sock_set_errno(sock, 0);
  792. done_socket(sock);
  793. return 0;
  794. }
  795. /**
  796. * Set a socket into listen mode.
  797. * The socket may not have been used for another connection previously.
  798. *
  799. * @param s the socket to set to listening mode
  800. * @param backlog (ATTENTION: needs TCP_LISTEN_BACKLOG=1)
  801. * @return 0 on success, non-zero on failure
  802. */
  803. int
  804. lwip_listen(int s, int backlog)
  805. {
  806. struct lwip_sock *sock;
  807. err_t err;
  808. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_listen(%d, backlog=%d)\n", s, backlog));
  809. sock = get_socket(s);
  810. if (!sock) {
  811. return -1;
  812. }
  813. /* limit the "backlog" parameter to fit in an u8_t */
  814. backlog = LWIP_MIN(LWIP_MAX(backlog, 0), 0xff);
  815. err = netconn_listen_with_backlog(sock->conn, (u8_t)backlog);
  816. if (err != ERR_OK) {
  817. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_listen(%d) failed, err=%d\n", s, err));
  818. if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) != NETCONN_TCP) {
  819. sock_set_errno(sock, EOPNOTSUPP);
  820. } else {
  821. sock_set_errno(sock, err_to_errno(err));
  822. }
  823. done_socket(sock);
  824. return -1;
  825. }
  826. sock_set_errno(sock, 0);
  827. done_socket(sock);
  828. return 0;
  829. }
  830. #if LWIP_TCP
  831. /* Helper function to loop over receiving pbufs from netconn
  832. * until "len" bytes are received or we're otherwise done.
  833. * Keeps sock->lastdata for peeking or partly copying.
  834. */
  835. static ssize_t
  836. lwip_recv_tcp(struct lwip_sock *sock, void *mem, size_t len, int flags)
  837. {
  838. u8_t apiflags = NETCONN_NOAUTORCVD;
  839. ssize_t recvd = 0;
  840. ssize_t recv_left = (len <= SSIZE_MAX) ? (ssize_t)len : SSIZE_MAX;
  841. LWIP_ASSERT("no socket given", sock != NULL);
  842. LWIP_ASSERT("this should be checked internally", NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_TCP);
  843. if (flags & MSG_DONTWAIT) {
  844. apiflags |= NETCONN_DONTBLOCK;
  845. }
  846. do {
  847. struct pbuf *p;
  848. err_t err;
  849. u16_t copylen;
  850. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recv_tcp: top while sock->lastdata=%p\n", (void *)sock->lastdata.pbuf));
  851. /* Check if there is data left from the last recv operation. */
  852. if (sock->lastdata.pbuf) {
  853. p = sock->lastdata.pbuf;
  854. } else {
  855. /* No data was left from the previous operation, so we try to get
  856. some from the network. */
  857. err = netconn_recv_tcp_pbuf_flags(sock->conn, &p, apiflags);
  858. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recv_tcp: netconn_recv err=%d, pbuf=%p\n",
  859. err, (void *)p));
  860. if (err != ERR_OK) {
  861. if (recvd > 0) {
  862. /* already received data, return that (this trusts in getting the same error from
  863. netconn layer again next time netconn_recv is called) */
  864. goto lwip_recv_tcp_done;
  865. }
  866. /* We should really do some error checking here. */
  867. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recv_tcp: p == NULL, error is \"%s\"!\n",
  868. lwip_strerr(err)));
  869. sock_set_errno(sock, err_to_errno(err));
  870. if (err == ERR_CLSD) {
  871. return 0;
  872. } else {
  873. return -1;
  874. }
  875. }
  876. LWIP_ASSERT("p != NULL", p != NULL);
  877. sock->lastdata.pbuf = p;
  878. }
  879. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recv_tcp: buflen=%"U16_F" recv_left=%d off=%d\n",
  880. p->tot_len, (int)recv_left, (int)recvd));
  881. if (recv_left > p->tot_len) {
  882. copylen = p->tot_len;
  883. } else {
  884. copylen = (u16_t)recv_left;
  885. }
  886. if (recvd + copylen < recvd) {
  887. /* overflow */
  888. copylen = (u16_t)(SSIZE_MAX - recvd);
  889. }
  890. /* copy the contents of the received buffer into
  891. the supplied memory pointer mem */
  892. pbuf_copy_partial(p, (u8_t *)mem + recvd, copylen, 0);
  893. recvd += copylen;
  894. /* TCP combines multiple pbufs for one recv */
  895. LWIP_ASSERT("invalid copylen, len would underflow", recv_left >= copylen);
  896. recv_left -= copylen;
  897. /* Unless we peek the incoming message... */
  898. if ((flags & MSG_PEEK) == 0) {
  899. /* ... check if there is data left in the pbuf */
  900. LWIP_ASSERT("invalid copylen", p->tot_len >= copylen);
  901. if (p->tot_len - copylen > 0) {
  902. /* If so, it should be saved in the sock structure for the next recv call.
  903. We store the pbuf but hide/free the consumed data: */
  904. sock->lastdata.pbuf = pbuf_free_header(p, copylen);
  905. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recv_tcp: lastdata now pbuf=%p\n", (void *)sock->lastdata.pbuf));
  906. } else {
  907. sock->lastdata.pbuf = NULL;
  908. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recv_tcp: deleting pbuf=%p\n", (void *)p));
  909. pbuf_free(p);
  910. }
  911. }
  912. /* once we have some data to return, only add more if we don't need to wait */
  913. apiflags |= NETCONN_DONTBLOCK | NETCONN_NOFIN;
  914. /* @todo: do we need to support peeking more than one pbuf? */
  915. } while ((recv_left > 0) && !(flags & MSG_PEEK));
  916. lwip_recv_tcp_done:
  917. if ((recvd > 0) && !(flags & MSG_PEEK)) {
  918. /* ensure window update after copying all data */
  919. netconn_tcp_recvd(sock->conn, (size_t)recvd);
  920. }
  921. sock_set_errno(sock, 0);
  922. return recvd;
  923. }
  924. #endif
  925. /* Convert a netbuf's address data to struct sockaddr */
  926. static int
  927. lwip_sock_make_addr(struct netconn *conn, ip_addr_t *fromaddr, u16_t port,
  928. struct sockaddr *from, socklen_t *fromlen)
  929. {
  930. int truncated = 0;
  931. union sockaddr_aligned saddr;
  932. LWIP_UNUSED_ARG(conn);
  933. LWIP_ASSERT("fromaddr != NULL", fromaddr != NULL);
  934. LWIP_ASSERT("from != NULL", from != NULL);
  935. LWIP_ASSERT("fromlen != NULL", fromlen != NULL);
  936. #if LWIP_IPV4 && LWIP_IPV6
  937. /* Dual-stack: Map IPv4 addresses to IPv4 mapped IPv6 */
  938. if (NETCONNTYPE_ISIPV6(netconn_type(conn)) && IP_IS_V4(fromaddr)) {
  939. ip4_2_ipv4_mapped_ipv6(ip_2_ip6(fromaddr), ip_2_ip4(fromaddr));
  940. IP_SET_TYPE(fromaddr, IPADDR_TYPE_V6);
  941. }
  942. #endif /* LWIP_IPV4 && LWIP_IPV6 */
  943. IPADDR_PORT_TO_SOCKADDR(&saddr, fromaddr, port);
  944. if (*fromlen < saddr.sa.sa_len) {
  945. truncated = 1;
  946. } else if (*fromlen > saddr.sa.sa_len) {
  947. *fromlen = saddr.sa.sa_len;
  948. }
  949. MEMCPY(from, &saddr, *fromlen);
  950. return truncated;
  951. }
  952. #if LWIP_TCP
  953. /* Helper function to get a tcp socket's remote address info */
  954. static int
  955. lwip_recv_tcp_from(struct lwip_sock *sock, struct sockaddr *from, socklen_t *fromlen, const char *dbg_fn, int dbg_s, ssize_t dbg_ret)
  956. {
  957. if (sock == NULL) {
  958. return 0;
  959. }
  960. LWIP_UNUSED_ARG(dbg_fn);
  961. LWIP_UNUSED_ARG(dbg_s);
  962. LWIP_UNUSED_ARG(dbg_ret);
  963. #if !SOCKETS_DEBUG
  964. if (from && fromlen)
  965. #endif /* !SOCKETS_DEBUG */
  966. {
  967. /* get remote addr/port from tcp_pcb */
  968. u16_t port;
  969. ip_addr_t tmpaddr;
  970. netconn_getaddr(sock->conn, &tmpaddr, &port, 0);
  971. LWIP_DEBUGF(SOCKETS_DEBUG, ("%s(%d): addr=", dbg_fn, dbg_s));
  972. ip_addr_debug_print_val(SOCKETS_DEBUG, tmpaddr);
  973. LWIP_DEBUGF(SOCKETS_DEBUG, (" port=%"U16_F" len=%d\n", port, (int)dbg_ret));
  974. if (from && fromlen) {
  975. return lwip_sock_make_addr(sock->conn, &tmpaddr, port, from, fromlen);
  976. }
  977. }
  978. return 0;
  979. }
  980. #endif
  981. /* Helper function to receive a netbuf from a udp or raw netconn.
  982. * Keeps sock->lastdata for peeking.
  983. */
  984. static err_t
  985. lwip_recvfrom_udp_raw(struct lwip_sock *sock, int flags, struct msghdr *msg, u16_t *datagram_len, int dbg_s)
  986. {
  987. struct netbuf *buf;
  988. u8_t apiflags;
  989. err_t err;
  990. u16_t buflen, copylen, copied;
  991. int i;
  992. LWIP_UNUSED_ARG(dbg_s);
  993. LWIP_ERROR("lwip_recvfrom_udp_raw: invalid arguments", (msg->msg_iov != NULL) || (msg->msg_iovlen <= 0), return ERR_ARG;);
  994. if (flags & MSG_DONTWAIT) {
  995. apiflags = NETCONN_DONTBLOCK;
  996. } else {
  997. apiflags = 0;
  998. }
  999. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recvfrom_udp_raw[UDP/RAW]: top sock->lastdata=%p\n", (void *)sock->lastdata.netbuf));
  1000. /* Check if there is data left from the last recv operation. */
  1001. buf = sock->lastdata.netbuf;
  1002. if (buf == NULL) {
  1003. /* No data was left from the previous operation, so we try to get
  1004. some from the network. */
  1005. err = netconn_recv_udp_raw_netbuf_flags(sock->conn, &buf, apiflags);
  1006. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recvfrom_udp_raw[UDP/RAW]: netconn_recv err=%d, netbuf=%p\n",
  1007. err, (void *)buf));
  1008. if (err != ERR_OK) {
  1009. return err;
  1010. }
  1011. LWIP_ASSERT("buf != NULL", buf != NULL);
  1012. sock->lastdata.netbuf = buf;
  1013. }
  1014. buflen = buf->p->tot_len;
  1015. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recvfrom_udp_raw: buflen=%"U16_F"\n", buflen));
  1016. copied = 0;
  1017. /* copy the pbuf payload into the iovs */
  1018. for (i = 0; (i < msg->msg_iovlen) && (copied < buflen); i++) {
  1019. u16_t len_left = (u16_t)(buflen - copied);
  1020. if (msg->msg_iov[i].iov_len > len_left) {
  1021. copylen = len_left;
  1022. } else {
  1023. copylen = (u16_t)msg->msg_iov[i].iov_len;
  1024. }
  1025. /* copy the contents of the received buffer into
  1026. the supplied memory buffer */
  1027. pbuf_copy_partial(buf->p, (u8_t *)msg->msg_iov[i].iov_base, copylen, copied);
  1028. copied = (u16_t)(copied + copylen);
  1029. }
  1030. /* Check to see from where the data was.*/
  1031. #if !SOCKETS_DEBUG
  1032. if (msg->msg_name && msg->msg_namelen)
  1033. #endif /* !SOCKETS_DEBUG */
  1034. {
  1035. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recvfrom_udp_raw(%d): addr=", dbg_s));
  1036. ip_addr_debug_print_val(SOCKETS_DEBUG, *netbuf_fromaddr(buf));
  1037. LWIP_DEBUGF(SOCKETS_DEBUG, (" port=%"U16_F" len=%d\n", netbuf_fromport(buf), copied));
  1038. if (msg->msg_name && msg->msg_namelen) {
  1039. lwip_sock_make_addr(sock->conn, netbuf_fromaddr(buf), netbuf_fromport(buf),
  1040. (struct sockaddr *)msg->msg_name, &msg->msg_namelen);
  1041. }
  1042. }
  1043. /* Initialize flag output */
  1044. msg->msg_flags = 0;
  1045. if (msg->msg_control) {
  1046. u8_t wrote_msg = 0;
  1047. #if LWIP_NETBUF_RECVINFO
  1048. /* Check if packet info was recorded */
  1049. if (buf->flags & NETBUF_FLAG_DESTADDR) {
  1050. if (IP_IS_V4(&buf->toaddr)) {
  1051. #if LWIP_IPV4
  1052. if (msg->msg_controllen >= CMSG_SPACE(sizeof(struct in_pktinfo))) {
  1053. struct cmsghdr *chdr = CMSG_FIRSTHDR(msg); /* This will always return a header!! */
  1054. struct in_pktinfo *pkti = (struct in_pktinfo *)CMSG_DATA(chdr);
  1055. chdr->cmsg_level = IPPROTO_IP;
  1056. chdr->cmsg_type = IP_PKTINFO;
  1057. chdr->cmsg_len = CMSG_LEN(sizeof(struct in_pktinfo));
  1058. pkti->ipi_ifindex = buf->p->if_idx;
  1059. inet_addr_from_ip4addr(&pkti->ipi_addr, ip_2_ip4(netbuf_destaddr(buf)));
  1060. msg->msg_controllen = CMSG_SPACE(sizeof(struct in_pktinfo));
  1061. wrote_msg = 1;
  1062. } else {
  1063. msg->msg_flags |= MSG_CTRUNC;
  1064. }
  1065. #endif /* LWIP_IPV4 */
  1066. }
  1067. }
  1068. #endif /* LWIP_NETBUF_RECVINFO */
  1069. if (!wrote_msg) {
  1070. msg->msg_controllen = 0;
  1071. }
  1072. }
  1073. /* If we don't peek the incoming message: zero lastdata pointer and free the netbuf */
  1074. if ((flags & MSG_PEEK) == 0) {
  1075. sock->lastdata.netbuf = NULL;
  1076. netbuf_delete(buf);
  1077. }
  1078. if (datagram_len) {
  1079. *datagram_len = buflen;
  1080. }
  1081. return ERR_OK;
  1082. }
  1083. ssize_t
  1084. lwip_recvfrom(int s, void *mem, size_t len, int flags,
  1085. struct sockaddr *from, socklen_t *fromlen)
  1086. {
  1087. struct lwip_sock *sock;
  1088. ssize_t ret;
  1089. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recvfrom(%d, %p, %"SZT_F", 0x%x, ..)\n", s, mem, len, flags));
  1090. sock = get_socket(s);
  1091. if (!sock) {
  1092. return -1;
  1093. }
  1094. #if LWIP_TCP
  1095. if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_TCP) {
  1096. ret = lwip_recv_tcp(sock, mem, len, flags);
  1097. lwip_recv_tcp_from(sock, from, fromlen, "lwip_recvfrom", s, ret);
  1098. done_socket(sock);
  1099. return ret;
  1100. } else
  1101. #endif
  1102. {
  1103. u16_t datagram_len = 0;
  1104. struct iovec vec;
  1105. struct msghdr msg;
  1106. err_t err;
  1107. vec.iov_base = mem;
  1108. vec.iov_len = len;
  1109. msg.msg_control = NULL;
  1110. msg.msg_controllen = 0;
  1111. msg.msg_flags = 0;
  1112. msg.msg_iov = &vec;
  1113. msg.msg_iovlen = 1;
  1114. msg.msg_name = from;
  1115. msg.msg_namelen = (fromlen ? *fromlen : 0);
  1116. err = lwip_recvfrom_udp_raw(sock, flags, &msg, &datagram_len, s);
  1117. if (err != ERR_OK) {
  1118. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recvfrom[UDP/RAW](%d): buf == NULL, error is \"%s\"!\n",
  1119. s, lwip_strerr(err)));
  1120. sock_set_errno(sock, err_to_errno(err));
  1121. done_socket(sock);
  1122. return -1;
  1123. }
  1124. ret = (ssize_t)LWIP_MIN(LWIP_MIN(len, datagram_len), SSIZE_MAX);
  1125. if (fromlen) {
  1126. *fromlen = msg.msg_namelen;
  1127. }
  1128. }
  1129. sock_set_errno(sock, 0);
  1130. done_socket(sock);
  1131. return ret;
  1132. }
  1133. ssize_t
  1134. lwip_read(int s, void *mem, size_t len)
  1135. {
  1136. return lwip_recvfrom(s, mem, len, 0, NULL, NULL);
  1137. }
  1138. ssize_t
  1139. lwip_readv(int s, const struct iovec *iov, int iovcnt)
  1140. {
  1141. struct msghdr msg;
  1142. msg.msg_name = NULL;
  1143. msg.msg_namelen = 0;
  1144. /* Hack: we have to cast via number to cast from 'const' pointer to non-const.
  1145. Blame the opengroup standard for this inconsistency. */
  1146. msg.msg_iov = LWIP_CONST_CAST(struct iovec *, iov);
  1147. msg.msg_iovlen = iovcnt;
  1148. msg.msg_control = NULL;
  1149. msg.msg_controllen = 0;
  1150. msg.msg_flags = 0;
  1151. return lwip_recvmsg(s, &msg, 0);
  1152. }
  1153. ssize_t
  1154. lwip_recv(int s, void *mem, size_t len, int flags)
  1155. {
  1156. return lwip_recvfrom(s, mem, len, flags, NULL, NULL);
  1157. }
  1158. ssize_t
  1159. lwip_recvmsg(int s, struct msghdr *message, int flags)
  1160. {
  1161. struct lwip_sock *sock;
  1162. int i;
  1163. ssize_t buflen;
  1164. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recvmsg(%d, message=%p, flags=0x%x)\n", s, (void *)message, flags));
  1165. LWIP_ERROR("lwip_recvmsg: invalid message pointer", message != NULL, return ERR_ARG;);
  1166. LWIP_ERROR("lwip_recvmsg: unsupported flags", (flags & ~(MSG_PEEK|MSG_DONTWAIT)) == 0,
  1167. set_errno(EOPNOTSUPP); return -1;);
  1168. if ((message->msg_iovlen <= 0) || (message->msg_iovlen > IOV_MAX)) {
  1169. set_errno(EMSGSIZE);
  1170. return -1;
  1171. }
  1172. sock = get_socket(s);
  1173. if (!sock) {
  1174. return -1;
  1175. }
  1176. /* check for valid vectors */
  1177. buflen = 0;
  1178. for (i = 0; i < message->msg_iovlen; i++) {
  1179. if ((message->msg_iov[i].iov_base == NULL) || ((ssize_t)message->msg_iov[i].iov_len <= 0) ||
  1180. ((size_t)(ssize_t)message->msg_iov[i].iov_len != message->msg_iov[i].iov_len) ||
  1181. ((ssize_t)(buflen + (ssize_t)message->msg_iov[i].iov_len) <= 0)) {
  1182. sock_set_errno(sock, err_to_errno(ERR_VAL));
  1183. done_socket(sock);
  1184. return -1;
  1185. }
  1186. buflen = (ssize_t)(buflen + (ssize_t)message->msg_iov[i].iov_len);
  1187. }
  1188. if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_TCP) {
  1189. #if LWIP_TCP
  1190. int recv_flags = flags;
  1191. message->msg_flags = 0;
  1192. /* recv the data */
  1193. buflen = 0;
  1194. for (i = 0; i < message->msg_iovlen; i++) {
  1195. /* try to receive into this vector's buffer */
  1196. ssize_t recvd_local = lwip_recv_tcp(sock, message->msg_iov[i].iov_base, message->msg_iov[i].iov_len, recv_flags);
  1197. if (recvd_local > 0) {
  1198. /* sum up received bytes */
  1199. buflen += recvd_local;
  1200. }
  1201. if ((recvd_local < 0) || (recvd_local < (int)message->msg_iov[i].iov_len) ||
  1202. (flags & MSG_PEEK)) {
  1203. /* returned prematurely (or peeking, which might actually be limitated to the first iov) */
  1204. if (buflen <= 0) {
  1205. /* nothing received at all, propagate the error */
  1206. buflen = recvd_local;
  1207. }
  1208. break;
  1209. }
  1210. /* pass MSG_DONTWAIT to lwip_recv_tcp() to prevent waiting for more data */
  1211. recv_flags |= MSG_DONTWAIT;
  1212. }
  1213. if (buflen > 0) {
  1214. /* reset socket error since we have received something */
  1215. sock_set_errno(sock, 0);
  1216. }
  1217. /* " If the socket is connected, the msg_name and msg_namelen members shall be ignored." */
  1218. done_socket(sock);
  1219. return buflen;
  1220. #else /* LWIP_TCP */
  1221. sock_set_errno(sock, err_to_errno(ERR_ARG));
  1222. done_socket(sock);
  1223. return -1;
  1224. #endif /* LWIP_TCP */
  1225. }
  1226. /* else, UDP and RAW NETCONNs */
  1227. #if LWIP_UDP || LWIP_RAW
  1228. {
  1229. u16_t datagram_len = 0;
  1230. err_t err;
  1231. err = lwip_recvfrom_udp_raw(sock, flags, message, &datagram_len, s);
  1232. if (err != ERR_OK) {
  1233. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recvmsg[UDP/RAW](%d): buf == NULL, error is \"%s\"!\n",
  1234. s, lwip_strerr(err)));
  1235. sock_set_errno(sock, err_to_errno(err));
  1236. done_socket(sock);
  1237. return -1;
  1238. }
  1239. if (datagram_len > buflen) {
  1240. message->msg_flags |= MSG_TRUNC;
  1241. }
  1242. sock_set_errno(sock, 0);
  1243. done_socket(sock);
  1244. return (int)datagram_len;
  1245. }
  1246. #else /* LWIP_UDP || LWIP_RAW */
  1247. sock_set_errno(sock, err_to_errno(ERR_ARG));
  1248. done_socket(sock);
  1249. return -1;
  1250. #endif /* LWIP_UDP || LWIP_RAW */
  1251. }
  1252. ssize_t
  1253. lwip_send(int s, const void *data, size_t size, int flags)
  1254. {
  1255. struct lwip_sock *sock;
  1256. err_t err;
  1257. u8_t write_flags;
  1258. size_t written;
  1259. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_send(%d, data=%p, size=%"SZT_F", flags=0x%x)\n",
  1260. s, data, size, flags));
  1261. sock = get_socket(s);
  1262. if (!sock) {
  1263. return -1;
  1264. }
  1265. if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) != NETCONN_TCP) {
  1266. #if (LWIP_UDP || LWIP_RAW)
  1267. done_socket(sock);
  1268. return lwip_sendto(s, data, size, flags, NULL, 0);
  1269. #else /* (LWIP_UDP || LWIP_RAW) */
  1270. sock_set_errno(sock, err_to_errno(ERR_ARG));
  1271. done_socket(sock);
  1272. return -1;
  1273. #endif /* (LWIP_UDP || LWIP_RAW) */
  1274. }
  1275. write_flags = (u8_t)(NETCONN_COPY |
  1276. ((flags & MSG_MORE) ? NETCONN_MORE : 0) |
  1277. ((flags & MSG_DONTWAIT) ? NETCONN_DONTBLOCK : 0));
  1278. written = 0;
  1279. err = netconn_write_partly(sock->conn, data, size, write_flags, &written);
  1280. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_send(%d) err=%d written=%"SZT_F"\n", s, err, written));
  1281. sock_set_errno(sock, err_to_errno(err));
  1282. done_socket(sock);
  1283. /* casting 'written' to ssize_t is OK here since the netconn API limits it to SSIZE_MAX */
  1284. return (err == ERR_OK ? (ssize_t)written : -1);
  1285. }
  1286. ssize_t
  1287. lwip_sendmsg(int s, const struct msghdr *msg, int flags)
  1288. {
  1289. struct lwip_sock *sock;
  1290. #if LWIP_TCP
  1291. u8_t write_flags;
  1292. size_t written;
  1293. #endif
  1294. err_t err = ERR_OK;
  1295. sock = get_socket(s);
  1296. if (!sock) {
  1297. return -1;
  1298. }
  1299. LWIP_ERROR("lwip_sendmsg: invalid msghdr", msg != NULL,
  1300. sock_set_errno(sock, err_to_errno(ERR_ARG)); done_socket(sock); return -1;);
  1301. LWIP_ERROR("lwip_sendmsg: invalid msghdr iov", msg->msg_iov != NULL,
  1302. sock_set_errno(sock, err_to_errno(ERR_ARG)); done_socket(sock); return -1;);
  1303. LWIP_ERROR("lwip_sendmsg: maximum iovs exceeded", (msg->msg_iovlen > 0) && (msg->msg_iovlen <= IOV_MAX),
  1304. sock_set_errno(sock, EMSGSIZE); done_socket(sock); return -1;);
  1305. LWIP_ERROR("lwip_sendmsg: unsupported flags", (flags & ~(MSG_DONTWAIT | MSG_MORE)) == 0,
  1306. sock_set_errno(sock, EOPNOTSUPP); done_socket(sock); return -1;);
  1307. LWIP_UNUSED_ARG(msg->msg_control);
  1308. LWIP_UNUSED_ARG(msg->msg_controllen);
  1309. LWIP_UNUSED_ARG(msg->msg_flags);
  1310. if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_TCP) {
  1311. #if LWIP_TCP
  1312. write_flags = (u8_t)(NETCONN_COPY |
  1313. ((flags & MSG_MORE) ? NETCONN_MORE : 0) |
  1314. ((flags & MSG_DONTWAIT) ? NETCONN_DONTBLOCK : 0));
  1315. written = 0;
  1316. err = netconn_write_vectors_partly(sock->conn, (struct netvector *)msg->msg_iov, (u16_t)msg->msg_iovlen, write_flags, &written);
  1317. sock_set_errno(sock, err_to_errno(err));
  1318. done_socket(sock);
  1319. /* casting 'written' to ssize_t is OK here since the netconn API limits it to SSIZE_MAX */
  1320. return (err == ERR_OK ? (ssize_t)written : -1);
  1321. #else /* LWIP_TCP */
  1322. sock_set_errno(sock, err_to_errno(ERR_ARG));
  1323. done_socket(sock);
  1324. return -1;
  1325. #endif /* LWIP_TCP */
  1326. }
  1327. /* else, UDP and RAW NETCONNs */
  1328. #if LWIP_UDP || LWIP_RAW
  1329. {
  1330. struct netbuf chain_buf;
  1331. int i;
  1332. ssize_t size = 0;
  1333. LWIP_UNUSED_ARG(flags);
  1334. LWIP_ERROR("lwip_sendmsg: invalid msghdr name", (((msg->msg_name == NULL) && (msg->msg_namelen == 0)) ||
  1335. IS_SOCK_ADDR_LEN_VALID(msg->msg_namelen)),
  1336. sock_set_errno(sock, err_to_errno(ERR_ARG)); done_socket(sock); return -1;);
  1337. /* initialize chain buffer with destination */
  1338. memset(&chain_buf, 0, sizeof(struct netbuf));
  1339. if (msg->msg_name) {
  1340. u16_t remote_port;
  1341. SOCKADDR_TO_IPADDR_PORT((const struct sockaddr *)msg->msg_name, &chain_buf.addr, remote_port);
  1342. netbuf_fromport(&chain_buf) = remote_port;
  1343. }
  1344. #if LWIP_NETIF_TX_SINGLE_PBUF
  1345. for (i = 0; i < msg->msg_iovlen; i++) {
  1346. size += msg->msg_iov[i].iov_len;
  1347. if ((msg->msg_iov[i].iov_len > INT_MAX) || (size < (int)msg->msg_iov[i].iov_len)) {
  1348. /* overflow */
  1349. goto sendmsg_emsgsize;
  1350. }
  1351. }
  1352. if (size > 0xFFFF) {
  1353. /* overflow */
  1354. goto sendmsg_emsgsize;
  1355. }
  1356. /* Allocate a new netbuf and copy the data into it. */
  1357. if (netbuf_alloc(&chain_buf, (u16_t)size) == NULL) {
  1358. err = ERR_MEM;
  1359. } else {
  1360. /* flatten the IO vectors */
  1361. size_t offset = 0;
  1362. for (i = 0; i < msg->msg_iovlen; i++) {
  1363. MEMCPY(&((u8_t *)chain_buf.p->payload)[offset], msg->msg_iov[i].iov_base, msg->msg_iov[i].iov_len);
  1364. offset += msg->msg_iov[i].iov_len;
  1365. }
  1366. #if LWIP_CHECKSUM_ON_COPY
  1367. {
  1368. /* This can be improved by using LWIP_CHKSUM_COPY() and aggregating the checksum for each IO vector */
  1369. u16_t chksum = ~inet_chksum_pbuf(chain_buf.p);
  1370. netbuf_set_chksum(&chain_buf, chksum);
  1371. }
  1372. #endif /* LWIP_CHECKSUM_ON_COPY */
  1373. err = ERR_OK;
  1374. }
  1375. #else /* LWIP_NETIF_TX_SINGLE_PBUF */
  1376. /* create a chained netbuf from the IO vectors. NOTE: we assemble a pbuf chain
  1377. manually to avoid having to allocate, chain, and delete a netbuf for each iov */
  1378. for (i = 0; i < msg->msg_iovlen; i++) {
  1379. struct pbuf *p;
  1380. if (msg->msg_iov[i].iov_len > 0xFFFF) {
  1381. /* overflow */
  1382. goto sendmsg_emsgsize;
  1383. }
  1384. p = pbuf_alloc(PBUF_TRANSPORT, 0, PBUF_REF);
  1385. if (p == NULL) {
  1386. err = ERR_MEM; /* let netbuf_delete() cleanup chain_buf */
  1387. break;
  1388. }
  1389. p->payload = msg->msg_iov[i].iov_base;
  1390. p->len = p->tot_len = (u16_t)msg->msg_iov[i].iov_len;
  1391. /* netbuf empty, add new pbuf */
  1392. if (chain_buf.p == NULL) {
  1393. chain_buf.p = chain_buf.ptr = p;
  1394. /* add pbuf to existing pbuf chain */
  1395. } else {
  1396. if (chain_buf.p->tot_len + p->len > 0xffff) {
  1397. /* overflow */
  1398. pbuf_free(p);
  1399. goto sendmsg_emsgsize;
  1400. }
  1401. pbuf_cat(chain_buf.p, p);
  1402. }
  1403. }
  1404. /* save size of total chain */
  1405. if (err == ERR_OK) {
  1406. size = netbuf_len(&chain_buf);
  1407. }
  1408. #endif /* LWIP_NETIF_TX_SINGLE_PBUF */
  1409. if (err == ERR_OK) {
  1410. #if LWIP_IPV4 && LWIP_IPV6
  1411. /* Dual-stack: Unmap IPv4 mapped IPv6 addresses */
  1412. if (IP_IS_V6_VAL(chain_buf.addr) && ip6_addr_isipv4mappedipv6(ip_2_ip6(&chain_buf.addr))) {
  1413. unmap_ipv4_mapped_ipv6(ip_2_ip4(&chain_buf.addr), ip_2_ip6(&chain_buf.addr));
  1414. IP_SET_TYPE_VAL(chain_buf.addr, IPADDR_TYPE_V4);
  1415. }
  1416. #endif /* LWIP_IPV4 && LWIP_IPV6 */
  1417. /* send the data */
  1418. err = netconn_send(sock->conn, &chain_buf);
  1419. }
  1420. /* deallocated the buffer */
  1421. netbuf_free(&chain_buf);
  1422. sock_set_errno(sock, err_to_errno(err));
  1423. done_socket(sock);
  1424. return (err == ERR_OK ? size : -1);
  1425. sendmsg_emsgsize:
  1426. sock_set_errno(sock, EMSGSIZE);
  1427. netbuf_free(&chain_buf);
  1428. done_socket(sock);
  1429. return -1;
  1430. }
  1431. #else /* LWIP_UDP || LWIP_RAW */
  1432. sock_set_errno(sock, err_to_errno(ERR_ARG));
  1433. done_socket(sock);
  1434. return -1;
  1435. #endif /* LWIP_UDP || LWIP_RAW */
  1436. }
  1437. ssize_t
  1438. lwip_sendto(int s, const void *data, size_t size, int flags,
  1439. const struct sockaddr *to, socklen_t tolen)
  1440. {
  1441. struct lwip_sock *sock;
  1442. err_t err;
  1443. u16_t short_size;
  1444. u16_t remote_port;
  1445. struct netbuf buf;
  1446. sock = get_socket(s);
  1447. if (!sock) {
  1448. return -1;
  1449. }
  1450. if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_TCP) {
  1451. #if LWIP_TCP
  1452. done_socket(sock);
  1453. return lwip_send(s, data, size, flags);
  1454. #else /* LWIP_TCP */
  1455. LWIP_UNUSED_ARG(flags);
  1456. sock_set_errno(sock, err_to_errno(ERR_ARG));
  1457. done_socket(sock);
  1458. return -1;
  1459. #endif /* LWIP_TCP */
  1460. }
  1461. if (size > LWIP_MIN(0xFFFF, SSIZE_MAX)) {
  1462. /* cannot fit into one datagram (at least for us) */
  1463. sock_set_errno(sock, EMSGSIZE);
  1464. done_socket(sock);
  1465. return -1;
  1466. }
  1467. short_size = (u16_t)size;
  1468. LWIP_ERROR("lwip_sendto: invalid address", (((to == NULL) && (tolen == 0)) ||
  1469. (IS_SOCK_ADDR_LEN_VALID(tolen) &&
  1470. ((to != NULL) && (IS_SOCK_ADDR_TYPE_VALID(to) && IS_SOCK_ADDR_ALIGNED(to))))),
  1471. sock_set_errno(sock, err_to_errno(ERR_ARG)); done_socket(sock); return -1;);
  1472. LWIP_UNUSED_ARG(tolen);
  1473. /* initialize a buffer */
  1474. buf.p = buf.ptr = NULL;
  1475. #if LWIP_CHECKSUM_ON_COPY
  1476. buf.flags = 0;
  1477. #endif /* LWIP_CHECKSUM_ON_COPY */
  1478. if (to) {
  1479. SOCKADDR_TO_IPADDR_PORT(to, &buf.addr, remote_port);
  1480. } else {
  1481. remote_port = 0;
  1482. ip_addr_set_any(NETCONNTYPE_ISIPV6(netconn_type(sock->conn)), &buf.addr);
  1483. }
  1484. netbuf_fromport(&buf) = remote_port;
  1485. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_sendto(%d, data=%p, short_size=%"U16_F", flags=0x%x to=",
  1486. s, data, short_size, flags));
  1487. ip_addr_debug_print_val(SOCKETS_DEBUG, buf.addr);
  1488. LWIP_DEBUGF(SOCKETS_DEBUG, (" port=%"U16_F"\n", remote_port));
  1489. /* make the buffer point to the data that should be sent */
  1490. #if LWIP_NETIF_TX_SINGLE_PBUF
  1491. /* Allocate a new netbuf and copy the data into it. */
  1492. if (netbuf_alloc(&buf, short_size) == NULL) {
  1493. err = ERR_MEM;
  1494. } else {
  1495. #if LWIP_CHECKSUM_ON_COPY
  1496. if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) != NETCONN_RAW) {
  1497. u16_t chksum = LWIP_CHKSUM_COPY(buf.p->payload, data, short_size);
  1498. netbuf_set_chksum(&buf, chksum);
  1499. } else
  1500. #endif /* LWIP_CHECKSUM_ON_COPY */
  1501. {
  1502. MEMCPY(buf.p->payload, data, short_size);
  1503. }
  1504. err = ERR_OK;
  1505. }
  1506. #else /* LWIP_NETIF_TX_SINGLE_PBUF */
  1507. err = netbuf_ref(&buf, data, short_size);
  1508. #endif /* LWIP_NETIF_TX_SINGLE_PBUF */
  1509. if (err == ERR_OK) {
  1510. #if LWIP_IPV4 && LWIP_IPV6
  1511. /* Dual-stack: Unmap IPv4 mapped IPv6 addresses */
  1512. if (IP_IS_V6_VAL(buf.addr) && ip6_addr_isipv4mappedipv6(ip_2_ip6(&buf.addr))) {
  1513. unmap_ipv4_mapped_ipv6(ip_2_ip4(&buf.addr), ip_2_ip6(&buf.addr));
  1514. IP_SET_TYPE_VAL(buf.addr, IPADDR_TYPE_V4);
  1515. }
  1516. #endif /* LWIP_IPV4 && LWIP_IPV6 */
  1517. /* send the data */
  1518. err = netconn_send(sock->conn, &buf);
  1519. }
  1520. /* deallocated the buffer */
  1521. netbuf_free(&buf);
  1522. sock_set_errno(sock, err_to_errno(err));
  1523. done_socket(sock);
  1524. return (err == ERR_OK ? short_size : -1);
  1525. }
  1526. int
  1527. lwip_socket(int domain, int type, int protocol)
  1528. {
  1529. struct netconn *conn;
  1530. int i;
  1531. LWIP_UNUSED_ARG(domain); /* @todo: check this */
  1532. /* create a netconn */
  1533. switch (type) {
  1534. case SOCK_RAW:
  1535. conn = netconn_new_with_proto_and_callback(DOMAIN_TO_NETCONN_TYPE(domain, NETCONN_RAW),
  1536. (u8_t)protocol, DEFAULT_SOCKET_EVENTCB);
  1537. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_socket(%s, SOCK_RAW, %d) = ",
  1538. domain == PF_INET ? "PF_INET" : "UNKNOWN", protocol));
  1539. break;
  1540. case SOCK_DGRAM:
  1541. conn = netconn_new_with_callback(DOMAIN_TO_NETCONN_TYPE(domain,
  1542. ((protocol == IPPROTO_UDPLITE) ? NETCONN_UDPLITE : NETCONN_UDP)),
  1543. DEFAULT_SOCKET_EVENTCB);
  1544. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_socket(%s, SOCK_DGRAM, %d) = ",
  1545. domain == PF_INET ? "PF_INET" : "UNKNOWN", protocol));
  1546. #if LWIP_NETBUF_RECVINFO
  1547. if (conn) {
  1548. /* netconn layer enables pktinfo by default, sockets default to off */
  1549. conn->flags &= ~NETCONN_FLAG_PKTINFO;
  1550. }
  1551. #endif /* LWIP_NETBUF_RECVINFO */
  1552. break;
  1553. case SOCK_STREAM:
  1554. conn = netconn_new_with_callback(DOMAIN_TO_NETCONN_TYPE(domain, NETCONN_TCP), DEFAULT_SOCKET_EVENTCB);
  1555. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_socket(%s, SOCK_STREAM, %d) = ",
  1556. domain == PF_INET ? "PF_INET" : "UNKNOWN", protocol));
  1557. break;
  1558. default:
  1559. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_socket(%d, %d/UNKNOWN, %d) = -1\n",
  1560. domain, type, protocol));
  1561. set_errno(EINVAL);
  1562. return -1;
  1563. }
  1564. if (!conn) {
  1565. LWIP_DEBUGF(SOCKETS_DEBUG, ("-1 / ENOBUFS (could not create netconn)\n"));
  1566. set_errno(ENOBUFS);
  1567. return -1;
  1568. }
  1569. i = alloc_socket(conn, 0);
  1570. if (i == -1) {
  1571. netconn_delete(conn);
  1572. set_errno(ENFILE);
  1573. return -1;
  1574. }
  1575. conn->socket = i;
  1576. done_socket(&sockets[i - LWIP_SOCKET_OFFSET]);
  1577. LWIP_DEBUGF(SOCKETS_DEBUG, ("%d\n", i));
  1578. set_errno(0);
  1579. return i;
  1580. }
  1581. ssize_t
  1582. lwip_write(int s, const void *data, size_t size)
  1583. {
  1584. return lwip_send(s, data, size, 0);
  1585. }
  1586. ssize_t
  1587. lwip_writev(int s, const struct iovec *iov, int iovcnt)
  1588. {
  1589. struct msghdr msg;
  1590. msg.msg_name = NULL;
  1591. msg.msg_namelen = 0;
  1592. /* Hack: we have to cast via number to cast from 'const' pointer to non-const.
  1593. Blame the opengroup standard for this inconsistency. */
  1594. msg.msg_iov = LWIP_CONST_CAST(struct iovec *, iov);
  1595. msg.msg_iovlen = iovcnt;
  1596. msg.msg_control = NULL;
  1597. msg.msg_controllen = 0;
  1598. msg.msg_flags = 0;
  1599. return lwip_sendmsg(s, &msg, 0);
  1600. }
  1601. #if LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL
  1602. /* Add select_cb to select_cb_list. */
  1603. static void
  1604. lwip_link_select_cb(struct lwip_select_cb *select_cb)
  1605. {
  1606. LWIP_SOCKET_SELECT_DECL_PROTECT(lev);
  1607. /* Protect the select_cb_list */
  1608. LWIP_SOCKET_SELECT_PROTECT(lev);
  1609. /* Put this select_cb on top of list */
  1610. select_cb->next = select_cb_list;
  1611. if (select_cb_list != NULL) {
  1612. select_cb_list->prev = select_cb;
  1613. }
  1614. select_cb_list = select_cb;
  1615. #if !LWIP_TCPIP_CORE_LOCKING
  1616. /* Increasing this counter tells select_check_waiters that the list has changed. */
  1617. select_cb_ctr++;
  1618. #endif
  1619. /* Now we can safely unprotect */
  1620. LWIP_SOCKET_SELECT_UNPROTECT(lev);
  1621. }
  1622. /* Remove select_cb from select_cb_list. */
  1623. static void
  1624. lwip_unlink_select_cb(struct lwip_select_cb *select_cb)
  1625. {
  1626. LWIP_SOCKET_SELECT_DECL_PROTECT(lev);
  1627. /* Take us off the list */
  1628. LWIP_SOCKET_SELECT_PROTECT(lev);
  1629. if (select_cb->next != NULL) {
  1630. select_cb->next->prev = select_cb->prev;
  1631. }
  1632. if (select_cb_list == select_cb) {
  1633. LWIP_ASSERT("select_cb->prev == NULL", select_cb->prev == NULL);
  1634. select_cb_list = select_cb->next;
  1635. } else {
  1636. LWIP_ASSERT("select_cb->prev != NULL", select_cb->prev != NULL);
  1637. select_cb->prev->next = select_cb->next;
  1638. }
  1639. #if !LWIP_TCPIP_CORE_LOCKING
  1640. /* Increasing this counter tells select_check_waiters that the list has changed. */
  1641. select_cb_ctr++;
  1642. #endif
  1643. LWIP_SOCKET_SELECT_UNPROTECT(lev);
  1644. }
  1645. #endif /* LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL */
  1646. #if LWIP_SOCKET_SELECT
  1647. /**
  1648. * Go through the readset and writeset lists and see which socket of the sockets
  1649. * set in the sets has events. On return, readset, writeset and exceptset have
  1650. * the sockets enabled that had events.
  1651. *
  1652. * @param maxfdp1 the highest socket index in the sets
  1653. * @param readset_in set of sockets to check for read events
  1654. * @param writeset_in set of sockets to check for write events
  1655. * @param exceptset_in set of sockets to check for error events
  1656. * @param readset_out set of sockets that had read events
  1657. * @param writeset_out set of sockets that had write events
  1658. * @param exceptset_out set os sockets that had error events
  1659. * @return number of sockets that had events (read/write/exception) (>= 0)
  1660. */
  1661. static int
  1662. lwip_selscan(int maxfdp1, fd_set *readset_in, fd_set *writeset_in, fd_set *exceptset_in,
  1663. fd_set *readset_out, fd_set *writeset_out, fd_set *exceptset_out)
  1664. {
  1665. int i, nready = 0;
  1666. fd_set lreadset, lwriteset, lexceptset;
  1667. struct lwip_sock *sock;
  1668. SYS_ARCH_DECL_PROTECT(lev);
  1669. FD_ZERO(&lreadset);
  1670. FD_ZERO(&lwriteset);
  1671. FD_ZERO(&lexceptset);
  1672. /* Go through each socket in each list to count number of sockets which
  1673. currently match */
  1674. for (i = LWIP_SOCKET_OFFSET; i < maxfdp1; i++) {
  1675. /* if this FD is not in the set, continue */
  1676. if (!(readset_in && FD_ISSET(i, readset_in)) &&
  1677. !(writeset_in && FD_ISSET(i, writeset_in)) &&
  1678. !(exceptset_in && FD_ISSET(i, exceptset_in))) {
  1679. continue;
  1680. }
  1681. /* First get the socket's status (protected)... */
  1682. SYS_ARCH_PROTECT(lev);
  1683. sock = tryget_socket_unconn_locked(i);
  1684. if (sock != NULL) {
  1685. void *lastdata = sock->lastdata.pbuf;
  1686. s16_t rcvevent = sock->rcvevent;
  1687. u16_t sendevent = sock->sendevent;
  1688. u16_t errevent = sock->errevent;
  1689. SYS_ARCH_UNPROTECT(lev);
  1690. /* ... then examine it: */
  1691. /* See if netconn of this socket is ready for read */
  1692. if (readset_in && FD_ISSET(i, readset_in) && ((lastdata != NULL) || (rcvevent > 0))) {
  1693. FD_SET(i, &lreadset);
  1694. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_selscan: fd=%d ready for reading\n", i));
  1695. nready++;
  1696. }
  1697. /* See if netconn of this socket is ready for write */
  1698. if (writeset_in && FD_ISSET(i, writeset_in) && (sendevent != 0)) {
  1699. FD_SET(i, &lwriteset);
  1700. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_selscan: fd=%d ready for writing\n", i));
  1701. nready++;
  1702. }
  1703. /* See if netconn of this socket had an error */
  1704. if (exceptset_in && FD_ISSET(i, exceptset_in) && (errevent != 0)) {
  1705. FD_SET(i, &lexceptset);
  1706. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_selscan: fd=%d ready for exception\n", i));
  1707. nready++;
  1708. }
  1709. done_socket(sock);
  1710. } else {
  1711. SYS_ARCH_UNPROTECT(lev);
  1712. /* no a valid open socket */
  1713. return -1;
  1714. }
  1715. }
  1716. /* copy local sets to the ones provided as arguments */
  1717. *readset_out = lreadset;
  1718. *writeset_out = lwriteset;
  1719. *exceptset_out = lexceptset;
  1720. LWIP_ASSERT("nready >= 0", nready >= 0);
  1721. return nready;
  1722. }
  1723. #if LWIP_NETCONN_FULLDUPLEX
  1724. /* Mark all of the set sockets in one of the three fdsets passed to select as used.
  1725. * All sockets are marked (and later unmarked), whether they are open or not.
  1726. * This is OK as lwip_selscan aborts select when non-open sockets are found.
  1727. */
  1728. static void
  1729. lwip_select_inc_sockets_used_set(int maxfdp, fd_set *fdset, fd_set *used_sockets)
  1730. {
  1731. SYS_ARCH_DECL_PROTECT(lev);
  1732. if (fdset) {
  1733. int i;
  1734. for (i = LWIP_SOCKET_OFFSET; i < maxfdp; i++) {
  1735. /* if this FD is in the set, lock it (unless already done) */
  1736. if (FD_ISSET(i, fdset) && !FD_ISSET(i, used_sockets)) {
  1737. struct lwip_sock *sock;
  1738. SYS_ARCH_PROTECT(lev);
  1739. sock = tryget_socket_unconn_locked(i);
  1740. if (sock != NULL) {
  1741. /* leave the socket used until released by lwip_select_dec_sockets_used */
  1742. FD_SET(i, used_sockets);
  1743. }
  1744. SYS_ARCH_UNPROTECT(lev);
  1745. }
  1746. }
  1747. }
  1748. }
  1749. /* Mark all sockets passed to select as used to prevent them from being freed
  1750. * from other threads while select is running.
  1751. * Marked sockets are added to 'used_sockets' to mark them only once an be able
  1752. * to unmark them correctly.
  1753. */
  1754. static void
  1755. lwip_select_inc_sockets_used(int maxfdp, fd_set *fdset1, fd_set *fdset2, fd_set *fdset3, fd_set *used_sockets)
  1756. {
  1757. FD_ZERO(used_sockets);
  1758. lwip_select_inc_sockets_used_set(maxfdp, fdset1, used_sockets);
  1759. lwip_select_inc_sockets_used_set(maxfdp, fdset2, used_sockets);
  1760. lwip_select_inc_sockets_used_set(maxfdp, fdset3, used_sockets);
  1761. }
  1762. /* Let go all sockets that were marked as used when starting select */
  1763. static void
  1764. lwip_select_dec_sockets_used(int maxfdp, fd_set *used_sockets)
  1765. {
  1766. int i;
  1767. for (i = LWIP_SOCKET_OFFSET; i < maxfdp; i++) {
  1768. /* if this FD is not in the set, continue */
  1769. if (FD_ISSET(i, used_sockets)) {
  1770. struct lwip_sock *sock = tryget_socket_unconn_nouse(i);
  1771. LWIP_ASSERT("socket gone at the end of select", sock != NULL);
  1772. if (sock != NULL) {
  1773. done_socket(sock);
  1774. }
  1775. }
  1776. }
  1777. }
  1778. #else /* LWIP_NETCONN_FULLDUPLEX */
  1779. #define lwip_select_inc_sockets_used(maxfdp1, readset, writeset, exceptset, used_sockets)
  1780. #define lwip_select_dec_sockets_used(maxfdp1, used_sockets)
  1781. #endif /* LWIP_NETCONN_FULLDUPLEX */
  1782. int
  1783. lwip_select(int maxfdp1, fd_set *readset, fd_set *writeset, fd_set *exceptset,
  1784. struct timeval *timeout)
  1785. {
  1786. u32_t waitres = 0;
  1787. int nready;
  1788. fd_set lreadset, lwriteset, lexceptset;
  1789. u32_t msectimeout;
  1790. int i;
  1791. int maxfdp2;
  1792. #if LWIP_NETCONN_SEM_PER_THREAD
  1793. int waited = 0;
  1794. #endif
  1795. #if LWIP_NETCONN_FULLDUPLEX
  1796. fd_set used_sockets;
  1797. #endif
  1798. SYS_ARCH_DECL_PROTECT(lev);
  1799. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_select(%d, %p, %p, %p, tvsec=%"S32_F" tvusec=%"S32_F")\n",
  1800. maxfdp1, (void *)readset, (void *) writeset, (void *) exceptset,
  1801. timeout ? (s32_t)timeout->tv_sec : (s32_t) - 1,
  1802. timeout ? (s32_t)timeout->tv_usec : (s32_t) - 1));
  1803. if ((maxfdp1 < 0) || (maxfdp1 > LWIP_SELECT_MAXNFDS)) {
  1804. set_errno(EINVAL);
  1805. return -1;
  1806. }
  1807. lwip_select_inc_sockets_used(maxfdp1, readset, writeset, exceptset, &used_sockets);
  1808. /* Go through each socket in each list to count number of sockets which
  1809. currently match */
  1810. nready = lwip_selscan(maxfdp1, readset, writeset, exceptset, &lreadset, &lwriteset, &lexceptset);
  1811. if (nready < 0) {
  1812. /* one of the sockets in one of the fd_sets was invalid */
  1813. set_errno(EBADF);
  1814. lwip_select_dec_sockets_used(maxfdp1, &used_sockets);
  1815. return -1;
  1816. } else if (nready > 0) {
  1817. /* one or more sockets are set, no need to wait */
  1818. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_select: nready=%d\n", nready));
  1819. } else {
  1820. /* If we don't have any current events, then suspend if we are supposed to */
  1821. if (timeout && timeout->tv_sec == 0 && timeout->tv_usec == 0) {
  1822. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_select: no timeout, returning 0\n"));
  1823. /* This is OK as the local fdsets are empty and nready is zero,
  1824. or we would have returned earlier. */
  1825. } else {
  1826. /* None ready: add our semaphore to list:
  1827. We don't actually need any dynamic memory. Our entry on the
  1828. list is only valid while we are in this function, so it's ok
  1829. to use local variables (unless we're running in MPU compatible
  1830. mode). */
  1831. API_SELECT_CB_VAR_DECLARE(select_cb);
  1832. API_SELECT_CB_VAR_ALLOC(select_cb, set_errno(ENOMEM); lwip_select_dec_sockets_used(maxfdp1, &used_sockets); return -1);
  1833. memset(&API_SELECT_CB_VAR_REF(select_cb), 0, sizeof(struct lwip_select_cb));
  1834. API_SELECT_CB_VAR_REF(select_cb).readset = readset;
  1835. API_SELECT_CB_VAR_REF(select_cb).writeset = writeset;
  1836. API_SELECT_CB_VAR_REF(select_cb).exceptset = exceptset;
  1837. #if LWIP_NETCONN_SEM_PER_THREAD
  1838. API_SELECT_CB_VAR_REF(select_cb).sem = LWIP_NETCONN_THREAD_SEM_GET();
  1839. #else /* LWIP_NETCONN_SEM_PER_THREAD */
  1840. if (sys_sem_new(&API_SELECT_CB_VAR_REF(select_cb).sem, 0) != ERR_OK) {
  1841. /* failed to create semaphore */
  1842. set_errno(ENOMEM);
  1843. lwip_select_dec_sockets_used(maxfdp1, &used_sockets);
  1844. API_SELECT_CB_VAR_FREE(select_cb);
  1845. return -1;
  1846. }
  1847. #endif /* LWIP_NETCONN_SEM_PER_THREAD */
  1848. lwip_link_select_cb(&API_SELECT_CB_VAR_REF(select_cb));
  1849. /* Increase select_waiting for each socket we are interested in */
  1850. maxfdp2 = maxfdp1;
  1851. for (i = LWIP_SOCKET_OFFSET; i < maxfdp1; i++) {
  1852. if ((readset && FD_ISSET(i, readset)) ||
  1853. (writeset && FD_ISSET(i, writeset)) ||
  1854. (exceptset && FD_ISSET(i, exceptset))) {
  1855. struct lwip_sock *sock;
  1856. SYS_ARCH_PROTECT(lev);
  1857. sock = tryget_socket_unconn_locked(i);
  1858. if (sock != NULL) {
  1859. sock->select_waiting++;
  1860. if (sock->select_waiting == 0) {
  1861. /* overflow - too many threads waiting */
  1862. sock->select_waiting--;
  1863. nready = -1;
  1864. maxfdp2 = i;
  1865. SYS_ARCH_UNPROTECT(lev);
  1866. done_socket(sock);
  1867. set_errno(EBUSY);
  1868. break;
  1869. }
  1870. SYS_ARCH_UNPROTECT(lev);
  1871. done_socket(sock);
  1872. } else {
  1873. /* Not a valid socket */
  1874. nready = -1;
  1875. maxfdp2 = i;
  1876. SYS_ARCH_UNPROTECT(lev);
  1877. set_errno(EBADF);
  1878. break;
  1879. }
  1880. }
  1881. }
  1882. if (nready >= 0) {
  1883. /* Call lwip_selscan again: there could have been events between
  1884. the last scan (without us on the list) and putting us on the list! */
  1885. nready = lwip_selscan(maxfdp1, readset, writeset, exceptset, &lreadset, &lwriteset, &lexceptset);
  1886. if (!nready) {
  1887. /* Still none ready, just wait to be woken */
  1888. if (timeout == 0) {
  1889. /* Wait forever */
  1890. msectimeout = 0;
  1891. } else {
  1892. long msecs_long = ((timeout->tv_sec * 1000) + ((timeout->tv_usec + 500) / 1000));
  1893. if (msecs_long <= 0) {
  1894. /* Wait 1ms at least (0 means wait forever) */
  1895. msectimeout = 1;
  1896. } else {
  1897. msectimeout = (u32_t)msecs_long;
  1898. }
  1899. }
  1900. waitres = sys_arch_sem_wait(SELECT_SEM_PTR(API_SELECT_CB_VAR_REF(select_cb).sem), msectimeout);
  1901. #if LWIP_NETCONN_SEM_PER_THREAD
  1902. waited = 1;
  1903. #endif
  1904. }
  1905. }
  1906. /* Decrease select_waiting for each socket we are interested in */
  1907. for (i = LWIP_SOCKET_OFFSET; i < maxfdp2; i++) {
  1908. if ((readset && FD_ISSET(i, readset)) ||
  1909. (writeset && FD_ISSET(i, writeset)) ||
  1910. (exceptset && FD_ISSET(i, exceptset))) {
  1911. struct lwip_sock *sock;
  1912. SYS_ARCH_PROTECT(lev);
  1913. sock = tryget_socket_unconn_locked(i);
  1914. if (sock != NULL) {
  1915. /* for now, handle select_waiting==0... */
  1916. LWIP_ASSERT("sock->select_waiting > 0", sock->select_waiting > 0);
  1917. if (sock->select_waiting > 0) {
  1918. sock->select_waiting--;
  1919. }
  1920. SYS_ARCH_UNPROTECT(lev);
  1921. done_socket(sock);
  1922. } else {
  1923. SYS_ARCH_UNPROTECT(lev);
  1924. /* Not a valid socket */
  1925. nready = -1;
  1926. set_errno(EBADF);
  1927. }
  1928. }
  1929. }
  1930. lwip_unlink_select_cb(&API_SELECT_CB_VAR_REF(select_cb));
  1931. #if LWIP_NETCONN_SEM_PER_THREAD
  1932. if (API_SELECT_CB_VAR_REF(select_cb).sem_signalled && (!waited || (waitres == SYS_ARCH_TIMEOUT))) {
  1933. /* don't leave the thread-local semaphore signalled */
  1934. sys_arch_sem_wait(API_SELECT_CB_VAR_REF(select_cb).sem, 1);
  1935. }
  1936. #else /* LWIP_NETCONN_SEM_PER_THREAD */
  1937. sys_sem_free(&API_SELECT_CB_VAR_REF(select_cb).sem);
  1938. #endif /* LWIP_NETCONN_SEM_PER_THREAD */
  1939. API_SELECT_CB_VAR_FREE(select_cb);
  1940. if (nready < 0) {
  1941. /* This happens when a socket got closed while waiting */
  1942. lwip_select_dec_sockets_used(maxfdp1, &used_sockets);
  1943. return -1;
  1944. }
  1945. if (waitres == SYS_ARCH_TIMEOUT) {
  1946. /* Timeout */
  1947. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_select: timeout expired\n"));
  1948. /* This is OK as the local fdsets are empty and nready is zero,
  1949. or we would have returned earlier. */
  1950. } else {
  1951. /* See what's set now after waiting */
  1952. nready = lwip_selscan(maxfdp1, readset, writeset, exceptset, &lreadset, &lwriteset, &lexceptset);
  1953. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_select: nready=%d\n", nready));
  1954. }
  1955. }
  1956. }
  1957. lwip_select_dec_sockets_used(maxfdp1, &used_sockets);
  1958. set_errno(0);
  1959. if (readset) {
  1960. *readset = lreadset;
  1961. }
  1962. if (writeset) {
  1963. *writeset = lwriteset;
  1964. }
  1965. if (exceptset) {
  1966. *exceptset = lexceptset;
  1967. }
  1968. return nready;
  1969. }
  1970. #endif /* LWIP_SOCKET_SELECT */
  1971. #if LWIP_SOCKET_POLL
  1972. /** Options for the lwip_pollscan function. */
  1973. enum lwip_pollscan_opts
  1974. {
  1975. /** Clear revents in each struct pollfd. */
  1976. LWIP_POLLSCAN_CLEAR = 1,
  1977. /** Increment select_waiting in each struct lwip_sock. */
  1978. LWIP_POLLSCAN_INC_WAIT = 2,
  1979. /** Decrement select_waiting in each struct lwip_sock. */
  1980. LWIP_POLLSCAN_DEC_WAIT = 4
  1981. };
  1982. /**
  1983. * Update revents in each struct pollfd.
  1984. * Optionally update select_waiting in struct lwip_sock.
  1985. *
  1986. * @param fds array of structures to update
  1987. * @param nfds number of structures in fds
  1988. * @param opts what to update and how
  1989. * @return number of structures that have revents != 0
  1990. */
  1991. static int
  1992. lwip_pollscan(struct pollfd *fds, nfds_t nfds, enum lwip_pollscan_opts opts)
  1993. {
  1994. int nready = 0;
  1995. nfds_t fdi;
  1996. struct lwip_sock *sock;
  1997. SYS_ARCH_DECL_PROTECT(lev);
  1998. /* Go through each struct pollfd in the array. */
  1999. for (fdi = 0; fdi < nfds; fdi++) {
  2000. if ((opts & LWIP_POLLSCAN_CLEAR) != 0) {
  2001. fds[fdi].revents = 0;
  2002. }
  2003. /* Negative fd means the caller wants us to ignore this struct.
  2004. POLLNVAL means we already detected that the fd is invalid;
  2005. if another thread has since opened a new socket with that fd,
  2006. we must not use that socket. */
  2007. if (fds[fdi].fd >= 0 && (fds[fdi].revents & POLLNVAL) == 0) {
  2008. /* First get the socket's status (protected)... */
  2009. SYS_ARCH_PROTECT(lev);
  2010. sock = tryget_socket_unconn_locked(fds[fdi].fd);
  2011. if (sock != NULL) {
  2012. void* lastdata = sock->lastdata.pbuf;
  2013. s16_t rcvevent = sock->rcvevent;
  2014. u16_t sendevent = sock->sendevent;
  2015. u16_t errevent = sock->errevent;
  2016. if ((opts & LWIP_POLLSCAN_INC_WAIT) != 0) {
  2017. sock->select_waiting++;
  2018. if (sock->select_waiting == 0) {
  2019. /* overflow - too many threads waiting */
  2020. sock->select_waiting--;
  2021. nready = -1;
  2022. SYS_ARCH_UNPROTECT(lev);
  2023. done_socket(sock);
  2024. break;
  2025. }
  2026. } else if ((opts & LWIP_POLLSCAN_DEC_WAIT) != 0) {
  2027. /* for now, handle select_waiting==0... */
  2028. LWIP_ASSERT("sock->select_waiting > 0", sock->select_waiting > 0);
  2029. if (sock->select_waiting > 0) {
  2030. sock->select_waiting--;
  2031. }
  2032. }
  2033. SYS_ARCH_UNPROTECT(lev);
  2034. done_socket(sock);
  2035. /* ... then examine it: */
  2036. /* See if netconn of this socket is ready for read */
  2037. if ((fds[fdi].events & POLLIN) != 0 && ((lastdata != NULL) || (rcvevent > 0))) {
  2038. fds[fdi].revents |= POLLIN;
  2039. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_pollscan: fd=%d ready for reading\n", fds[fdi].fd));
  2040. }
  2041. /* See if netconn of this socket is ready for write */
  2042. if ((fds[fdi].events & POLLOUT) != 0 && (sendevent != 0)) {
  2043. fds[fdi].revents |= POLLOUT;
  2044. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_pollscan: fd=%d ready for writing\n", fds[fdi].fd));
  2045. }
  2046. /* See if netconn of this socket had an error */
  2047. if (errevent != 0) {
  2048. /* POLLERR is output only. */
  2049. fds[fdi].revents |= POLLERR;
  2050. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_pollscan: fd=%d ready for exception\n", fds[fdi].fd));
  2051. }
  2052. } else {
  2053. /* Not a valid socket */
  2054. SYS_ARCH_UNPROTECT(lev);
  2055. /* POLLNVAL is output only. */
  2056. fds[fdi].revents |= POLLNVAL;
  2057. return -1;
  2058. }
  2059. }
  2060. /* Will return the number of structures that have events,
  2061. not the number of events. */
  2062. if (fds[fdi].revents != 0) {
  2063. nready++;
  2064. }
  2065. }
  2066. LWIP_ASSERT("nready >= 0", nready >= 0);
  2067. return nready;
  2068. }
  2069. #if LWIP_NETCONN_FULLDUPLEX
  2070. /* Mark all sockets as used.
  2071. *
  2072. * All sockets are marked (and later unmarked), whether they are open or not.
  2073. * This is OK as lwip_pollscan aborts select when non-open sockets are found.
  2074. */
  2075. static void
  2076. lwip_poll_inc_sockets_used(struct pollfd *fds, nfds_t nfds)
  2077. {
  2078. nfds_t fdi;
  2079. if(fds) {
  2080. /* Go through each struct pollfd in the array. */
  2081. for (fdi = 0; fdi < nfds; fdi++) {
  2082. /* Increase the reference counter */
  2083. tryget_socket_unconn(fds[fdi].fd);
  2084. }
  2085. }
  2086. }
  2087. /* Let go all sockets that were marked as used when starting poll */
  2088. static void
  2089. lwip_poll_dec_sockets_used(struct pollfd *fds, nfds_t nfds)
  2090. {
  2091. nfds_t fdi;
  2092. if(fds) {
  2093. /* Go through each struct pollfd in the array. */
  2094. for (fdi = 0; fdi < nfds; fdi++) {
  2095. struct lwip_sock *sock = tryget_socket_unconn_nouse(fds[fdi].fd);
  2096. if (sock != NULL) {
  2097. done_socket(sock);
  2098. }
  2099. }
  2100. }
  2101. }
  2102. #else /* LWIP_NETCONN_FULLDUPLEX */
  2103. #define lwip_poll_inc_sockets_used(fds, nfds)
  2104. #define lwip_poll_dec_sockets_used(fds, nfds)
  2105. #endif /* LWIP_NETCONN_FULLDUPLEX */
  2106. int
  2107. lwip_poll(struct pollfd *fds, nfds_t nfds, int timeout)
  2108. {
  2109. u32_t waitres = 0;
  2110. int nready;
  2111. u32_t msectimeout;
  2112. #if LWIP_NETCONN_SEM_PER_THREAD
  2113. int waited = 0;
  2114. #endif
  2115. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_poll(%p, %d, %d)\n",
  2116. (void*)fds, (int)nfds, timeout));
  2117. LWIP_ERROR("lwip_poll: invalid fds", ((fds != NULL && nfds > 0) || (fds == NULL && nfds == 0)),
  2118. set_errno(EINVAL); return -1;);
  2119. lwip_poll_inc_sockets_used(fds, nfds);
  2120. /* Go through each struct pollfd to count number of structures
  2121. which currently match */
  2122. nready = lwip_pollscan(fds, nfds, LWIP_POLLSCAN_CLEAR);
  2123. if (nready < 0) {
  2124. lwip_poll_dec_sockets_used(fds, nfds);
  2125. return -1;
  2126. }
  2127. /* If we don't have any current events, then suspend if we are supposed to */
  2128. if (!nready) {
  2129. API_SELECT_CB_VAR_DECLARE(select_cb);
  2130. if (timeout == 0) {
  2131. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_poll: no timeout, returning 0\n"));
  2132. goto return_success;
  2133. }
  2134. API_SELECT_CB_VAR_ALLOC(select_cb, set_errno(EAGAIN); lwip_poll_dec_sockets_used(fds, nfds); return -1);
  2135. memset(&API_SELECT_CB_VAR_REF(select_cb), 0, sizeof(struct lwip_select_cb));
  2136. /* None ready: add our semaphore to list:
  2137. We don't actually need any dynamic memory. Our entry on the
  2138. list is only valid while we are in this function, so it's ok
  2139. to use local variables. */
  2140. API_SELECT_CB_VAR_REF(select_cb).poll_fds = fds;
  2141. API_SELECT_CB_VAR_REF(select_cb).poll_nfds = nfds;
  2142. #if LWIP_NETCONN_SEM_PER_THREAD
  2143. API_SELECT_CB_VAR_REF(select_cb).sem = LWIP_NETCONN_THREAD_SEM_GET();
  2144. #else /* LWIP_NETCONN_SEM_PER_THREAD */
  2145. if (sys_sem_new(&API_SELECT_CB_VAR_REF(select_cb).sem, 0) != ERR_OK) {
  2146. /* failed to create semaphore */
  2147. set_errno(EAGAIN);
  2148. lwip_poll_dec_sockets_used(fds, nfds);
  2149. API_SELECT_CB_VAR_FREE(select_cb);
  2150. return -1;
  2151. }
  2152. #endif /* LWIP_NETCONN_SEM_PER_THREAD */
  2153. lwip_link_select_cb(&API_SELECT_CB_VAR_REF(select_cb));
  2154. /* Increase select_waiting for each socket we are interested in.
  2155. Also, check for events again: there could have been events between
  2156. the last scan (without us on the list) and putting us on the list! */
  2157. nready = lwip_pollscan(fds, nfds, LWIP_POLLSCAN_INC_WAIT);
  2158. if (!nready) {
  2159. /* Still none ready, just wait to be woken */
  2160. if (timeout < 0) {
  2161. /* Wait forever */
  2162. msectimeout = 0;
  2163. } else {
  2164. /* timeout == 0 would have been handled earlier. */
  2165. LWIP_ASSERT("timeout > 0", timeout > 0);
  2166. msectimeout = timeout;
  2167. }
  2168. waitres = sys_arch_sem_wait(SELECT_SEM_PTR(API_SELECT_CB_VAR_REF(select_cb).sem), msectimeout);
  2169. #if LWIP_NETCONN_SEM_PER_THREAD
  2170. waited = 1;
  2171. #endif
  2172. }
  2173. /* Decrease select_waiting for each socket we are interested in,
  2174. and check which events occurred while we waited. */
  2175. nready = lwip_pollscan(fds, nfds, LWIP_POLLSCAN_DEC_WAIT);
  2176. lwip_unlink_select_cb(&API_SELECT_CB_VAR_REF(select_cb));
  2177. #if LWIP_NETCONN_SEM_PER_THREAD
  2178. if (select_cb.sem_signalled && (!waited || (waitres == SYS_ARCH_TIMEOUT))) {
  2179. /* don't leave the thread-local semaphore signalled */
  2180. sys_arch_sem_wait(API_SELECT_CB_VAR_REF(select_cb).sem, 1);
  2181. }
  2182. #else /* LWIP_NETCONN_SEM_PER_THREAD */
  2183. sys_sem_free(&API_SELECT_CB_VAR_REF(select_cb).sem);
  2184. #endif /* LWIP_NETCONN_SEM_PER_THREAD */
  2185. API_SELECT_CB_VAR_FREE(select_cb);
  2186. if (nready < 0) {
  2187. /* This happens when a socket got closed while waiting */
  2188. lwip_poll_dec_sockets_used(fds, nfds);
  2189. return -1;
  2190. }
  2191. if (waitres == SYS_ARCH_TIMEOUT) {
  2192. /* Timeout */
  2193. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_poll: timeout expired\n"));
  2194. goto return_success;
  2195. }
  2196. }
  2197. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_poll: nready=%d\n", nready));
  2198. return_success:
  2199. lwip_poll_dec_sockets_used(fds, nfds);
  2200. set_errno(0);
  2201. return nready;
  2202. }
  2203. /**
  2204. * Check whether event_callback should wake up a thread waiting in
  2205. * lwip_poll.
  2206. */
  2207. static int
  2208. lwip_poll_should_wake(const struct lwip_select_cb *scb, int fd, int has_recvevent, int has_sendevent, int has_errevent)
  2209. {
  2210. nfds_t fdi;
  2211. for (fdi = 0; fdi < scb->poll_nfds; fdi++) {
  2212. const struct pollfd *pollfd = &scb->poll_fds[fdi];
  2213. if (pollfd->fd == fd) {
  2214. /* Do not update pollfd->revents right here;
  2215. that would be a data race because lwip_pollscan
  2216. accesses revents without protecting. */
  2217. if (has_recvevent && (pollfd->events & POLLIN) != 0) {
  2218. return 1;
  2219. }
  2220. if (has_sendevent && (pollfd->events & POLLOUT) != 0) {
  2221. return 1;
  2222. }
  2223. if (has_errevent) {
  2224. /* POLLERR is output only. */
  2225. return 1;
  2226. }
  2227. }
  2228. }
  2229. return 0;
  2230. }
  2231. #endif /* LWIP_SOCKET_POLL */
  2232. #if LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL
  2233. /**
  2234. * Callback registered in the netconn layer for each socket-netconn.
  2235. * Processes recvevent (data available) and wakes up tasks waiting for select.
  2236. *
  2237. * @note for LWIP_TCPIP_CORE_LOCKING any caller of this function
  2238. * must have the core lock held when signaling the following events
  2239. * as they might cause select_list_cb to be checked:
  2240. * NETCONN_EVT_RCVPLUS
  2241. * NETCONN_EVT_SENDPLUS
  2242. * NETCONN_EVT_ERROR
  2243. * This requirement will be asserted in select_check_waiters()
  2244. */
  2245. static void
  2246. event_callback(struct netconn *conn, enum netconn_evt evt, u16_t len)
  2247. {
  2248. int s, check_waiters;
  2249. struct lwip_sock *sock;
  2250. SYS_ARCH_DECL_PROTECT(lev);
  2251. LWIP_UNUSED_ARG(len);
  2252. /* Get socket */
  2253. if (conn) {
  2254. s = conn->socket;
  2255. if (s < 0) {
  2256. /* Data comes in right away after an accept, even though
  2257. * the server task might not have created a new socket yet.
  2258. * Just count down (or up) if that's the case and we
  2259. * will use the data later. Note that only receive events
  2260. * can happen before the new socket is set up. */
  2261. SYS_ARCH_PROTECT(lev);
  2262. if (conn->socket < 0) {
  2263. if (evt == NETCONN_EVT_RCVPLUS) {
  2264. /* conn->socket is -1 on initialization
  2265. lwip_accept adjusts sock->recvevent if conn->socket < -1 */
  2266. conn->socket--;
  2267. }
  2268. SYS_ARCH_UNPROTECT(lev);
  2269. return;
  2270. }
  2271. s = conn->socket;
  2272. SYS_ARCH_UNPROTECT(lev);
  2273. }
  2274. sock = get_socket(s);
  2275. if (!sock) {
  2276. return;
  2277. }
  2278. } else {
  2279. return;
  2280. }
  2281. check_waiters = 1;
  2282. SYS_ARCH_PROTECT(lev);
  2283. /* Set event as required */
  2284. switch (evt) {
  2285. case NETCONN_EVT_RCVPLUS:
  2286. sock->rcvevent++;
  2287. if (sock->rcvevent > 1) {
  2288. check_waiters = 0;
  2289. }
  2290. break;
  2291. case NETCONN_EVT_RCVMINUS:
  2292. sock->rcvevent--;
  2293. check_waiters = 0;
  2294. break;
  2295. case NETCONN_EVT_SENDPLUS:
  2296. if (sock->sendevent) {
  2297. check_waiters = 0;
  2298. }
  2299. sock->sendevent = 1;
  2300. break;
  2301. case NETCONN_EVT_SENDMINUS:
  2302. sock->sendevent = 0;
  2303. check_waiters = 0;
  2304. break;
  2305. case NETCONN_EVT_ERROR:
  2306. sock->errevent = 1;
  2307. break;
  2308. default:
  2309. LWIP_ASSERT("unknown event", 0);
  2310. break;
  2311. }
  2312. if (sock->select_waiting && check_waiters) {
  2313. /* Save which events are active */
  2314. int has_recvevent, has_sendevent, has_errevent;
  2315. has_recvevent = sock->rcvevent > 0;
  2316. has_sendevent = sock->sendevent != 0;
  2317. has_errevent = sock->errevent != 0;
  2318. SYS_ARCH_UNPROTECT(lev);
  2319. /* Check any select calls waiting on this socket */
  2320. select_check_waiters(s, has_recvevent, has_sendevent, has_errevent);
  2321. } else {
  2322. SYS_ARCH_UNPROTECT(lev);
  2323. }
  2324. done_socket(sock);
  2325. }
  2326. /**
  2327. * Check if any select waiters are waiting on this socket and its events
  2328. *
  2329. * @note on synchronization of select_cb_list:
  2330. * LWIP_TCPIP_CORE_LOCKING: the select_cb_list must only be accessed while holding
  2331. * the core lock. We do a single pass through the list and signal any waiters.
  2332. * Core lock should already be held when calling here!!!!
  2333. * !LWIP_TCPIP_CORE_LOCKING: we use SYS_ARCH_PROTECT but unlock on each iteration
  2334. * of the loop, thus creating a possibility where a thread could modify the
  2335. * select_cb_list during our UNPROTECT/PROTECT. We use a generational counter to
  2336. * detect this change and restart the list walk. The list is expected to be small
  2337. */
  2338. static void select_check_waiters(int s, int has_recvevent, int has_sendevent, int has_errevent)
  2339. {
  2340. struct lwip_select_cb *scb;
  2341. #if !LWIP_TCPIP_CORE_LOCKING
  2342. int last_select_cb_ctr;
  2343. SYS_ARCH_DECL_PROTECT(lev);
  2344. #endif /* !LWIP_TCPIP_CORE_LOCKING */
  2345. LWIP_ASSERT_CORE_LOCKED();
  2346. #if !LWIP_TCPIP_CORE_LOCKING
  2347. SYS_ARCH_PROTECT(lev);
  2348. again:
  2349. /* remember the state of select_cb_list to detect changes */
  2350. last_select_cb_ctr = select_cb_ctr;
  2351. #endif /* !LWIP_TCPIP_CORE_LOCKING */
  2352. for (scb = select_cb_list; scb != NULL; scb = scb->next) {
  2353. if (scb->sem_signalled == 0) {
  2354. /* semaphore not signalled yet */
  2355. int do_signal = 0;
  2356. #if LWIP_SOCKET_POLL
  2357. if (scb->poll_fds != NULL) {
  2358. do_signal = lwip_poll_should_wake(scb, s, has_recvevent, has_sendevent, has_errevent);
  2359. }
  2360. #endif /* LWIP_SOCKET_POLL */
  2361. #if LWIP_SOCKET_SELECT && LWIP_SOCKET_POLL
  2362. else
  2363. #endif /* LWIP_SOCKET_SELECT && LWIP_SOCKET_POLL */
  2364. #if LWIP_SOCKET_SELECT
  2365. {
  2366. /* Test this select call for our socket */
  2367. if (has_recvevent) {
  2368. if (scb->readset && FD_ISSET(s, scb->readset)) {
  2369. do_signal = 1;
  2370. }
  2371. }
  2372. if (has_sendevent) {
  2373. if (!do_signal && scb->writeset && FD_ISSET(s, scb->writeset)) {
  2374. do_signal = 1;
  2375. }
  2376. }
  2377. if (has_errevent) {
  2378. if (!do_signal && scb->exceptset && FD_ISSET(s, scb->exceptset)) {
  2379. do_signal = 1;
  2380. }
  2381. }
  2382. }
  2383. #endif /* LWIP_SOCKET_SELECT */
  2384. if (do_signal) {
  2385. scb->sem_signalled = 1;
  2386. /* For !LWIP_TCPIP_CORE_LOCKING, we don't call SYS_ARCH_UNPROTECT() before signaling
  2387. the semaphore, as this might lead to the select thread taking itself off the list,
  2388. invalidating the semaphore. */
  2389. sys_sem_signal(SELECT_SEM_PTR(scb->sem));
  2390. }
  2391. }
  2392. #if LWIP_TCPIP_CORE_LOCKING
  2393. }
  2394. #else
  2395. /* unlock interrupts with each step */
  2396. SYS_ARCH_UNPROTECT(lev);
  2397. /* this makes sure interrupt protection time is short */
  2398. SYS_ARCH_PROTECT(lev);
  2399. if (last_select_cb_ctr != select_cb_ctr) {
  2400. /* someone has changed select_cb_list, restart at the beginning */
  2401. goto again;
  2402. }
  2403. /* remember the state of select_cb_list to detect changes */
  2404. last_select_cb_ctr = select_cb_ctr;
  2405. }
  2406. SYS_ARCH_UNPROTECT(lev);
  2407. #endif
  2408. }
  2409. #endif /* LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL */
  2410. /**
  2411. * Close one end of a full-duplex connection.
  2412. */
  2413. int
  2414. lwip_shutdown(int s, int how)
  2415. {
  2416. struct lwip_sock *sock;
  2417. err_t err;
  2418. u8_t shut_rx = 0, shut_tx = 0;
  2419. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_shutdown(%d, how=%d)\n", s, how));
  2420. sock = get_socket(s);
  2421. if (!sock) {
  2422. return -1;
  2423. }
  2424. if (sock->conn != NULL) {
  2425. if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) != NETCONN_TCP) {
  2426. sock_set_errno(sock, EOPNOTSUPP);
  2427. done_socket(sock);
  2428. return -1;
  2429. }
  2430. } else {
  2431. sock_set_errno(sock, ENOTCONN);
  2432. done_socket(sock);
  2433. return -1;
  2434. }
  2435. if (how == SHUT_RD) {
  2436. shut_rx = 1;
  2437. } else if (how == SHUT_WR) {
  2438. shut_tx = 1;
  2439. } else if (how == SHUT_RDWR) {
  2440. shut_rx = 1;
  2441. shut_tx = 1;
  2442. } else {
  2443. sock_set_errno(sock, EINVAL);
  2444. done_socket(sock);
  2445. return -1;
  2446. }
  2447. err = netconn_shutdown(sock->conn, shut_rx, shut_tx);
  2448. sock_set_errno(sock, err_to_errno(err));
  2449. done_socket(sock);
  2450. return (err == ERR_OK ? 0 : -1);
  2451. }
  2452. static int
  2453. lwip_getaddrname(int s, struct sockaddr *name, socklen_t *namelen, u8_t local)
  2454. {
  2455. struct lwip_sock *sock;
  2456. union sockaddr_aligned saddr;
  2457. ip_addr_t naddr;
  2458. u16_t port;
  2459. err_t err;
  2460. sock = get_socket(s);
  2461. if (!sock) {
  2462. return -1;
  2463. }
  2464. /* get the IP address and port */
  2465. err = netconn_getaddr(sock->conn, &naddr, &port, local);
  2466. if (err != ERR_OK) {
  2467. sock_set_errno(sock, err_to_errno(err));
  2468. done_socket(sock);
  2469. return -1;
  2470. }
  2471. #if LWIP_IPV4 && LWIP_IPV6
  2472. /* Dual-stack: Map IPv4 addresses to IPv4 mapped IPv6 */
  2473. if (NETCONNTYPE_ISIPV6(netconn_type(sock->conn)) &&
  2474. IP_IS_V4_VAL(naddr)) {
  2475. ip4_2_ipv4_mapped_ipv6(ip_2_ip6(&naddr), ip_2_ip4(&naddr));
  2476. IP_SET_TYPE_VAL(naddr, IPADDR_TYPE_V6);
  2477. }
  2478. #endif /* LWIP_IPV4 && LWIP_IPV6 */
  2479. IPADDR_PORT_TO_SOCKADDR(&saddr, &naddr, port);
  2480. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getaddrname(%d, addr=", s));
  2481. ip_addr_debug_print_val(SOCKETS_DEBUG, naddr);
  2482. LWIP_DEBUGF(SOCKETS_DEBUG, (" port=%"U16_F")\n", port));
  2483. if (*namelen > saddr.sa.sa_len) {
  2484. *namelen = saddr.sa.sa_len;
  2485. }
  2486. MEMCPY(name, &saddr, *namelen);
  2487. sock_set_errno(sock, 0);
  2488. done_socket(sock);
  2489. return 0;
  2490. }
  2491. int
  2492. lwip_getpeername(int s, struct sockaddr *name, socklen_t *namelen)
  2493. {
  2494. return lwip_getaddrname(s, name, namelen, 0);
  2495. }
  2496. int
  2497. lwip_getsockname(int s, struct sockaddr *name, socklen_t *namelen)
  2498. {
  2499. return lwip_getaddrname(s, name, namelen, 1);
  2500. }
  2501. int
  2502. lwip_getsockopt(int s, int level, int optname, void *optval, socklen_t *optlen)
  2503. {
  2504. int err;
  2505. struct lwip_sock *sock = get_socket(s);
  2506. #if !LWIP_TCPIP_CORE_LOCKING
  2507. err_t cberr;
  2508. LWIP_SETGETSOCKOPT_DATA_VAR_DECLARE(data);
  2509. #endif /* !LWIP_TCPIP_CORE_LOCKING */
  2510. if (!sock) {
  2511. return -1;
  2512. }
  2513. if ((NULL == optval) || (NULL == optlen)) {
  2514. sock_set_errno(sock, EFAULT);
  2515. done_socket(sock);
  2516. return -1;
  2517. }
  2518. #if LWIP_TCPIP_CORE_LOCKING
  2519. /* core-locking can just call the -impl function */
  2520. LOCK_TCPIP_CORE();
  2521. err = lwip_getsockopt_impl(s, level, optname, optval, optlen);
  2522. UNLOCK_TCPIP_CORE();
  2523. #else /* LWIP_TCPIP_CORE_LOCKING */
  2524. #if LWIP_MPU_COMPATIBLE
  2525. /* MPU_COMPATIBLE copies the optval data, so check for max size here */
  2526. if (*optlen > LWIP_SETGETSOCKOPT_MAXOPTLEN) {
  2527. sock_set_errno(sock, ENOBUFS);
  2528. done_socket(sock);
  2529. return -1;
  2530. }
  2531. #endif /* LWIP_MPU_COMPATIBLE */
  2532. LWIP_SETGETSOCKOPT_DATA_VAR_ALLOC(data, sock);
  2533. LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).s = s;
  2534. LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).level = level;
  2535. LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optname = optname;
  2536. LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optlen = *optlen;
  2537. #if !LWIP_MPU_COMPATIBLE
  2538. LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optval.p = optval;
  2539. #endif /* !LWIP_MPU_COMPATIBLE */
  2540. LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).err = 0;
  2541. #if LWIP_NETCONN_SEM_PER_THREAD
  2542. LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).completed_sem = LWIP_NETCONN_THREAD_SEM_GET();
  2543. #else
  2544. LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).completed_sem = &sock->conn->op_completed;
  2545. #endif
  2546. cberr = tcpip_callback(lwip_getsockopt_callback, &LWIP_SETGETSOCKOPT_DATA_VAR_REF(data));
  2547. if (cberr != ERR_OK) {
  2548. LWIP_SETGETSOCKOPT_DATA_VAR_FREE(data);
  2549. sock_set_errno(sock, err_to_errno(cberr));
  2550. done_socket(sock);
  2551. return -1;
  2552. }
  2553. sys_arch_sem_wait((sys_sem_t *)(LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).completed_sem), 0);
  2554. /* write back optlen and optval */
  2555. *optlen = LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optlen;
  2556. #if LWIP_MPU_COMPATIBLE
  2557. MEMCPY(optval, LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optval,
  2558. LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optlen);
  2559. #endif /* LWIP_MPU_COMPATIBLE */
  2560. /* maybe lwip_getsockopt_internal has changed err */
  2561. err = LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).err;
  2562. LWIP_SETGETSOCKOPT_DATA_VAR_FREE(data);
  2563. #endif /* LWIP_TCPIP_CORE_LOCKING */
  2564. sock_set_errno(sock, err);
  2565. done_socket(sock);
  2566. return err ? -1 : 0;
  2567. }
  2568. #if !LWIP_TCPIP_CORE_LOCKING
  2569. /** lwip_getsockopt_callback: only used without CORE_LOCKING
  2570. * to get into the tcpip_thread
  2571. */
  2572. static void
  2573. lwip_getsockopt_callback(void *arg)
  2574. {
  2575. struct lwip_setgetsockopt_data *data;
  2576. LWIP_ASSERT("arg != NULL", arg != NULL);
  2577. data = (struct lwip_setgetsockopt_data *)arg;
  2578. data->err = lwip_getsockopt_impl(data->s, data->level, data->optname,
  2579. #if LWIP_MPU_COMPATIBLE
  2580. data->optval,
  2581. #else /* LWIP_MPU_COMPATIBLE */
  2582. data->optval.p,
  2583. #endif /* LWIP_MPU_COMPATIBLE */
  2584. &data->optlen);
  2585. sys_sem_signal((sys_sem_t *)(data->completed_sem));
  2586. }
  2587. #endif /* LWIP_TCPIP_CORE_LOCKING */
  2588. static int
  2589. lwip_sockopt_to_ipopt(int optname)
  2590. {
  2591. /* Map SO_* values to our internal SOF_* values
  2592. * We should not rely on #defines in socket.h
  2593. * being in sync with ip.h.
  2594. */
  2595. switch (optname) {
  2596. case SO_BROADCAST:
  2597. return SOF_BROADCAST;
  2598. case SO_KEEPALIVE:
  2599. return SOF_KEEPALIVE;
  2600. case SO_REUSEADDR:
  2601. return SOF_REUSEADDR;
  2602. default:
  2603. LWIP_ASSERT("Unknown socket option", 0);
  2604. return 0;
  2605. }
  2606. }
  2607. /** lwip_getsockopt_impl: the actual implementation of getsockopt:
  2608. * same argument as lwip_getsockopt, either called directly or through callback
  2609. */
  2610. static int
  2611. lwip_getsockopt_impl(int s, int level, int optname, void *optval, socklen_t *optlen)
  2612. {
  2613. int err = 0;
  2614. struct lwip_sock *sock = tryget_socket(s);
  2615. if (!sock) {
  2616. return EBADF;
  2617. }
  2618. #ifdef LWIP_HOOK_SOCKETS_GETSOCKOPT
  2619. if (LWIP_HOOK_SOCKETS_GETSOCKOPT(s, sock, level, optname, optval, optlen, &err)) {
  2620. return err;
  2621. }
  2622. #endif
  2623. switch (level) {
  2624. /* Level: SOL_SOCKET */
  2625. case SOL_SOCKET:
  2626. switch (optname) {
  2627. #if LWIP_TCP
  2628. case SO_ACCEPTCONN:
  2629. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, *optlen, int);
  2630. if (NETCONNTYPE_GROUP(sock->conn->type) != NETCONN_TCP) {
  2631. done_socket(sock);
  2632. return ENOPROTOOPT;
  2633. }
  2634. if ((sock->conn->pcb.tcp != NULL) && (sock->conn->pcb.tcp->state == LISTEN)) {
  2635. *(int *)optval = 1;
  2636. } else {
  2637. *(int *)optval = 0;
  2638. }
  2639. break;
  2640. #endif /* LWIP_TCP */
  2641. /* The option flags */
  2642. case SO_BROADCAST:
  2643. case SO_KEEPALIVE:
  2644. #if SO_REUSE
  2645. case SO_REUSEADDR:
  2646. #endif /* SO_REUSE */
  2647. if ((optname == SO_BROADCAST) &&
  2648. (NETCONNTYPE_GROUP(sock->conn->type) != NETCONN_UDP)) {
  2649. done_socket(sock);
  2650. return ENOPROTOOPT;
  2651. }
  2652. optname = lwip_sockopt_to_ipopt(optname);
  2653. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, *optlen, int);
  2654. *(int *)optval = ip_get_option(sock->conn->pcb.ip, optname);
  2655. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, SOL_SOCKET, optname=0x%x, ..) = %s\n",
  2656. s, optname, (*(int *)optval ? "on" : "off")));
  2657. break;
  2658. case SO_TYPE:
  2659. LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, *optlen, int);
  2660. switch (NETCONNTYPE_GROUP(netconn_type(sock->conn))) {
  2661. case NETCONN_RAW:
  2662. *(int *)optval = SOCK_RAW;
  2663. break;
  2664. case NETCONN_TCP:
  2665. *(int *)optval = SOCK_STREAM;
  2666. break;
  2667. case NETCONN_UDP:
  2668. *(int *)optval = SOCK_DGRAM;
  2669. break;
  2670. default: /* unrecognized socket type */
  2671. *(int *)optval = netconn_type(sock->conn);
  2672. LWIP_DEBUGF(SOCKETS_DEBUG,
  2673. ("lwip_getsockopt(%d, SOL_SOCKET, SO_TYPE): unrecognized socket type %d\n",
  2674. s, *(int *)optval));
  2675. } /* switch (netconn_type(sock->conn)) */
  2676. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, SOL_SOCKET, SO_TYPE) = %d\n",
  2677. s, *(int *)optval));
  2678. break;
  2679. case SO_ERROR:
  2680. LWIP_SOCKOPT_CHECK_OPTLEN(sock, *optlen, int);
  2681. *(int *)optval = err_to_errno(netconn_err(sock->conn));
  2682. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, SOL_SOCKET, SO_ERROR) = %d\n",
  2683. s, *(int *)optval));
  2684. break;
  2685. #if LWIP_SO_SNDTIMEO
  2686. case SO_SNDTIMEO:
  2687. LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, *optlen, LWIP_SO_SNDRCVTIMEO_OPTTYPE);
  2688. LWIP_SO_SNDRCVTIMEO_SET(optval, netconn_get_sendtimeout(sock->conn));
  2689. break;
  2690. #endif /* LWIP_SO_SNDTIMEO */
  2691. #if LWIP_SO_RCVTIMEO
  2692. case SO_RCVTIMEO:
  2693. LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, *optlen, LWIP_SO_SNDRCVTIMEO_OPTTYPE);
  2694. LWIP_SO_SNDRCVTIMEO_SET(optval, netconn_get_recvtimeout(sock->conn));
  2695. break;
  2696. #endif /* LWIP_SO_RCVTIMEO */
  2697. #if LWIP_SO_RCVBUF
  2698. case SO_RCVBUF:
  2699. LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, *optlen, int);
  2700. *(int *)optval = netconn_get_recvbufsize(sock->conn);
  2701. break;
  2702. #endif /* LWIP_SO_RCVBUF */
  2703. #if LWIP_SO_LINGER
  2704. case SO_LINGER: {
  2705. s16_t conn_linger;
  2706. struct linger *linger = (struct linger *)optval;
  2707. LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, *optlen, struct linger);
  2708. conn_linger = sock->conn->linger;
  2709. if (conn_linger >= 0) {
  2710. linger->l_onoff = 1;
  2711. linger->l_linger = (int)conn_linger;
  2712. } else {
  2713. linger->l_onoff = 0;
  2714. linger->l_linger = 0;
  2715. }
  2716. }
  2717. break;
  2718. #endif /* LWIP_SO_LINGER */
  2719. #if LWIP_UDP
  2720. case SO_NO_CHECK:
  2721. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, *optlen, int, NETCONN_UDP);
  2722. #if LWIP_UDPLITE
  2723. if (udp_is_flag_set(sock->conn->pcb.udp, UDP_FLAGS_UDPLITE)) {
  2724. /* this flag is only available for UDP, not for UDP lite */
  2725. done_socket(sock);
  2726. return EAFNOSUPPORT;
  2727. }
  2728. #endif /* LWIP_UDPLITE */
  2729. *(int *)optval = udp_is_flag_set(sock->conn->pcb.udp, UDP_FLAGS_NOCHKSUM) ? 1 : 0;
  2730. break;
  2731. #endif /* LWIP_UDP*/
  2732. default:
  2733. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, SOL_SOCKET, UNIMPL: optname=0x%x, ..)\n",
  2734. s, optname));
  2735. err = ENOPROTOOPT;
  2736. break;
  2737. } /* switch (optname) */
  2738. break;
  2739. /* Level: IPPROTO_IP */
  2740. case IPPROTO_IP:
  2741. switch (optname) {
  2742. case IP_TTL:
  2743. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, *optlen, int);
  2744. *(int *)optval = sock->conn->pcb.ip->ttl;
  2745. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_IP, IP_TTL) = %d\n",
  2746. s, *(int *)optval));
  2747. break;
  2748. case IP_TOS:
  2749. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, *optlen, int);
  2750. *(int *)optval = sock->conn->pcb.ip->tos;
  2751. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_IP, IP_TOS) = %d\n",
  2752. s, *(int *)optval));
  2753. break;
  2754. #if LWIP_IPV4 && LWIP_MULTICAST_TX_OPTIONS && LWIP_UDP
  2755. case IP_MULTICAST_TTL:
  2756. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, *optlen, u8_t);
  2757. if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) != NETCONN_UDP) {
  2758. done_socket(sock);
  2759. return ENOPROTOOPT;
  2760. }
  2761. *(u8_t *)optval = udp_get_multicast_ttl(sock->conn->pcb.udp);
  2762. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_IP, IP_MULTICAST_TTL) = %d\n",
  2763. s, *(int *)optval));
  2764. break;
  2765. case IP_MULTICAST_IF:
  2766. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, *optlen, struct in_addr);
  2767. if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) != NETCONN_UDP) {
  2768. done_socket(sock);
  2769. return ENOPROTOOPT;
  2770. }
  2771. inet_addr_from_ip4addr((struct in_addr *)optval, udp_get_multicast_netif_addr(sock->conn->pcb.udp));
  2772. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_IP, IP_MULTICAST_IF) = 0x%"X32_F"\n",
  2773. s, *(u32_t *)optval));
  2774. break;
  2775. case IP_MULTICAST_LOOP:
  2776. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, *optlen, u8_t);
  2777. if ((sock->conn->pcb.udp->flags & UDP_FLAGS_MULTICAST_LOOP) != 0) {
  2778. *(u8_t *)optval = 1;
  2779. } else {
  2780. *(u8_t *)optval = 0;
  2781. }
  2782. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_IP, IP_MULTICAST_LOOP) = %d\n",
  2783. s, *(int *)optval));
  2784. break;
  2785. #endif /* LWIP_IPV4 && LWIP_MULTICAST_TX_OPTIONS && LWIP_UDP */
  2786. default:
  2787. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_IP, UNIMPL: optname=0x%x, ..)\n",
  2788. s, optname));
  2789. err = ENOPROTOOPT;
  2790. break;
  2791. } /* switch (optname) */
  2792. break;
  2793. #if LWIP_TCP
  2794. /* Level: IPPROTO_TCP */
  2795. case IPPROTO_TCP:
  2796. /* Special case: all IPPROTO_TCP option take an int */
  2797. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, *optlen, int, NETCONN_TCP);
  2798. if (sock->conn->pcb.tcp->state == LISTEN) {
  2799. done_socket(sock);
  2800. return EINVAL;
  2801. }
  2802. switch (optname) {
  2803. case TCP_NODELAY:
  2804. *(int *)optval = tcp_nagle_disabled(sock->conn->pcb.tcp);
  2805. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_TCP, TCP_NODELAY) = %s\n",
  2806. s, (*(int *)optval) ? "on" : "off") );
  2807. break;
  2808. case TCP_KEEPALIVE:
  2809. *(int *)optval = (int)sock->conn->pcb.tcp->keep_idle;
  2810. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_TCP, TCP_KEEPALIVE) = %d\n",
  2811. s, *(int *)optval));
  2812. break;
  2813. #if LWIP_TCP_KEEPALIVE
  2814. case TCP_KEEPIDLE:
  2815. *(int *)optval = (int)(sock->conn->pcb.tcp->keep_idle / 1000);
  2816. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_TCP, TCP_KEEPIDLE) = %d\n",
  2817. s, *(int *)optval));
  2818. break;
  2819. case TCP_KEEPINTVL:
  2820. *(int *)optval = (int)(sock->conn->pcb.tcp->keep_intvl / 1000);
  2821. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_TCP, TCP_KEEPINTVL) = %d\n",
  2822. s, *(int *)optval));
  2823. break;
  2824. case TCP_KEEPCNT:
  2825. *(int *)optval = (int)sock->conn->pcb.tcp->keep_cnt;
  2826. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_TCP, TCP_KEEPCNT) = %d\n",
  2827. s, *(int *)optval));
  2828. break;
  2829. #endif /* LWIP_TCP_KEEPALIVE */
  2830. default:
  2831. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_TCP, UNIMPL: optname=0x%x, ..)\n",
  2832. s, optname));
  2833. err = ENOPROTOOPT;
  2834. break;
  2835. } /* switch (optname) */
  2836. break;
  2837. #endif /* LWIP_TCP */
  2838. #if LWIP_IPV6
  2839. /* Level: IPPROTO_IPV6 */
  2840. case IPPROTO_IPV6:
  2841. switch (optname) {
  2842. case IPV6_V6ONLY:
  2843. LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, *optlen, int);
  2844. *(int *)optval = (netconn_get_ipv6only(sock->conn) ? 1 : 0);
  2845. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_IPV6, IPV6_V6ONLY) = %d\n",
  2846. s, *(int *)optval));
  2847. break;
  2848. default:
  2849. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_IPV6, UNIMPL: optname=0x%x, ..)\n",
  2850. s, optname));
  2851. err = ENOPROTOOPT;
  2852. break;
  2853. } /* switch (optname) */
  2854. break;
  2855. #endif /* LWIP_IPV6 */
  2856. #if LWIP_UDP && LWIP_UDPLITE
  2857. /* Level: IPPROTO_UDPLITE */
  2858. case IPPROTO_UDPLITE:
  2859. /* Special case: all IPPROTO_UDPLITE option take an int */
  2860. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, *optlen, int);
  2861. /* If this is no UDP lite socket, ignore any options. */
  2862. if (!NETCONNTYPE_ISUDPLITE(netconn_type(sock->conn))) {
  2863. done_socket(sock);
  2864. return ENOPROTOOPT;
  2865. }
  2866. switch (optname) {
  2867. case UDPLITE_SEND_CSCOV:
  2868. *(int *)optval = sock->conn->pcb.udp->chksum_len_tx;
  2869. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_UDPLITE, UDPLITE_SEND_CSCOV) = %d\n",
  2870. s, (*(int *)optval)) );
  2871. break;
  2872. case UDPLITE_RECV_CSCOV:
  2873. *(int *)optval = sock->conn->pcb.udp->chksum_len_rx;
  2874. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_UDPLITE, UDPLITE_RECV_CSCOV) = %d\n",
  2875. s, (*(int *)optval)) );
  2876. break;
  2877. default:
  2878. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_UDPLITE, UNIMPL: optname=0x%x, ..)\n",
  2879. s, optname));
  2880. err = ENOPROTOOPT;
  2881. break;
  2882. } /* switch (optname) */
  2883. break;
  2884. #endif /* LWIP_UDP */
  2885. /* Level: IPPROTO_RAW */
  2886. case IPPROTO_RAW:
  2887. switch (optname) {
  2888. #if LWIP_IPV6 && LWIP_RAW
  2889. case IPV6_CHECKSUM:
  2890. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, *optlen, int, NETCONN_RAW);
  2891. if (sock->conn->pcb.raw->chksum_reqd == 0) {
  2892. *(int *)optval = -1;
  2893. } else {
  2894. *(int *)optval = sock->conn->pcb.raw->chksum_offset;
  2895. }
  2896. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_RAW, IPV6_CHECKSUM) = %d\n",
  2897. s, (*(int *)optval)) );
  2898. break;
  2899. #endif /* LWIP_IPV6 && LWIP_RAW */
  2900. default:
  2901. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_RAW, UNIMPL: optname=0x%x, ..)\n",
  2902. s, optname));
  2903. err = ENOPROTOOPT;
  2904. break;
  2905. } /* switch (optname) */
  2906. break;
  2907. default:
  2908. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, level=0x%x, UNIMPL: optname=0x%x, ..)\n",
  2909. s, level, optname));
  2910. err = ENOPROTOOPT;
  2911. break;
  2912. } /* switch (level) */
  2913. done_socket(sock);
  2914. return err;
  2915. }
  2916. int
  2917. lwip_setsockopt(int s, int level, int optname, const void *optval, socklen_t optlen)
  2918. {
  2919. int err = 0;
  2920. struct lwip_sock *sock = get_socket(s);
  2921. #if !LWIP_TCPIP_CORE_LOCKING
  2922. err_t cberr;
  2923. LWIP_SETGETSOCKOPT_DATA_VAR_DECLARE(data);
  2924. #endif /* !LWIP_TCPIP_CORE_LOCKING */
  2925. if (!sock) {
  2926. return -1;
  2927. }
  2928. if (NULL == optval) {
  2929. sock_set_errno(sock, EFAULT);
  2930. done_socket(sock);
  2931. return -1;
  2932. }
  2933. #if LWIP_TCPIP_CORE_LOCKING
  2934. /* core-locking can just call the -impl function */
  2935. LOCK_TCPIP_CORE();
  2936. err = lwip_setsockopt_impl(s, level, optname, optval, optlen);
  2937. UNLOCK_TCPIP_CORE();
  2938. #else /* LWIP_TCPIP_CORE_LOCKING */
  2939. #if LWIP_MPU_COMPATIBLE
  2940. /* MPU_COMPATIBLE copies the optval data, so check for max size here */
  2941. if (optlen > LWIP_SETGETSOCKOPT_MAXOPTLEN) {
  2942. sock_set_errno(sock, ENOBUFS);
  2943. done_socket(sock);
  2944. return -1;
  2945. }
  2946. #endif /* LWIP_MPU_COMPATIBLE */
  2947. LWIP_SETGETSOCKOPT_DATA_VAR_ALLOC(data, sock);
  2948. LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).s = s;
  2949. LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).level = level;
  2950. LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optname = optname;
  2951. LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optlen = optlen;
  2952. #if LWIP_MPU_COMPATIBLE
  2953. MEMCPY(LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optval, optval, optlen);
  2954. #else /* LWIP_MPU_COMPATIBLE */
  2955. LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optval.pc = (const void *)optval;
  2956. #endif /* LWIP_MPU_COMPATIBLE */
  2957. LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).err = 0;
  2958. #if LWIP_NETCONN_SEM_PER_THREAD
  2959. LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).completed_sem = LWIP_NETCONN_THREAD_SEM_GET();
  2960. #else
  2961. LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).completed_sem = &sock->conn->op_completed;
  2962. #endif
  2963. cberr = tcpip_callback(lwip_setsockopt_callback, &LWIP_SETGETSOCKOPT_DATA_VAR_REF(data));
  2964. if (cberr != ERR_OK) {
  2965. LWIP_SETGETSOCKOPT_DATA_VAR_FREE(data);
  2966. sock_set_errno(sock, err_to_errno(cberr));
  2967. done_socket(sock);
  2968. return -1;
  2969. }
  2970. sys_arch_sem_wait((sys_sem_t *)(LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).completed_sem), 0);
  2971. /* maybe lwip_getsockopt_internal has changed err */
  2972. err = LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).err;
  2973. LWIP_SETGETSOCKOPT_DATA_VAR_FREE(data);
  2974. #endif /* LWIP_TCPIP_CORE_LOCKING */
  2975. sock_set_errno(sock, err);
  2976. done_socket(sock);
  2977. return err ? -1 : 0;
  2978. }
  2979. #if !LWIP_TCPIP_CORE_LOCKING
  2980. /** lwip_setsockopt_callback: only used without CORE_LOCKING
  2981. * to get into the tcpip_thread
  2982. */
  2983. static void
  2984. lwip_setsockopt_callback(void *arg)
  2985. {
  2986. struct lwip_setgetsockopt_data *data;
  2987. LWIP_ASSERT("arg != NULL", arg != NULL);
  2988. data = (struct lwip_setgetsockopt_data *)arg;
  2989. data->err = lwip_setsockopt_impl(data->s, data->level, data->optname,
  2990. #if LWIP_MPU_COMPATIBLE
  2991. data->optval,
  2992. #else /* LWIP_MPU_COMPATIBLE */
  2993. data->optval.pc,
  2994. #endif /* LWIP_MPU_COMPATIBLE */
  2995. data->optlen);
  2996. sys_sem_signal((sys_sem_t *)(data->completed_sem));
  2997. }
  2998. #endif /* LWIP_TCPIP_CORE_LOCKING */
  2999. /** lwip_setsockopt_impl: the actual implementation of setsockopt:
  3000. * same argument as lwip_setsockopt, either called directly or through callback
  3001. */
  3002. static int
  3003. lwip_setsockopt_impl(int s, int level, int optname, const void *optval, socklen_t optlen)
  3004. {
  3005. int err = 0;
  3006. struct lwip_sock *sock = tryget_socket(s);
  3007. if (!sock) {
  3008. return EBADF;
  3009. }
  3010. #ifdef LWIP_HOOK_SOCKETS_SETSOCKOPT
  3011. if (LWIP_HOOK_SOCKETS_SETSOCKOPT(s, sock, level, optname, optval, optlen, &err)) {
  3012. return err;
  3013. }
  3014. #endif
  3015. switch (level) {
  3016. /* Level: SOL_SOCKET */
  3017. case SOL_SOCKET:
  3018. switch (optname) {
  3019. /* SO_ACCEPTCONN is get-only */
  3020. /* The option flags */
  3021. case SO_BROADCAST:
  3022. case SO_KEEPALIVE:
  3023. #if SO_REUSE
  3024. case SO_REUSEADDR:
  3025. #endif /* SO_REUSE */
  3026. if ((optname == SO_BROADCAST) &&
  3027. (NETCONNTYPE_GROUP(sock->conn->type) != NETCONN_UDP)) {
  3028. done_socket(sock);
  3029. return ENOPROTOOPT;
  3030. }
  3031. optname = lwip_sockopt_to_ipopt(optname);
  3032. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, optlen, int);
  3033. if (*(const int *)optval) {
  3034. ip_set_option(sock->conn->pcb.ip, optname);
  3035. } else {
  3036. ip_reset_option(sock->conn->pcb.ip, optname);
  3037. }
  3038. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, SOL_SOCKET, optname=0x%x, ..) -> %s\n",
  3039. s, optname, (*(const int *)optval ? "on" : "off")));
  3040. break;
  3041. /* SO_TYPE is get-only */
  3042. /* SO_ERROR is get-only */
  3043. #if LWIP_SO_SNDTIMEO
  3044. case SO_SNDTIMEO: {
  3045. long ms_long;
  3046. LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, optlen, LWIP_SO_SNDRCVTIMEO_OPTTYPE);
  3047. ms_long = LWIP_SO_SNDRCVTIMEO_GET_MS(optval);
  3048. if (ms_long < 0) {
  3049. done_socket(sock);
  3050. return EINVAL;
  3051. }
  3052. netconn_set_sendtimeout(sock->conn, ms_long);
  3053. break;
  3054. }
  3055. #endif /* LWIP_SO_SNDTIMEO */
  3056. #if LWIP_SO_RCVTIMEO
  3057. case SO_RCVTIMEO: {
  3058. long ms_long;
  3059. LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, optlen, LWIP_SO_SNDRCVTIMEO_OPTTYPE);
  3060. ms_long = LWIP_SO_SNDRCVTIMEO_GET_MS(optval);
  3061. if (ms_long < 0) {
  3062. done_socket(sock);
  3063. return EINVAL;
  3064. }
  3065. netconn_set_recvtimeout(sock->conn, (u32_t)ms_long);
  3066. break;
  3067. }
  3068. #endif /* LWIP_SO_RCVTIMEO */
  3069. #if LWIP_SO_RCVBUF
  3070. case SO_RCVBUF:
  3071. LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, optlen, int);
  3072. netconn_set_recvbufsize(sock->conn, *(const int *)optval);
  3073. break;
  3074. #endif /* LWIP_SO_RCVBUF */
  3075. #if LWIP_SO_LINGER
  3076. case SO_LINGER: {
  3077. const struct linger *linger = (const struct linger *)optval;
  3078. LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, optlen, struct linger);
  3079. if (linger->l_onoff) {
  3080. int lingersec = linger->l_linger;
  3081. if (lingersec < 0) {
  3082. done_socket(sock);
  3083. return EINVAL;
  3084. }
  3085. if (lingersec > 0xFFFF) {
  3086. lingersec = 0xFFFF;
  3087. }
  3088. sock->conn->linger = (s16_t)lingersec;
  3089. } else {
  3090. sock->conn->linger = -1;
  3091. }
  3092. }
  3093. break;
  3094. #endif /* LWIP_SO_LINGER */
  3095. #if LWIP_UDP
  3096. case SO_NO_CHECK:
  3097. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, int, NETCONN_UDP);
  3098. #if LWIP_UDPLITE
  3099. if (udp_is_flag_set(sock->conn->pcb.udp, UDP_FLAGS_UDPLITE)) {
  3100. /* this flag is only available for UDP, not for UDP lite */
  3101. done_socket(sock);
  3102. return EAFNOSUPPORT;
  3103. }
  3104. #endif /* LWIP_UDPLITE */
  3105. if (*(const int *)optval) {
  3106. udp_set_flags(sock->conn->pcb.udp, UDP_FLAGS_NOCHKSUM);
  3107. } else {
  3108. udp_clear_flags(sock->conn->pcb.udp, UDP_FLAGS_NOCHKSUM);
  3109. }
  3110. break;
  3111. #endif /* LWIP_UDP */
  3112. case SO_BINDTODEVICE: {
  3113. const struct ifreq *iface;
  3114. struct netif *n = NULL;
  3115. LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, optlen, struct ifreq);
  3116. iface = (const struct ifreq *)optval;
  3117. if (iface->ifr_name[0] != 0) {
  3118. n = netif_find(iface->ifr_name);
  3119. if (n == NULL) {
  3120. done_socket(sock);
  3121. return ENODEV;
  3122. }
  3123. }
  3124. switch (NETCONNTYPE_GROUP(netconn_type(sock->conn))) {
  3125. #if LWIP_TCP
  3126. case NETCONN_TCP:
  3127. tcp_bind_netif(sock->conn->pcb.tcp, n);
  3128. break;
  3129. #endif
  3130. #if LWIP_UDP
  3131. case NETCONN_UDP:
  3132. udp_bind_netif(sock->conn->pcb.udp, n);
  3133. break;
  3134. #endif
  3135. #if LWIP_RAW
  3136. case NETCONN_RAW:
  3137. raw_bind_netif(sock->conn->pcb.raw, n);
  3138. break;
  3139. #endif
  3140. default:
  3141. LWIP_ASSERT("Unhandled netconn type in SO_BINDTODEVICE", 0);
  3142. break;
  3143. }
  3144. }
  3145. break;
  3146. default:
  3147. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, SOL_SOCKET, UNIMPL: optname=0x%x, ..)\n",
  3148. s, optname));
  3149. err = ENOPROTOOPT;
  3150. break;
  3151. } /* switch (optname) */
  3152. break;
  3153. /* Level: IPPROTO_IP */
  3154. case IPPROTO_IP:
  3155. switch (optname) {
  3156. case IP_TTL:
  3157. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, optlen, int);
  3158. sock->conn->pcb.ip->ttl = (u8_t)(*(const int *)optval);
  3159. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_IP, IP_TTL, ..) -> %d\n",
  3160. s, sock->conn->pcb.ip->ttl));
  3161. break;
  3162. case IP_TOS:
  3163. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, optlen, int);
  3164. sock->conn->pcb.ip->tos = (u8_t)(*(const int *)optval);
  3165. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_IP, IP_TOS, ..)-> %d\n",
  3166. s, sock->conn->pcb.ip->tos));
  3167. break;
  3168. #if LWIP_NETBUF_RECVINFO
  3169. case IP_PKTINFO:
  3170. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, int, NETCONN_UDP);
  3171. if (*(const int *)optval) {
  3172. sock->conn->flags |= NETCONN_FLAG_PKTINFO;
  3173. } else {
  3174. sock->conn->flags &= ~NETCONN_FLAG_PKTINFO;
  3175. }
  3176. break;
  3177. #endif /* LWIP_NETBUF_RECVINFO */
  3178. #if LWIP_IPV4 && LWIP_MULTICAST_TX_OPTIONS && LWIP_UDP
  3179. case IP_MULTICAST_TTL:
  3180. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, u8_t, NETCONN_UDP);
  3181. udp_set_multicast_ttl(sock->conn->pcb.udp, (u8_t)(*(const u8_t *)optval));
  3182. break;
  3183. case IP_MULTICAST_IF: {
  3184. ip4_addr_t if_addr;
  3185. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, struct in_addr, NETCONN_UDP);
  3186. inet_addr_to_ip4addr(&if_addr, (const struct in_addr *)optval);
  3187. udp_set_multicast_netif_addr(sock->conn->pcb.udp, &if_addr);
  3188. }
  3189. break;
  3190. case IP_MULTICAST_LOOP:
  3191. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, u8_t, NETCONN_UDP);
  3192. if (*(const u8_t *)optval) {
  3193. udp_set_flags(sock->conn->pcb.udp, UDP_FLAGS_MULTICAST_LOOP);
  3194. } else {
  3195. udp_clear_flags(sock->conn->pcb.udp, UDP_FLAGS_MULTICAST_LOOP);
  3196. }
  3197. break;
  3198. #endif /* LWIP_IPV4 && LWIP_MULTICAST_TX_OPTIONS && LWIP_UDP */
  3199. #if LWIP_IGMP
  3200. case IP_ADD_MEMBERSHIP:
  3201. case IP_DROP_MEMBERSHIP: {
  3202. /* If this is a TCP or a RAW socket, ignore these options. */
  3203. err_t igmp_err;
  3204. const struct ip_mreq *imr = (const struct ip_mreq *)optval;
  3205. ip4_addr_t if_addr;
  3206. ip4_addr_t multi_addr;
  3207. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, struct ip_mreq, NETCONN_UDP);
  3208. inet_addr_to_ip4addr(&if_addr, &imr->imr_interface);
  3209. inet_addr_to_ip4addr(&multi_addr, &imr->imr_multiaddr);
  3210. if (optname == IP_ADD_MEMBERSHIP) {
  3211. if (!lwip_socket_register_membership(s, &if_addr, &multi_addr)) {
  3212. /* cannot track membership (out of memory) */
  3213. err = ENOMEM;
  3214. igmp_err = ERR_OK;
  3215. } else {
  3216. igmp_err = igmp_joingroup(&if_addr, &multi_addr);
  3217. }
  3218. } else {
  3219. igmp_err = igmp_leavegroup(&if_addr, &multi_addr);
  3220. lwip_socket_unregister_membership(s, &if_addr, &multi_addr);
  3221. }
  3222. if (igmp_err != ERR_OK) {
  3223. err = EADDRNOTAVAIL;
  3224. }
  3225. }
  3226. break;
  3227. #endif /* LWIP_IGMP */
  3228. default:
  3229. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_IP, UNIMPL: optname=0x%x, ..)\n",
  3230. s, optname));
  3231. err = ENOPROTOOPT;
  3232. break;
  3233. } /* switch (optname) */
  3234. break;
  3235. #if LWIP_TCP
  3236. /* Level: IPPROTO_TCP */
  3237. case IPPROTO_TCP:
  3238. /* Special case: all IPPROTO_TCP option take an int */
  3239. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, int, NETCONN_TCP);
  3240. if (sock->conn->pcb.tcp->state == LISTEN) {
  3241. done_socket(sock);
  3242. return EINVAL;
  3243. }
  3244. switch (optname) {
  3245. case TCP_NODELAY:
  3246. if (*(const int *)optval) {
  3247. tcp_nagle_disable(sock->conn->pcb.tcp);
  3248. } else {
  3249. tcp_nagle_enable(sock->conn->pcb.tcp);
  3250. }
  3251. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_TCP, TCP_NODELAY) -> %s\n",
  3252. s, (*(const int *)optval) ? "on" : "off") );
  3253. break;
  3254. case TCP_KEEPALIVE:
  3255. sock->conn->pcb.tcp->keep_idle = (u32_t)(*(const int *)optval);
  3256. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_TCP, TCP_KEEPALIVE) -> %"U32_F"\n",
  3257. s, sock->conn->pcb.tcp->keep_idle));
  3258. break;
  3259. #if LWIP_TCP_KEEPALIVE
  3260. case TCP_KEEPIDLE:
  3261. sock->conn->pcb.tcp->keep_idle = 1000 * (u32_t)(*(const int *)optval);
  3262. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_TCP, TCP_KEEPIDLE) -> %"U32_F"\n",
  3263. s, sock->conn->pcb.tcp->keep_idle));
  3264. break;
  3265. case TCP_KEEPINTVL:
  3266. sock->conn->pcb.tcp->keep_intvl = 1000 * (u32_t)(*(const int *)optval);
  3267. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_TCP, TCP_KEEPINTVL) -> %"U32_F"\n",
  3268. s, sock->conn->pcb.tcp->keep_intvl));
  3269. break;
  3270. case TCP_KEEPCNT:
  3271. sock->conn->pcb.tcp->keep_cnt = (u32_t)(*(const int *)optval);
  3272. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_TCP, TCP_KEEPCNT) -> %"U32_F"\n",
  3273. s, sock->conn->pcb.tcp->keep_cnt));
  3274. break;
  3275. #endif /* LWIP_TCP_KEEPALIVE */
  3276. default:
  3277. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_TCP, UNIMPL: optname=0x%x, ..)\n",
  3278. s, optname));
  3279. err = ENOPROTOOPT;
  3280. break;
  3281. } /* switch (optname) */
  3282. break;
  3283. #endif /* LWIP_TCP*/
  3284. #if LWIP_IPV6
  3285. /* Level: IPPROTO_IPV6 */
  3286. case IPPROTO_IPV6:
  3287. switch (optname) {
  3288. case IPV6_V6ONLY:
  3289. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, optlen, int);
  3290. if (*(const int *)optval) {
  3291. netconn_set_ipv6only(sock->conn, 1);
  3292. } else {
  3293. netconn_set_ipv6only(sock->conn, 0);
  3294. }
  3295. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_IPV6, IPV6_V6ONLY, ..) -> %d\n",
  3296. s, (netconn_get_ipv6only(sock->conn) ? 1 : 0)));
  3297. break;
  3298. #if LWIP_IPV6_MLD
  3299. case IPV6_JOIN_GROUP:
  3300. case IPV6_LEAVE_GROUP: {
  3301. /* If this is a TCP or a RAW socket, ignore these options. */
  3302. err_t mld6_err;
  3303. struct netif *netif;
  3304. ip6_addr_t multi_addr;
  3305. const struct ipv6_mreq *imr = (const struct ipv6_mreq *)optval;
  3306. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, struct ipv6_mreq, NETCONN_UDP);
  3307. inet6_addr_to_ip6addr(&multi_addr, &imr->ipv6mr_multiaddr);
  3308. LWIP_ASSERT("Invalid netif index", imr->ipv6mr_interface <= 0xFFu);
  3309. netif = netif_get_by_index((u8_t)imr->ipv6mr_interface);
  3310. if (netif == NULL) {
  3311. err = EADDRNOTAVAIL;
  3312. break;
  3313. }
  3314. if (optname == IPV6_JOIN_GROUP) {
  3315. if (!lwip_socket_register_mld6_membership(s, imr->ipv6mr_interface, &multi_addr)) {
  3316. /* cannot track membership (out of memory) */
  3317. err = ENOMEM;
  3318. mld6_err = ERR_OK;
  3319. } else {
  3320. mld6_err = mld6_joingroup_netif(netif, &multi_addr);
  3321. }
  3322. } else {
  3323. mld6_err = mld6_leavegroup_netif(netif, &multi_addr);
  3324. lwip_socket_unregister_mld6_membership(s, imr->ipv6mr_interface, &multi_addr);
  3325. }
  3326. if (mld6_err != ERR_OK) {
  3327. err = EADDRNOTAVAIL;
  3328. }
  3329. }
  3330. break;
  3331. #endif /* LWIP_IPV6_MLD */
  3332. default:
  3333. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_IPV6, UNIMPL: optname=0x%x, ..)\n",
  3334. s, optname));
  3335. err = ENOPROTOOPT;
  3336. break;
  3337. } /* switch (optname) */
  3338. break;
  3339. #endif /* LWIP_IPV6 */
  3340. #if LWIP_UDP && LWIP_UDPLITE
  3341. /* Level: IPPROTO_UDPLITE */
  3342. case IPPROTO_UDPLITE:
  3343. /* Special case: all IPPROTO_UDPLITE option take an int */
  3344. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, optlen, int);
  3345. /* If this is no UDP lite socket, ignore any options. */
  3346. if (!NETCONNTYPE_ISUDPLITE(netconn_type(sock->conn))) {
  3347. done_socket(sock);
  3348. return ENOPROTOOPT;
  3349. }
  3350. switch (optname) {
  3351. case UDPLITE_SEND_CSCOV:
  3352. if ((*(const int *)optval != 0) && ((*(const int *)optval < 8) || (*(const int *)optval > 0xffff))) {
  3353. /* don't allow illegal values! */
  3354. sock->conn->pcb.udp->chksum_len_tx = 8;
  3355. } else {
  3356. sock->conn->pcb.udp->chksum_len_tx = (u16_t) * (const int *)optval;
  3357. }
  3358. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_UDPLITE, UDPLITE_SEND_CSCOV) -> %d\n",
  3359. s, (*(const int *)optval)) );
  3360. break;
  3361. case UDPLITE_RECV_CSCOV:
  3362. if ((*(const int *)optval != 0) && ((*(const int *)optval < 8) || (*(const int *)optval > 0xffff))) {
  3363. /* don't allow illegal values! */
  3364. sock->conn->pcb.udp->chksum_len_rx = 8;
  3365. } else {
  3366. sock->conn->pcb.udp->chksum_len_rx = (u16_t) * (const int *)optval;
  3367. }
  3368. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_UDPLITE, UDPLITE_RECV_CSCOV) -> %d\n",
  3369. s, (*(const int *)optval)) );
  3370. break;
  3371. default:
  3372. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_UDPLITE, UNIMPL: optname=0x%x, ..)\n",
  3373. s, optname));
  3374. err = ENOPROTOOPT;
  3375. break;
  3376. } /* switch (optname) */
  3377. break;
  3378. #endif /* LWIP_UDP */
  3379. /* Level: IPPROTO_RAW */
  3380. case IPPROTO_RAW:
  3381. switch (optname) {
  3382. #if LWIP_IPV6 && LWIP_RAW
  3383. case IPV6_CHECKSUM:
  3384. /* It should not be possible to disable the checksum generation with ICMPv6
  3385. * as per RFC 3542 chapter 3.1 */
  3386. if (sock->conn->pcb.raw->protocol == IPPROTO_ICMPV6) {
  3387. done_socket(sock);
  3388. return EINVAL;
  3389. }
  3390. LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, int, NETCONN_RAW);
  3391. if (*(const int *)optval < 0) {
  3392. sock->conn->pcb.raw->chksum_reqd = 0;
  3393. } else if (*(const int *)optval & 1) {
  3394. /* Per RFC3542, odd offsets are not allowed */
  3395. done_socket(sock);
  3396. return EINVAL;
  3397. } else {
  3398. sock->conn->pcb.raw->chksum_reqd = 1;
  3399. sock->conn->pcb.raw->chksum_offset = (u16_t) * (const int *)optval;
  3400. }
  3401. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_RAW, IPV6_CHECKSUM, ..) -> %d\n",
  3402. s, sock->conn->pcb.raw->chksum_reqd));
  3403. break;
  3404. #endif /* LWIP_IPV6 && LWIP_RAW */
  3405. default:
  3406. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_RAW, UNIMPL: optname=0x%x, ..)\n",
  3407. s, optname));
  3408. err = ENOPROTOOPT;
  3409. break;
  3410. } /* switch (optname) */
  3411. break;
  3412. default:
  3413. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, level=0x%x, UNIMPL: optname=0x%x, ..)\n",
  3414. s, level, optname));
  3415. err = ENOPROTOOPT;
  3416. break;
  3417. } /* switch (level) */
  3418. done_socket(sock);
  3419. return err;
  3420. }
  3421. int
  3422. lwip_ioctl(int s, long cmd, void *argp)
  3423. {
  3424. struct lwip_sock *sock = get_socket(s);
  3425. u8_t val;
  3426. #if LWIP_SO_RCVBUF
  3427. int recv_avail;
  3428. #endif /* LWIP_SO_RCVBUF */
  3429. if (!sock) {
  3430. return -1;
  3431. }
  3432. switch (cmd) {
  3433. #if LWIP_SO_RCVBUF || LWIP_FIONREAD_LINUXMODE
  3434. case FIONREAD:
  3435. if (!argp) {
  3436. sock_set_errno(sock, EINVAL);
  3437. done_socket(sock);
  3438. return -1;
  3439. }
  3440. #if LWIP_FIONREAD_LINUXMODE
  3441. if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) != NETCONN_TCP) {
  3442. struct netbuf *nb;
  3443. if (sock->lastdata.netbuf) {
  3444. nb = sock->lastdata.netbuf;
  3445. *((int *)argp) = nb->p->tot_len;
  3446. } else {
  3447. struct netbuf *rxbuf;
  3448. err_t err = netconn_recv_udp_raw_netbuf_flags(sock->conn, &rxbuf, NETCONN_DONTBLOCK);
  3449. if (err != ERR_OK) {
  3450. *((int *)argp) = 0;
  3451. } else {
  3452. sock->lastdata.netbuf = rxbuf;
  3453. *((int *)argp) = rxbuf->p->tot_len;
  3454. }
  3455. }
  3456. done_socket(sock);
  3457. return 0;
  3458. }
  3459. #endif /* LWIP_FIONREAD_LINUXMODE */
  3460. #if LWIP_SO_RCVBUF
  3461. /* we come here if either LWIP_FIONREAD_LINUXMODE==0 or this is a TCP socket */
  3462. SYS_ARCH_GET(sock->conn->recv_avail, recv_avail);
  3463. if (recv_avail < 0) {
  3464. recv_avail = 0;
  3465. }
  3466. /* Check if there is data left from the last recv operation. /maq 041215 */
  3467. if (sock->lastdata.netbuf) {
  3468. if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_TCP) {
  3469. recv_avail += sock->lastdata.pbuf->tot_len;
  3470. } else {
  3471. recv_avail += sock->lastdata.netbuf->p->tot_len;
  3472. }
  3473. }
  3474. *((int *)argp) = recv_avail;
  3475. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_ioctl(%d, FIONREAD, %p) = %"U16_F"\n", s, argp, *((u16_t *)argp)));
  3476. sock_set_errno(sock, 0);
  3477. done_socket(sock);
  3478. return 0;
  3479. #else /* LWIP_SO_RCVBUF */
  3480. break;
  3481. #endif /* LWIP_SO_RCVBUF */
  3482. #endif /* LWIP_SO_RCVBUF || LWIP_FIONREAD_LINUXMODE */
  3483. case (long)FIONBIO:
  3484. val = 0;
  3485. if (argp && *(int *)argp) {
  3486. val = 1;
  3487. }
  3488. netconn_set_nonblocking(sock->conn, val);
  3489. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_ioctl(%d, FIONBIO, %d)\n", s, val));
  3490. sock_set_errno(sock, 0);
  3491. done_socket(sock);
  3492. return 0;
  3493. default:
  3494. break;
  3495. } /* switch (cmd) */
  3496. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_ioctl(%d, UNIMPL: 0x%lx, %p)\n", s, cmd, argp));
  3497. sock_set_errno(sock, ENOSYS); /* not yet implemented */
  3498. done_socket(sock);
  3499. return -1;
  3500. }
  3501. /** A minimal implementation of fcntl.
  3502. * Currently only the commands F_GETFL and F_SETFL are implemented.
  3503. * The flag O_NONBLOCK and access modes are supported for F_GETFL, only
  3504. * the flag O_NONBLOCK is implemented for F_SETFL.
  3505. */
  3506. int
  3507. lwip_fcntl(int s, int cmd, int val)
  3508. {
  3509. struct lwip_sock *sock = get_socket(s);
  3510. int ret = -1;
  3511. int op_mode = 0;
  3512. if (!sock) {
  3513. return -1;
  3514. }
  3515. switch (cmd) {
  3516. case F_GETFL:
  3517. ret = netconn_is_nonblocking(sock->conn) ? O_NONBLOCK : 0;
  3518. sock_set_errno(sock, 0);
  3519. if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_TCP) {
  3520. #if LWIP_TCPIP_CORE_LOCKING
  3521. LOCK_TCPIP_CORE();
  3522. #else
  3523. SYS_ARCH_DECL_PROTECT(lev);
  3524. /* the proper thing to do here would be to get into the tcpip_thread,
  3525. but locking should be OK as well since we only *read* some flags */
  3526. SYS_ARCH_PROTECT(lev);
  3527. #endif
  3528. #if LWIP_TCP
  3529. if (sock->conn->pcb.tcp) {
  3530. if (!(sock->conn->pcb.tcp->flags & TF_RXCLOSED)) {
  3531. op_mode |= O_RDONLY;
  3532. }
  3533. if (!(sock->conn->pcb.tcp->flags & TF_FIN)) {
  3534. op_mode |= O_WRONLY;
  3535. }
  3536. }
  3537. #endif
  3538. #if LWIP_TCPIP_CORE_LOCKING
  3539. UNLOCK_TCPIP_CORE();
  3540. #else
  3541. SYS_ARCH_UNPROTECT(lev);
  3542. #endif
  3543. } else {
  3544. op_mode |= O_RDWR;
  3545. }
  3546. /* ensure O_RDWR for (O_RDONLY|O_WRONLY) != O_RDWR cases */
  3547. ret |= (op_mode == (O_RDONLY | O_WRONLY)) ? O_RDWR : op_mode;
  3548. break;
  3549. case F_SETFL:
  3550. /* Bits corresponding to the file access mode and the file creation flags [..] that are set in arg shall be ignored */
  3551. val &= ~(O_RDONLY | O_WRONLY | O_RDWR);
  3552. if ((val & ~O_NONBLOCK) == 0) {
  3553. /* only O_NONBLOCK, all other bits are zero */
  3554. netconn_set_nonblocking(sock->conn, val & O_NONBLOCK);
  3555. ret = 0;
  3556. sock_set_errno(sock, 0);
  3557. } else {
  3558. sock_set_errno(sock, ENOSYS); /* not yet implemented */
  3559. }
  3560. break;
  3561. default:
  3562. LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_fcntl(%d, UNIMPL: %d, %d)\n", s, cmd, val));
  3563. sock_set_errno(sock, ENOSYS); /* not yet implemented */
  3564. break;
  3565. }
  3566. done_socket(sock);
  3567. return ret;
  3568. }
  3569. #if LWIP_COMPAT_SOCKETS == 2 && LWIP_POSIX_SOCKETS_IO_NAMES
  3570. int
  3571. fcntl(int s, int cmd, ...)
  3572. {
  3573. va_list ap;
  3574. int val;
  3575. va_start(ap, cmd);
  3576. val = va_arg(ap, int);
  3577. va_end(ap);
  3578. return lwip_fcntl(s, cmd, val);
  3579. }
  3580. #endif
  3581. const char *
  3582. lwip_inet_ntop(int af, const void *src, char *dst, socklen_t size)
  3583. {
  3584. const char *ret = NULL;
  3585. int size_int = (int)size;
  3586. if (size_int < 0) {
  3587. set_errno(ENOSPC);
  3588. return NULL;
  3589. }
  3590. switch (af) {
  3591. #if LWIP_IPV4
  3592. case AF_INET:
  3593. ret = ip4addr_ntoa_r((const ip4_addr_t *)src, dst, size_int);
  3594. if (ret == NULL) {
  3595. set_errno(ENOSPC);
  3596. }
  3597. break;
  3598. #endif
  3599. #if LWIP_IPV6
  3600. case AF_INET6:
  3601. ret = ip6addr_ntoa_r((const ip6_addr_t *)src, dst, size_int);
  3602. if (ret == NULL) {
  3603. set_errno(ENOSPC);
  3604. }
  3605. break;
  3606. #endif
  3607. default:
  3608. set_errno(EAFNOSUPPORT);
  3609. break;
  3610. }
  3611. return ret;
  3612. }
  3613. int
  3614. lwip_inet_pton(int af, const char *src, void *dst)
  3615. {
  3616. int err;
  3617. switch (af) {
  3618. #if LWIP_IPV4
  3619. case AF_INET:
  3620. err = ip4addr_aton(src, (ip4_addr_t *)dst);
  3621. break;
  3622. #endif
  3623. #if LWIP_IPV6
  3624. case AF_INET6: {
  3625. /* convert into temporary variable since ip6_addr_t might be larger
  3626. than in6_addr when scopes are enabled */
  3627. ip6_addr_t addr;
  3628. err = ip6addr_aton(src, &addr);
  3629. if (err) {
  3630. memcpy(dst, &addr.addr, sizeof(addr.addr));
  3631. }
  3632. break;
  3633. }
  3634. #endif
  3635. default:
  3636. err = -1;
  3637. set_errno(EAFNOSUPPORT);
  3638. break;
  3639. }
  3640. return err;
  3641. }
  3642. #if LWIP_IGMP
  3643. /** Register a new IGMP membership. On socket close, the membership is dropped automatically.
  3644. *
  3645. * ATTENTION: this function is called from tcpip_thread (or under CORE_LOCK).
  3646. *
  3647. * @return 1 on success, 0 on failure
  3648. */
  3649. static int
  3650. lwip_socket_register_membership(int s, const ip4_addr_t *if_addr, const ip4_addr_t *multi_addr)
  3651. {
  3652. struct lwip_sock *sock = get_socket(s);
  3653. int i;
  3654. if (!sock) {
  3655. return 0;
  3656. }
  3657. for (i = 0; i < LWIP_SOCKET_MAX_MEMBERSHIPS; i++) {
  3658. if (socket_ipv4_multicast_memberships[i].sock == NULL) {
  3659. socket_ipv4_multicast_memberships[i].sock = sock;
  3660. ip4_addr_copy(socket_ipv4_multicast_memberships[i].if_addr, *if_addr);
  3661. ip4_addr_copy(socket_ipv4_multicast_memberships[i].multi_addr, *multi_addr);
  3662. done_socket(sock);
  3663. return 1;
  3664. }
  3665. }
  3666. done_socket(sock);
  3667. return 0;
  3668. }
  3669. /** Unregister a previously registered membership. This prevents dropping the membership
  3670. * on socket close.
  3671. *
  3672. * ATTENTION: this function is called from tcpip_thread (or under CORE_LOCK).
  3673. */
  3674. static void
  3675. lwip_socket_unregister_membership(int s, const ip4_addr_t *if_addr, const ip4_addr_t *multi_addr)
  3676. {
  3677. struct lwip_sock *sock = get_socket(s);
  3678. int i;
  3679. if (!sock) {
  3680. return;
  3681. }
  3682. for (i = 0; i < LWIP_SOCKET_MAX_MEMBERSHIPS; i++) {
  3683. if ((socket_ipv4_multicast_memberships[i].sock == sock) &&
  3684. ip4_addr_cmp(&socket_ipv4_multicast_memberships[i].if_addr, if_addr) &&
  3685. ip4_addr_cmp(&socket_ipv4_multicast_memberships[i].multi_addr, multi_addr)) {
  3686. socket_ipv4_multicast_memberships[i].sock = NULL;
  3687. ip4_addr_set_zero(&socket_ipv4_multicast_memberships[i].if_addr);
  3688. ip4_addr_set_zero(&socket_ipv4_multicast_memberships[i].multi_addr);
  3689. break;
  3690. }
  3691. }
  3692. done_socket(sock);
  3693. }
  3694. /** Drop all memberships of a socket that were not dropped explicitly via setsockopt.
  3695. *
  3696. * ATTENTION: this function is NOT called from tcpip_thread (or under CORE_LOCK).
  3697. */
  3698. static void
  3699. lwip_socket_drop_registered_memberships(int s)
  3700. {
  3701. struct lwip_sock *sock = get_socket(s);
  3702. int i;
  3703. if (!sock) {
  3704. return;
  3705. }
  3706. for (i = 0; i < LWIP_SOCKET_MAX_MEMBERSHIPS; i++) {
  3707. if (socket_ipv4_multicast_memberships[i].sock == sock) {
  3708. ip_addr_t multi_addr, if_addr;
  3709. ip_addr_copy_from_ip4(multi_addr, socket_ipv4_multicast_memberships[i].multi_addr);
  3710. ip_addr_copy_from_ip4(if_addr, socket_ipv4_multicast_memberships[i].if_addr);
  3711. socket_ipv4_multicast_memberships[i].sock = NULL;
  3712. ip4_addr_set_zero(&socket_ipv4_multicast_memberships[i].if_addr);
  3713. ip4_addr_set_zero(&socket_ipv4_multicast_memberships[i].multi_addr);
  3714. netconn_join_leave_group(sock->conn, &multi_addr, &if_addr, NETCONN_LEAVE);
  3715. }
  3716. }
  3717. done_socket(sock);
  3718. }
  3719. #endif /* LWIP_IGMP */
  3720. #if LWIP_IPV6_MLD
  3721. /** Register a new MLD6 membership. On socket close, the membership is dropped automatically.
  3722. *
  3723. * ATTENTION: this function is called from tcpip_thread (or under CORE_LOCK).
  3724. *
  3725. * @return 1 on success, 0 on failure
  3726. */
  3727. static int
  3728. lwip_socket_register_mld6_membership(int s, unsigned int if_idx, const ip6_addr_t *multi_addr)
  3729. {
  3730. struct lwip_sock *sock = get_socket(s);
  3731. int i;
  3732. if (!sock) {
  3733. return 0;
  3734. }
  3735. for (i = 0; i < LWIP_SOCKET_MAX_MEMBERSHIPS; i++) {
  3736. if (socket_ipv6_multicast_memberships[i].sock == NULL) {
  3737. socket_ipv6_multicast_memberships[i].sock = sock;
  3738. socket_ipv6_multicast_memberships[i].if_idx = (u8_t)if_idx;
  3739. ip6_addr_copy(socket_ipv6_multicast_memberships[i].multi_addr, *multi_addr);
  3740. done_socket(sock);
  3741. return 1;
  3742. }
  3743. }
  3744. done_socket(sock);
  3745. return 0;
  3746. }
  3747. /** Unregister a previously registered MLD6 membership. This prevents dropping the membership
  3748. * on socket close.
  3749. *
  3750. * ATTENTION: this function is called from tcpip_thread (or under CORE_LOCK).
  3751. */
  3752. static void
  3753. lwip_socket_unregister_mld6_membership(int s, unsigned int if_idx, const ip6_addr_t *multi_addr)
  3754. {
  3755. struct lwip_sock *sock = get_socket(s);
  3756. int i;
  3757. if (!sock) {
  3758. return;
  3759. }
  3760. for (i = 0; i < LWIP_SOCKET_MAX_MEMBERSHIPS; i++) {
  3761. if ((socket_ipv6_multicast_memberships[i].sock == sock) &&
  3762. (socket_ipv6_multicast_memberships[i].if_idx == if_idx) &&
  3763. ip6_addr_cmp(&socket_ipv6_multicast_memberships[i].multi_addr, multi_addr)) {
  3764. socket_ipv6_multicast_memberships[i].sock = NULL;
  3765. socket_ipv6_multicast_memberships[i].if_idx = NETIF_NO_INDEX;
  3766. ip6_addr_set_zero(&socket_ipv6_multicast_memberships[i].multi_addr);
  3767. break;
  3768. }
  3769. }
  3770. done_socket(sock);
  3771. }
  3772. /** Drop all MLD6 memberships of a socket that were not dropped explicitly via setsockopt.
  3773. *
  3774. * ATTENTION: this function is NOT called from tcpip_thread (or under CORE_LOCK).
  3775. */
  3776. static void
  3777. lwip_socket_drop_registered_mld6_memberships(int s)
  3778. {
  3779. struct lwip_sock *sock = get_socket(s);
  3780. int i;
  3781. if (!sock) {
  3782. return;
  3783. }
  3784. for (i = 0; i < LWIP_SOCKET_MAX_MEMBERSHIPS; i++) {
  3785. if (socket_ipv6_multicast_memberships[i].sock == sock) {
  3786. ip_addr_t multi_addr;
  3787. u8_t if_idx;
  3788. ip_addr_copy_from_ip6(multi_addr, socket_ipv6_multicast_memberships[i].multi_addr);
  3789. if_idx = socket_ipv6_multicast_memberships[i].if_idx;
  3790. socket_ipv6_multicast_memberships[i].sock = NULL;
  3791. socket_ipv6_multicast_memberships[i].if_idx = NETIF_NO_INDEX;
  3792. ip6_addr_set_zero(&socket_ipv6_multicast_memberships[i].multi_addr);
  3793. netconn_join_leave_group_netif(sock->conn, &multi_addr, if_idx, NETCONN_LEAVE);
  3794. }
  3795. }
  3796. done_socket(sock);
  3797. }
  3798. #endif /* LWIP_IPV6_MLD */
  3799. #endif /* LWIP_SOCKET */