123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182 |
- /**
- * @file
- * Sockets BSD-Like API module
- */
- /*
- * Copyright (c) 2001-2004 Swedish Institute of Computer Science.
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without modification,
- * are permitted provided that the following conditions are met:
- *
- * 1. Redistributions of source code must retain the above copyright notice,
- * this list of conditions and the following disclaimer.
- * 2. Redistributions in binary form must reproduce the above copyright notice,
- * this list of conditions and the following disclaimer in the documentation
- * and/or other materials provided with the distribution.
- * 3. The name of the author may not be used to endorse or promote products
- * derived from this software without specific prior written permission.
- *
- * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED
- * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
- * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT
- * SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
- * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
- * OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
- * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
- * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
- * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
- * OF SUCH DAMAGE.
- *
- * This file is part of the lwIP TCP/IP stack.
- *
- * Author: Adam Dunkels <adam@sics.se>
- *
- * Improved by Marc Boucher <marc@mbsi.ca> and David Haas <dhaas@alum.rpi.edu>
- *
- */
- #include "lwip/opt.h"
- #if LWIP_SOCKET /* don't build if not configured for use in lwipopts.h */
- #include "lwip/sockets.h"
- #include "lwip/priv/sockets_priv.h"
- #include "lwip/api.h"
- #include "lwip/igmp.h"
- #include "lwip/inet.h"
- #include "lwip/tcp.h"
- #include "lwip/raw.h"
- #include "lwip/udp.h"
- #include "lwip/memp.h"
- #include "lwip/pbuf.h"
- #include "lwip/netif.h"
- #include "lwip/priv/tcpip_priv.h"
- #include "lwip/mld6.h"
- #if LWIP_CHECKSUM_ON_COPY
- #include "lwip/inet_chksum.h"
- #endif
- #if LWIP_COMPAT_SOCKETS == 2 && LWIP_POSIX_SOCKETS_IO_NAMES
- #include <stdarg.h>
- #endif
- #include <string.h>
- #ifdef LWIP_HOOK_FILENAME
- #include LWIP_HOOK_FILENAME
- #endif
- /* If the netconn API is not required publicly, then we include the necessary
- files here to get the implementation */
- #if !LWIP_NETCONN
- #undef LWIP_NETCONN
- #define LWIP_NETCONN 1
- #include "api_msg.c"
- #include "api_lib.c"
- #include "netbuf.c"
- #undef LWIP_NETCONN
- #define LWIP_NETCONN 0
- #endif
- #define API_SELECT_CB_VAR_REF(name) API_VAR_REF(name)
- #define API_SELECT_CB_VAR_DECLARE(name) API_VAR_DECLARE(struct lwip_select_cb, name)
- #define API_SELECT_CB_VAR_ALLOC(name, retblock) API_VAR_ALLOC_EXT(struct lwip_select_cb, MEMP_SELECT_CB, name, retblock)
- #define API_SELECT_CB_VAR_FREE(name) API_VAR_FREE(MEMP_SELECT_CB, name)
- #if LWIP_IPV4
- #define IP4ADDR_PORT_TO_SOCKADDR(sin, ipaddr, port) do { \
- (sin)->sin_len = sizeof(struct sockaddr_in); \
- (sin)->sin_family = AF_INET; \
- (sin)->sin_port = lwip_htons((port)); \
- inet_addr_from_ip4addr(&(sin)->sin_addr, ipaddr); \
- memset((sin)->sin_zero, 0, SIN_ZERO_LEN); }while(0)
- #define SOCKADDR4_TO_IP4ADDR_PORT(sin, ipaddr, port) do { \
- inet_addr_to_ip4addr(ip_2_ip4(ipaddr), &((sin)->sin_addr)); \
- (port) = lwip_ntohs((sin)->sin_port); }while(0)
- #endif /* LWIP_IPV4 */
- #if LWIP_IPV6
- #define IP6ADDR_PORT_TO_SOCKADDR(sin6, ipaddr, port) do { \
- (sin6)->sin6_len = sizeof(struct sockaddr_in6); \
- (sin6)->sin6_family = AF_INET6; \
- (sin6)->sin6_port = lwip_htons((port)); \
- (sin6)->sin6_flowinfo = 0; \
- inet6_addr_from_ip6addr(&(sin6)->sin6_addr, ipaddr); \
- (sin6)->sin6_scope_id = ip6_addr_zone(ipaddr); }while(0)
- #define SOCKADDR6_TO_IP6ADDR_PORT(sin6, ipaddr, port) do { \
- inet6_addr_to_ip6addr(ip_2_ip6(ipaddr), &((sin6)->sin6_addr)); \
- if (ip6_addr_has_scope(ip_2_ip6(ipaddr), IP6_UNKNOWN)) { \
- ip6_addr_set_zone(ip_2_ip6(ipaddr), (u8_t)((sin6)->sin6_scope_id)); \
- } \
- (port) = lwip_ntohs((sin6)->sin6_port); }while(0)
- #endif /* LWIP_IPV6 */
- #if LWIP_IPV4 && LWIP_IPV6
- static void sockaddr_to_ipaddr_port(const struct sockaddr *sockaddr, ip_addr_t *ipaddr, u16_t *port);
- #define IS_SOCK_ADDR_LEN_VALID(namelen) (((namelen) == sizeof(struct sockaddr_in)) || \
- ((namelen) == sizeof(struct sockaddr_in6)))
- #define IS_SOCK_ADDR_TYPE_VALID(name) (((name)->sa_family == AF_INET) || \
- ((name)->sa_family == AF_INET6))
- #define SOCK_ADDR_TYPE_MATCH(name, sock) \
- ((((name)->sa_family == AF_INET) && !(NETCONNTYPE_ISIPV6((sock)->conn->type))) || \
- (((name)->sa_family == AF_INET6) && (NETCONNTYPE_ISIPV6((sock)->conn->type))))
- #define IPADDR_PORT_TO_SOCKADDR(sockaddr, ipaddr, port) do { \
- if (IP_IS_ANY_TYPE_VAL(*ipaddr) || IP_IS_V6_VAL(*ipaddr)) { \
- IP6ADDR_PORT_TO_SOCKADDR((struct sockaddr_in6*)(void*)(sockaddr), ip_2_ip6(ipaddr), port); \
- } else { \
- IP4ADDR_PORT_TO_SOCKADDR((struct sockaddr_in*)(void*)(sockaddr), ip_2_ip4(ipaddr), port); \
- } } while(0)
- #define SOCKADDR_TO_IPADDR_PORT(sockaddr, ipaddr, port) sockaddr_to_ipaddr_port(sockaddr, ipaddr, &(port))
- #define DOMAIN_TO_NETCONN_TYPE(domain, type) (((domain) == AF_INET) ? \
- (type) : (enum netconn_type)((type) | NETCONN_TYPE_IPV6))
- #elif LWIP_IPV6 /* LWIP_IPV4 && LWIP_IPV6 */
- #define IS_SOCK_ADDR_LEN_VALID(namelen) ((namelen) == sizeof(struct sockaddr_in6))
- #define IS_SOCK_ADDR_TYPE_VALID(name) ((name)->sa_family == AF_INET6)
- #define SOCK_ADDR_TYPE_MATCH(name, sock) 1
- #define IPADDR_PORT_TO_SOCKADDR(sockaddr, ipaddr, port) \
- IP6ADDR_PORT_TO_SOCKADDR((struct sockaddr_in6*)(void*)(sockaddr), ip_2_ip6(ipaddr), port)
- #define SOCKADDR_TO_IPADDR_PORT(sockaddr, ipaddr, port) \
- SOCKADDR6_TO_IP6ADDR_PORT((const struct sockaddr_in6*)(const void*)(sockaddr), ipaddr, port)
- #define DOMAIN_TO_NETCONN_TYPE(domain, netconn_type) (netconn_type)
- #else /*-> LWIP_IPV4: LWIP_IPV4 && LWIP_IPV6 */
- #define IS_SOCK_ADDR_LEN_VALID(namelen) ((namelen) == sizeof(struct sockaddr_in))
- #define IS_SOCK_ADDR_TYPE_VALID(name) ((name)->sa_family == AF_INET)
- #define SOCK_ADDR_TYPE_MATCH(name, sock) 1
- #define IPADDR_PORT_TO_SOCKADDR(sockaddr, ipaddr, port) \
- IP4ADDR_PORT_TO_SOCKADDR((struct sockaddr_in*)(void*)(sockaddr), ip_2_ip4(ipaddr), port)
- #define SOCKADDR_TO_IPADDR_PORT(sockaddr, ipaddr, port) \
- SOCKADDR4_TO_IP4ADDR_PORT((const struct sockaddr_in*)(const void*)(sockaddr), ipaddr, port)
- #define DOMAIN_TO_NETCONN_TYPE(domain, netconn_type) (netconn_type)
- #endif /* LWIP_IPV6 */
- #define IS_SOCK_ADDR_TYPE_VALID_OR_UNSPEC(name) (((name)->sa_family == AF_UNSPEC) || \
- IS_SOCK_ADDR_TYPE_VALID(name))
- #define SOCK_ADDR_TYPE_MATCH_OR_UNSPEC(name, sock) (((name)->sa_family == AF_UNSPEC) || \
- SOCK_ADDR_TYPE_MATCH(name, sock))
- #define IS_SOCK_ADDR_ALIGNED(name) ((((mem_ptr_t)(name)) % 4) == 0)
- #define LWIP_SOCKOPT_CHECK_OPTLEN(sock, optlen, opttype) do { if ((optlen) < sizeof(opttype)) { done_socket(sock); return EINVAL; }}while(0)
- #define LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, optlen, opttype) do { \
- LWIP_SOCKOPT_CHECK_OPTLEN(sock, optlen, opttype); \
- if ((sock)->conn == NULL) { done_socket(sock); return EINVAL; } }while(0)
- #define LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, optlen, opttype) do { \
- LWIP_SOCKOPT_CHECK_OPTLEN(sock, optlen, opttype); \
- if (((sock)->conn == NULL) || ((sock)->conn->pcb.tcp == NULL)) { done_socket(sock); return EINVAL; } }while(0)
- #define LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, opttype, netconntype) do { \
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, optlen, opttype); \
- if (NETCONNTYPE_GROUP(netconn_type((sock)->conn)) != netconntype) { done_socket(sock); return ENOPROTOOPT; } }while(0)
- #define LWIP_SETGETSOCKOPT_DATA_VAR_REF(name) API_VAR_REF(name)
- #define LWIP_SETGETSOCKOPT_DATA_VAR_DECLARE(name) API_VAR_DECLARE(struct lwip_setgetsockopt_data, name)
- #define LWIP_SETGETSOCKOPT_DATA_VAR_FREE(name) API_VAR_FREE(MEMP_SOCKET_SETGETSOCKOPT_DATA, name)
- #if LWIP_MPU_COMPATIBLE
- #define LWIP_SETGETSOCKOPT_DATA_VAR_ALLOC(name, sock) do { \
- name = (struct lwip_setgetsockopt_data *)memp_malloc(MEMP_SOCKET_SETGETSOCKOPT_DATA); \
- if (name == NULL) { \
- sock_set_errno(sock, ENOMEM); \
- done_socket(sock); \
- return -1; \
- } }while(0)
- #else /* LWIP_MPU_COMPATIBLE */
- #define LWIP_SETGETSOCKOPT_DATA_VAR_ALLOC(name, sock)
- #endif /* LWIP_MPU_COMPATIBLE */
- #if LWIP_SO_SNDRCVTIMEO_NONSTANDARD
- #define LWIP_SO_SNDRCVTIMEO_OPTTYPE int
- #define LWIP_SO_SNDRCVTIMEO_SET(optval, val) (*(int *)(optval) = (val))
- #define LWIP_SO_SNDRCVTIMEO_GET_MS(optval) ((long)*(const int*)(optval))
- #else
- #define LWIP_SO_SNDRCVTIMEO_OPTTYPE struct timeval
- #define LWIP_SO_SNDRCVTIMEO_SET(optval, val) do { \
- u32_t loc = (val); \
- ((struct timeval *)(optval))->tv_sec = (long)((loc) / 1000U); \
- ((struct timeval *)(optval))->tv_usec = (long)(((loc) % 1000U) * 1000U); }while(0)
- #define LWIP_SO_SNDRCVTIMEO_GET_MS(optval) ((((const struct timeval *)(optval))->tv_sec * 1000) + (((const struct timeval *)(optval))->tv_usec / 1000))
- #endif
- /** A struct sockaddr replacement that has the same alignment as sockaddr_in/
- * sockaddr_in6 if instantiated.
- */
- union sockaddr_aligned {
- struct sockaddr sa;
- #if LWIP_IPV6
- struct sockaddr_in6 sin6;
- #endif /* LWIP_IPV6 */
- #if LWIP_IPV4
- struct sockaddr_in sin;
- #endif /* LWIP_IPV4 */
- };
- /* Define the number of IPv4 multicast memberships, default is one per socket */
- #ifndef LWIP_SOCKET_MAX_MEMBERSHIPS
- #define LWIP_SOCKET_MAX_MEMBERSHIPS NUM_SOCKETS
- #endif
- #if LWIP_IGMP
- /* This is to keep track of IP_ADD_MEMBERSHIP calls to drop the membership when
- a socket is closed */
- struct lwip_socket_multicast_pair {
- /** the socket */
- struct lwip_sock *sock;
- /** the interface address */
- ip4_addr_t if_addr;
- /** the group address */
- ip4_addr_t multi_addr;
- };
- static struct lwip_socket_multicast_pair socket_ipv4_multicast_memberships[LWIP_SOCKET_MAX_MEMBERSHIPS];
- static int lwip_socket_register_membership(int s, const ip4_addr_t *if_addr, const ip4_addr_t *multi_addr);
- static void lwip_socket_unregister_membership(int s, const ip4_addr_t *if_addr, const ip4_addr_t *multi_addr);
- static void lwip_socket_drop_registered_memberships(int s);
- #endif /* LWIP_IGMP */
- #if LWIP_IPV6_MLD
- /* This is to keep track of IP_JOIN_GROUP calls to drop the membership when
- a socket is closed */
- struct lwip_socket_multicast_mld6_pair {
- /** the socket */
- struct lwip_sock *sock;
- /** the interface index */
- u8_t if_idx;
- /** the group address */
- ip6_addr_t multi_addr;
- };
- static struct lwip_socket_multicast_mld6_pair socket_ipv6_multicast_memberships[LWIP_SOCKET_MAX_MEMBERSHIPS];
- static int lwip_socket_register_mld6_membership(int s, unsigned int if_idx, const ip6_addr_t *multi_addr);
- static void lwip_socket_unregister_mld6_membership(int s, unsigned int if_idx, const ip6_addr_t *multi_addr);
- static void lwip_socket_drop_registered_mld6_memberships(int s);
- #endif /* LWIP_IPV6_MLD */
- /** The global array of available sockets */
- static struct lwip_sock sockets[NUM_SOCKETS];
- #if LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL
- #if LWIP_TCPIP_CORE_LOCKING
- /* protect the select_cb_list using core lock */
- #define LWIP_SOCKET_SELECT_DECL_PROTECT(lev)
- #define LWIP_SOCKET_SELECT_PROTECT(lev) LOCK_TCPIP_CORE()
- #define LWIP_SOCKET_SELECT_UNPROTECT(lev) UNLOCK_TCPIP_CORE()
- #else /* LWIP_TCPIP_CORE_LOCKING */
- /* protect the select_cb_list using SYS_LIGHTWEIGHT_PROT */
- #define LWIP_SOCKET_SELECT_DECL_PROTECT(lev) SYS_ARCH_DECL_PROTECT(lev)
- #define LWIP_SOCKET_SELECT_PROTECT(lev) SYS_ARCH_PROTECT(lev)
- #define LWIP_SOCKET_SELECT_UNPROTECT(lev) SYS_ARCH_UNPROTECT(lev)
- /** This counter is increased from lwip_select when the list is changed
- and checked in select_check_waiters to see if it has changed. */
- static volatile int select_cb_ctr;
- #endif /* LWIP_TCPIP_CORE_LOCKING */
- /** The global list of tasks waiting for select */
- static struct lwip_select_cb *select_cb_list;
- #endif /* LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL */
- #define sock_set_errno(sk, e) do { \
- const int sockerr = (e); \
- set_errno(sockerr); \
- } while (0)
- /* Forward declaration of some functions */
- #if LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL
- static void event_callback(struct netconn *conn, enum netconn_evt evt, u16_t len);
- #define DEFAULT_SOCKET_EVENTCB event_callback
- static void select_check_waiters(int s, int has_recvevent, int has_sendevent, int has_errevent);
- #else
- #define DEFAULT_SOCKET_EVENTCB NULL
- #endif
- #if !LWIP_TCPIP_CORE_LOCKING
- static void lwip_getsockopt_callback(void *arg);
- static void lwip_setsockopt_callback(void *arg);
- #endif
- static int lwip_getsockopt_impl(int s, int level, int optname, void *optval, socklen_t *optlen);
- static int lwip_setsockopt_impl(int s, int level, int optname, const void *optval, socklen_t optlen);
- static int free_socket_locked(struct lwip_sock *sock, int is_tcp, struct netconn **conn,
- union lwip_sock_lastdata *lastdata);
- static void free_socket_free_elements(int is_tcp, struct netconn *conn, union lwip_sock_lastdata *lastdata);
- #if LWIP_IPV4 && LWIP_IPV6
- static void
- sockaddr_to_ipaddr_port(const struct sockaddr *sockaddr, ip_addr_t *ipaddr, u16_t *port)
- {
- if ((sockaddr->sa_family) == AF_INET6) {
- SOCKADDR6_TO_IP6ADDR_PORT((const struct sockaddr_in6 *)(const void *)(sockaddr), ipaddr, *port);
- ipaddr->type = IPADDR_TYPE_V6;
- } else {
- SOCKADDR4_TO_IP4ADDR_PORT((const struct sockaddr_in *)(const void *)(sockaddr), ipaddr, *port);
- ipaddr->type = IPADDR_TYPE_V4;
- }
- }
- #endif /* LWIP_IPV4 && LWIP_IPV6 */
- /** LWIP_NETCONN_SEM_PER_THREAD==1: initialize thread-local semaphore */
- void
- lwip_socket_thread_init(void)
- {
- netconn_thread_init();
- }
- /** LWIP_NETCONN_SEM_PER_THREAD==1: destroy thread-local semaphore */
- void
- lwip_socket_thread_cleanup(void)
- {
- netconn_thread_cleanup();
- }
- #if LWIP_NETCONN_FULLDUPLEX
- /* Thread-safe increment of sock->fd_used, with overflow check */
- static int
- sock_inc_used(struct lwip_sock *sock)
- {
- int ret;
- SYS_ARCH_DECL_PROTECT(lev);
- LWIP_ASSERT("sock != NULL", sock != NULL);
- SYS_ARCH_PROTECT(lev);
- if (sock->fd_free_pending) {
- /* prevent new usage of this socket if free is pending */
- ret = 0;
- } else {
- ++sock->fd_used;
- ret = 1;
- LWIP_ASSERT("sock->fd_used != 0", sock->fd_used != 0);
- }
- SYS_ARCH_UNPROTECT(lev);
- return ret;
- }
- /* Like sock_inc_used(), but called under SYS_ARCH_PROTECT lock. */
- static int
- sock_inc_used_locked(struct lwip_sock *sock)
- {
- LWIP_ASSERT("sock != NULL", sock != NULL);
- if (sock->fd_free_pending) {
- LWIP_ASSERT("sock->fd_used != 0", sock->fd_used != 0);
- return 0;
- }
- ++sock->fd_used;
- LWIP_ASSERT("sock->fd_used != 0", sock->fd_used != 0);
- return 1;
- }
- /* In full-duplex mode,sock->fd_used != 0 prevents a socket descriptor from being
- * released (and possibly reused) when used from more than one thread
- * (e.g. read-while-write or close-while-write, etc)
- * This function is called at the end of functions using (try)get_socket*().
- */
- static void
- done_socket(struct lwip_sock *sock)
- {
- int freed = 0;
- int is_tcp = 0;
- struct netconn *conn = NULL;
- union lwip_sock_lastdata lastdata;
- SYS_ARCH_DECL_PROTECT(lev);
- LWIP_ASSERT("sock != NULL", sock != NULL);
- SYS_ARCH_PROTECT(lev);
- LWIP_ASSERT("sock->fd_used > 0", sock->fd_used > 0);
- if (--sock->fd_used == 0) {
- if (sock->fd_free_pending) {
- /* free the socket */
- sock->fd_used = 1;
- is_tcp = sock->fd_free_pending & LWIP_SOCK_FD_FREE_TCP;
- freed = free_socket_locked(sock, is_tcp, &conn, &lastdata);
- }
- }
- SYS_ARCH_UNPROTECT(lev);
- if (freed) {
- free_socket_free_elements(is_tcp, conn, &lastdata);
- }
- }
- #else /* LWIP_NETCONN_FULLDUPLEX */
- #define sock_inc_used(sock) 1
- #define sock_inc_used_locked(sock) 1
- #define done_socket(sock)
- #endif /* LWIP_NETCONN_FULLDUPLEX */
- /* Translate a socket 'int' into a pointer (only fails if the index is invalid) */
- static struct lwip_sock *
- tryget_socket_unconn_nouse(int fd)
- {
- int s = fd - LWIP_SOCKET_OFFSET;
- if ((s < 0) || (s >= NUM_SOCKETS)) {
- LWIP_DEBUGF(SOCKETS_DEBUG, ("tryget_socket_unconn(%d): invalid\n", fd));
- return NULL;
- }
- return &sockets[s];
- }
- struct lwip_sock *
- lwip_socket_dbg_get_socket(int fd)
- {
- return tryget_socket_unconn_nouse(fd);
- }
- /* Translate a socket 'int' into a pointer (only fails if the index is invalid) */
- static struct lwip_sock *
- tryget_socket_unconn(int fd)
- {
- struct lwip_sock *ret = tryget_socket_unconn_nouse(fd);
- if (ret != NULL) {
- if (!sock_inc_used(ret)) {
- return NULL;
- }
- }
- return ret;
- }
- /* Like tryget_socket_unconn(), but called under SYS_ARCH_PROTECT lock. */
- static struct lwip_sock *
- tryget_socket_unconn_locked(int fd)
- {
- struct lwip_sock *ret = tryget_socket_unconn_nouse(fd);
- if (ret != NULL) {
- if (!sock_inc_used_locked(ret)) {
- return NULL;
- }
- }
- return ret;
- }
- /**
- * Same as get_socket but doesn't set errno
- *
- * @param fd externally used socket index
- * @return struct lwip_sock for the socket or NULL if not found
- */
- static struct lwip_sock *
- tryget_socket(int fd)
- {
- struct lwip_sock *sock = tryget_socket_unconn(fd);
- if (sock != NULL) {
- if (sock->conn) {
- return sock;
- }
- done_socket(sock);
- }
- return NULL;
- }
- /**
- * Same as tryget_socket but a global routine.
- *
- * @param fd externally used socket index
- * @return struct lwip_sock for the socket or NULL if not found
- */
- struct lwip_sock *
- lwip_tryget_socket(int fd)
- {
- struct lwip_sock *sock = tryget_socket_unconn(fd);
- if (sock != NULL) {
- if (sock->conn) {
- return sock;
- }
- done_socket(sock);
- }
- return NULL;
- }
- /**
- * Map a externally used socket index to the internal socket representation.
- *
- * @param fd externally used socket index
- * @return struct lwip_sock for the socket or NULL if not found
- */
- static struct lwip_sock *
- get_socket(int fd)
- {
- struct lwip_sock *sock = tryget_socket(fd);
- if (!sock) {
- if ((fd < LWIP_SOCKET_OFFSET) || (fd >= (LWIP_SOCKET_OFFSET + NUM_SOCKETS))) {
- LWIP_DEBUGF(SOCKETS_DEBUG, ("get_socket(%d): invalid\n", fd));
- }
- set_errno(EBADF);
- return NULL;
- }
- return sock;
- }
- /**
- * Allocate a new socket for a given netconn.
- *
- * @param newconn the netconn for which to allocate a socket
- * @param accepted 1 if socket has been created by accept(),
- * 0 if socket has been created by socket()
- * @return the index of the new socket; -1 on error
- */
- static int
- alloc_socket(struct netconn *newconn, int accepted)
- {
- int i;
- SYS_ARCH_DECL_PROTECT(lev);
- LWIP_UNUSED_ARG(accepted);
- /* allocate a new socket identifier */
- for (i = 0; i < NUM_SOCKETS; ++i) {
- /* Protect socket array */
- SYS_ARCH_PROTECT(lev);
- if (!sockets[i].conn) {
- #if LWIP_NETCONN_FULLDUPLEX
- if (sockets[i].fd_used) {
- SYS_ARCH_UNPROTECT(lev);
- continue;
- }
- sockets[i].fd_used = 1;
- sockets[i].fd_free_pending = 0;
- #endif
- sockets[i].conn = newconn;
- /* The socket is not yet known to anyone, so no need to protect
- after having marked it as used. */
- SYS_ARCH_UNPROTECT(lev);
- sockets[i].lastdata.pbuf = NULL;
- #if LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL
- LWIP_ASSERT("sockets[i].select_waiting == 0", sockets[i].select_waiting == 0);
- sockets[i].rcvevent = 0;
- /* TCP sendbuf is empty, but the socket is not yet writable until connected
- * (unless it has been created by accept()). */
- sockets[i].sendevent = (NETCONNTYPE_GROUP(newconn->type) == NETCONN_TCP ? (accepted != 0) : 1);
- sockets[i].errevent = 0;
- #endif /* LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL */
- #ifdef SAL_USING_POSIX
- rt_wqueue_init(&sockets[i].wait_head);
- #endif
- return i + LWIP_SOCKET_OFFSET;
- }
- SYS_ARCH_UNPROTECT(lev);
- }
- return -1;
- }
- /** Free a socket (under lock)
- *
- * @param sock the socket to free
- * @param is_tcp != 0 for TCP sockets, used to free lastdata
- * @param conn the socekt's netconn is stored here, must be freed externally
- * @param lastdata lastdata is stored here, must be freed externally
- */
- static int
- free_socket_locked(struct lwip_sock *sock, int is_tcp, struct netconn **conn,
- union lwip_sock_lastdata *lastdata)
- {
- #if LWIP_NETCONN_FULLDUPLEX
- LWIP_ASSERT("sock->fd_used > 0", sock->fd_used > 0);
- sock->fd_used--;
- if (sock->fd_used > 0) {
- sock->fd_free_pending = LWIP_SOCK_FD_FREE_FREE | (is_tcp ? LWIP_SOCK_FD_FREE_TCP : 0);
- return 0;
- }
- #else /* LWIP_NETCONN_FULLDUPLEX */
- LWIP_UNUSED_ARG(is_tcp);
- #endif /* LWIP_NETCONN_FULLDUPLEX */
- *lastdata = sock->lastdata;
- sock->lastdata.pbuf = NULL;
- *conn = sock->conn;
- sock->conn = NULL;
- return 1;
- }
- /** Free a socket's leftover members.
- */
- static void
- free_socket_free_elements(int is_tcp, struct netconn *conn, union lwip_sock_lastdata *lastdata)
- {
- if (lastdata->pbuf != NULL) {
- if (is_tcp) {
- pbuf_free(lastdata->pbuf);
- } else {
- netbuf_delete(lastdata->netbuf);
- }
- }
- if (conn != NULL) {
- /* netconn_prepare_delete() has already been called, here we only free the conn */
- netconn_delete(conn);
- }
- }
- /** Free a socket. The socket's netconn must have been
- * delete before!
- *
- * @param sock the socket to free
- * @param is_tcp != 0 for TCP sockets, used to free lastdata
- */
- static void
- free_socket(struct lwip_sock *sock, int is_tcp)
- {
- int freed;
- struct netconn *conn;
- union lwip_sock_lastdata lastdata;
- SYS_ARCH_DECL_PROTECT(lev);
- /* Protect socket array */
- SYS_ARCH_PROTECT(lev);
- freed = free_socket_locked(sock, is_tcp, &conn, &lastdata);
- SYS_ARCH_UNPROTECT(lev);
- /* don't use 'sock' after this line, as another task might have allocated it */
- if (freed) {
- free_socket_free_elements(is_tcp, conn, &lastdata);
- }
- }
- /* Below this, the well-known socket functions are implemented.
- * Use google.com or opengroup.org to get a good description :-)
- *
- * Exceptions are documented!
- */
- int
- lwip_accept(int s, struct sockaddr *addr, socklen_t *addrlen)
- {
- struct lwip_sock *sock, *nsock;
- struct netconn *newconn;
- ip_addr_t naddr;
- u16_t port = 0;
- int newsock;
- err_t err;
- int recvevent;
- SYS_ARCH_DECL_PROTECT(lev);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_accept(%d)...\n", s));
- sock = get_socket(s);
- if (!sock) {
- return -1;
- }
- /* wait for a new connection */
- err = netconn_accept(sock->conn, &newconn);
- if (err != ERR_OK) {
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_accept(%d): netconn_acept failed, err=%d\n", s, err));
- if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) != NETCONN_TCP) {
- sock_set_errno(sock, EOPNOTSUPP);
- } else if (err == ERR_CLSD) {
- sock_set_errno(sock, EINVAL);
- } else {
- sock_set_errno(sock, err_to_errno(err));
- }
- done_socket(sock);
- return -1;
- }
- LWIP_ASSERT("newconn != NULL", newconn != NULL);
- newsock = alloc_socket(newconn, 1);
- if (newsock == -1) {
- netconn_delete(newconn);
- sock_set_errno(sock, ENFILE);
- done_socket(sock);
- return -1;
- }
- LWIP_ASSERT("invalid socket index", (newsock >= LWIP_SOCKET_OFFSET) && (newsock < NUM_SOCKETS + LWIP_SOCKET_OFFSET));
- nsock = &sockets[newsock - LWIP_SOCKET_OFFSET];
- /* See event_callback: If data comes in right away after an accept, even
- * though the server task might not have created a new socket yet.
- * In that case, newconn->socket is counted down (newconn->socket--),
- * so nsock->rcvevent is >= 1 here!
- */
- SYS_ARCH_PROTECT(lev);
- recvevent = (s16_t)(-1 - newconn->socket);
- newconn->socket = newsock;
- SYS_ARCH_UNPROTECT(lev);
- if (newconn->callback) {
- LOCK_TCPIP_CORE();
- while (recvevent > 0) {
- recvevent--;
- newconn->callback(newconn, NETCONN_EVT_RCVPLUS, 0);
- }
- UNLOCK_TCPIP_CORE();
- }
- /* Note that POSIX only requires us to check addr is non-NULL. addrlen must
- * not be NULL if addr is valid.
- */
- if ((addr != NULL) && (addrlen != NULL)) {
- union sockaddr_aligned tempaddr;
- /* get the IP address and port of the remote host */
- err = netconn_peer(newconn, &naddr, &port);
- if (err != ERR_OK) {
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_accept(%d): netconn_peer failed, err=%d\n", s, err));
- netconn_delete(newconn);
- free_socket(nsock, 1);
- sock_set_errno(sock, err_to_errno(err));
- done_socket(sock);
- return -1;
- }
- IPADDR_PORT_TO_SOCKADDR(&tempaddr, &naddr, port);
- if (*addrlen > tempaddr.sa.sa_len) {
- *addrlen = tempaddr.sa.sa_len;
- }
- MEMCPY(addr, &tempaddr, *addrlen);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_accept(%d) returning new sock=%d addr=", s, newsock));
- ip_addr_debug_print_val(SOCKETS_DEBUG, naddr);
- LWIP_DEBUGF(SOCKETS_DEBUG, (" port=%"U16_F"\n", port));
- } else {
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_accept(%d) returning new sock=%d", s, newsock));
- }
- sock_set_errno(sock, 0);
- done_socket(sock);
- done_socket(nsock);
- return newsock;
- }
- int
- lwip_bind(int s, const struct sockaddr *name, socklen_t namelen)
- {
- struct lwip_sock *sock;
- ip_addr_t local_addr;
- u16_t local_port;
- err_t err;
- sock = get_socket(s);
- if (!sock) {
- return -1;
- }
- if (!SOCK_ADDR_TYPE_MATCH(name, sock)) {
- /* sockaddr does not match socket type (IPv4/IPv6) */
- sock_set_errno(sock, err_to_errno(ERR_VAL));
- done_socket(sock);
- return -1;
- }
- /* check size, family and alignment of 'name' */
- LWIP_ERROR("lwip_bind: invalid address", (IS_SOCK_ADDR_LEN_VALID(namelen) &&
- IS_SOCK_ADDR_TYPE_VALID(name) && IS_SOCK_ADDR_ALIGNED(name)),
- sock_set_errno(sock, err_to_errno(ERR_ARG)); done_socket(sock); return -1;);
- LWIP_UNUSED_ARG(namelen);
- SOCKADDR_TO_IPADDR_PORT(name, &local_addr, local_port);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_bind(%d, addr=", s));
- ip_addr_debug_print_val(SOCKETS_DEBUG, local_addr);
- LWIP_DEBUGF(SOCKETS_DEBUG, (" port=%"U16_F")\n", local_port));
- #if LWIP_IPV4 && LWIP_IPV6
- /* Dual-stack: Unmap IPv4 mapped IPv6 addresses */
- if (IP_IS_V6_VAL(local_addr) && ip6_addr_isipv4mappedipv6(ip_2_ip6(&local_addr))) {
- unmap_ipv4_mapped_ipv6(ip_2_ip4(&local_addr), ip_2_ip6(&local_addr));
- IP_SET_TYPE_VAL(local_addr, IPADDR_TYPE_V4);
- }
- #endif /* LWIP_IPV4 && LWIP_IPV6 */
- err = netconn_bind(sock->conn, &local_addr, local_port);
- if (err != ERR_OK) {
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_bind(%d) failed, err=%d\n", s, err));
- sock_set_errno(sock, err_to_errno(err));
- done_socket(sock);
- return -1;
- }
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_bind(%d) succeeded\n", s));
- sock_set_errno(sock, 0);
- done_socket(sock);
- return 0;
- }
- int
- lwip_close(int s)
- {
- struct lwip_sock *sock;
- int is_tcp = 0;
- err_t err;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_close(%d)\n", s));
- sock = get_socket(s);
- if (!sock) {
- return -1;
- }
- if (sock->conn != NULL) {
- is_tcp = NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_TCP;
- } else {
- LWIP_ASSERT("sock->lastdata == NULL", sock->lastdata.pbuf == NULL);
- }
- #if LWIP_IGMP
- /* drop all possibly joined IGMP memberships */
- lwip_socket_drop_registered_memberships(s);
- #endif /* LWIP_IGMP */
- #if LWIP_IPV6_MLD
- /* drop all possibly joined MLD6 memberships */
- lwip_socket_drop_registered_mld6_memberships(s);
- #endif /* LWIP_IPV6_MLD */
- err = netconn_prepare_delete(sock->conn);
- if (err != ERR_OK) {
- sock_set_errno(sock, err_to_errno(err));
- done_socket(sock);
- return -1;
- }
- free_socket(sock, is_tcp);
- set_errno(0);
- return 0;
- }
- int
- lwip_connect(int s, const struct sockaddr *name, socklen_t namelen)
- {
- struct lwip_sock *sock;
- err_t err;
- sock = get_socket(s);
- if (!sock) {
- return -1;
- }
- if (!SOCK_ADDR_TYPE_MATCH_OR_UNSPEC(name, sock)) {
- /* sockaddr does not match socket type (IPv4/IPv6) */
- sock_set_errno(sock, err_to_errno(ERR_VAL));
- done_socket(sock);
- return -1;
- }
- LWIP_UNUSED_ARG(namelen);
- if (name->sa_family == AF_UNSPEC) {
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_connect(%d, AF_UNSPEC)\n", s));
- err = netconn_disconnect(sock->conn);
- } else {
- ip_addr_t remote_addr;
- u16_t remote_port;
- /* check size, family and alignment of 'name' */
- LWIP_ERROR("lwip_connect: invalid address", IS_SOCK_ADDR_LEN_VALID(namelen) &&
- IS_SOCK_ADDR_TYPE_VALID_OR_UNSPEC(name) && IS_SOCK_ADDR_ALIGNED(name),
- sock_set_errno(sock, err_to_errno(ERR_ARG)); done_socket(sock); return -1;);
- SOCKADDR_TO_IPADDR_PORT(name, &remote_addr, remote_port);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_connect(%d, addr=", s));
- ip_addr_debug_print_val(SOCKETS_DEBUG, remote_addr);
- LWIP_DEBUGF(SOCKETS_DEBUG, (" port=%"U16_F")\n", remote_port));
- #if LWIP_IPV4 && LWIP_IPV6
- /* Dual-stack: Unmap IPv4 mapped IPv6 addresses */
- if (IP_IS_V6_VAL(remote_addr) && ip6_addr_isipv4mappedipv6(ip_2_ip6(&remote_addr))) {
- unmap_ipv4_mapped_ipv6(ip_2_ip4(&remote_addr), ip_2_ip6(&remote_addr));
- IP_SET_TYPE_VAL(remote_addr, IPADDR_TYPE_V4);
- }
- #endif /* LWIP_IPV4 && LWIP_IPV6 */
- err = netconn_connect(sock->conn, &remote_addr, remote_port);
- }
- if (err != ERR_OK) {
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_connect(%d) failed, err=%d\n", s, err));
- sock_set_errno(sock, err_to_errno(err));
- done_socket(sock);
- return -1;
- }
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_connect(%d) succeeded\n", s));
- sock_set_errno(sock, 0);
- done_socket(sock);
- return 0;
- }
- /**
- * Set a socket into listen mode.
- * The socket may not have been used for another connection previously.
- *
- * @param s the socket to set to listening mode
- * @param backlog (ATTENTION: needs TCP_LISTEN_BACKLOG=1)
- * @return 0 on success, non-zero on failure
- */
- int
- lwip_listen(int s, int backlog)
- {
- struct lwip_sock *sock;
- err_t err;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_listen(%d, backlog=%d)\n", s, backlog));
- sock = get_socket(s);
- if (!sock) {
- return -1;
- }
- /* limit the "backlog" parameter to fit in an u8_t */
- backlog = LWIP_MIN(LWIP_MAX(backlog, 0), 0xff);
- err = netconn_listen_with_backlog(sock->conn, (u8_t)backlog);
- if (err != ERR_OK) {
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_listen(%d) failed, err=%d\n", s, err));
- if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) != NETCONN_TCP) {
- sock_set_errno(sock, EOPNOTSUPP);
- } else {
- sock_set_errno(sock, err_to_errno(err));
- }
- done_socket(sock);
- return -1;
- }
- sock_set_errno(sock, 0);
- done_socket(sock);
- return 0;
- }
- #if LWIP_TCP
- /* Helper function to loop over receiving pbufs from netconn
- * until "len" bytes are received or we're otherwise done.
- * Keeps sock->lastdata for peeking or partly copying.
- */
- static ssize_t
- lwip_recv_tcp(struct lwip_sock *sock, void *mem, size_t len, int flags)
- {
- u8_t apiflags = NETCONN_NOAUTORCVD;
- ssize_t recvd = 0;
- ssize_t recv_left = (len <= SSIZE_MAX) ? (ssize_t)len : SSIZE_MAX;
- LWIP_ASSERT("no socket given", sock != NULL);
- LWIP_ASSERT("this should be checked internally", NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_TCP);
- if (flags & MSG_DONTWAIT) {
- apiflags |= NETCONN_DONTBLOCK;
- }
- do {
- struct pbuf *p;
- err_t err;
- u16_t copylen;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recv_tcp: top while sock->lastdata=%p\n", (void *)sock->lastdata.pbuf));
- /* Check if there is data left from the last recv operation. */
- if (sock->lastdata.pbuf) {
- p = sock->lastdata.pbuf;
- } else {
- /* No data was left from the previous operation, so we try to get
- some from the network. */
- err = netconn_recv_tcp_pbuf_flags(sock->conn, &p, apiflags);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recv_tcp: netconn_recv err=%d, pbuf=%p\n",
- err, (void *)p));
- if (err != ERR_OK) {
- if (recvd > 0) {
- /* already received data, return that (this trusts in getting the same error from
- netconn layer again next time netconn_recv is called) */
- goto lwip_recv_tcp_done;
- }
- /* We should really do some error checking here. */
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recv_tcp: p == NULL, error is \"%s\"!\n",
- lwip_strerr(err)));
- sock_set_errno(sock, err_to_errno(err));
- if (err == ERR_CLSD) {
- return 0;
- } else {
- return -1;
- }
- }
- LWIP_ASSERT("p != NULL", p != NULL);
- sock->lastdata.pbuf = p;
- }
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recv_tcp: buflen=%"U16_F" recv_left=%d off=%d\n",
- p->tot_len, (int)recv_left, (int)recvd));
- if (recv_left > p->tot_len) {
- copylen = p->tot_len;
- } else {
- copylen = (u16_t)recv_left;
- }
- if (recvd + copylen < recvd) {
- /* overflow */
- copylen = (u16_t)(SSIZE_MAX - recvd);
- }
- /* copy the contents of the received buffer into
- the supplied memory pointer mem */
- pbuf_copy_partial(p, (u8_t *)mem + recvd, copylen, 0);
- recvd += copylen;
- /* TCP combines multiple pbufs for one recv */
- LWIP_ASSERT("invalid copylen, len would underflow", recv_left >= copylen);
- recv_left -= copylen;
- /* Unless we peek the incoming message... */
- if ((flags & MSG_PEEK) == 0) {
- /* ... check if there is data left in the pbuf */
- LWIP_ASSERT("invalid copylen", p->tot_len >= copylen);
- if (p->tot_len - copylen > 0) {
- /* If so, it should be saved in the sock structure for the next recv call.
- We store the pbuf but hide/free the consumed data: */
- sock->lastdata.pbuf = pbuf_free_header(p, copylen);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recv_tcp: lastdata now pbuf=%p\n", (void *)sock->lastdata.pbuf));
- } else {
- sock->lastdata.pbuf = NULL;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recv_tcp: deleting pbuf=%p\n", (void *)p));
- pbuf_free(p);
- }
- }
- /* once we have some data to return, only add more if we don't need to wait */
- apiflags |= NETCONN_DONTBLOCK | NETCONN_NOFIN;
- /* @todo: do we need to support peeking more than one pbuf? */
- } while ((recv_left > 0) && !(flags & MSG_PEEK));
- lwip_recv_tcp_done:
- if ((recvd > 0) && !(flags & MSG_PEEK)) {
- /* ensure window update after copying all data */
- netconn_tcp_recvd(sock->conn, (size_t)recvd);
- }
- sock_set_errno(sock, 0);
- return recvd;
- }
- #endif
- /* Convert a netbuf's address data to struct sockaddr */
- static int
- lwip_sock_make_addr(struct netconn *conn, ip_addr_t *fromaddr, u16_t port,
- struct sockaddr *from, socklen_t *fromlen)
- {
- int truncated = 0;
- union sockaddr_aligned saddr;
- LWIP_UNUSED_ARG(conn);
- LWIP_ASSERT("fromaddr != NULL", fromaddr != NULL);
- LWIP_ASSERT("from != NULL", from != NULL);
- LWIP_ASSERT("fromlen != NULL", fromlen != NULL);
- #if LWIP_IPV4 && LWIP_IPV6
- /* Dual-stack: Map IPv4 addresses to IPv4 mapped IPv6 */
- if (NETCONNTYPE_ISIPV6(netconn_type(conn)) && IP_IS_V4(fromaddr)) {
- ip4_2_ipv4_mapped_ipv6(ip_2_ip6(fromaddr), ip_2_ip4(fromaddr));
- IP_SET_TYPE(fromaddr, IPADDR_TYPE_V6);
- }
- #endif /* LWIP_IPV4 && LWIP_IPV6 */
- IPADDR_PORT_TO_SOCKADDR(&saddr, fromaddr, port);
- if (*fromlen < saddr.sa.sa_len) {
- truncated = 1;
- } else if (*fromlen > saddr.sa.sa_len) {
- *fromlen = saddr.sa.sa_len;
- }
- MEMCPY(from, &saddr, *fromlen);
- return truncated;
- }
- #if LWIP_TCP
- /* Helper function to get a tcp socket's remote address info */
- static int
- 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)
- {
- if (sock == NULL) {
- return 0;
- }
- LWIP_UNUSED_ARG(dbg_fn);
- LWIP_UNUSED_ARG(dbg_s);
- LWIP_UNUSED_ARG(dbg_ret);
- #if !SOCKETS_DEBUG
- if (from && fromlen)
- #endif /* !SOCKETS_DEBUG */
- {
- /* get remote addr/port from tcp_pcb */
- u16_t port;
- ip_addr_t tmpaddr;
- netconn_getaddr(sock->conn, &tmpaddr, &port, 0);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("%s(%d): addr=", dbg_fn, dbg_s));
- ip_addr_debug_print_val(SOCKETS_DEBUG, tmpaddr);
- LWIP_DEBUGF(SOCKETS_DEBUG, (" port=%"U16_F" len=%d\n", port, (int)dbg_ret));
- if (from && fromlen) {
- return lwip_sock_make_addr(sock->conn, &tmpaddr, port, from, fromlen);
- }
- }
- return 0;
- }
- #endif
- /* Helper function to receive a netbuf from a udp or raw netconn.
- * Keeps sock->lastdata for peeking.
- */
- static err_t
- lwip_recvfrom_udp_raw(struct lwip_sock *sock, int flags, struct msghdr *msg, u16_t *datagram_len, int dbg_s)
- {
- struct netbuf *buf;
- u8_t apiflags;
- err_t err;
- u16_t buflen, copylen, copied;
- int i;
- LWIP_UNUSED_ARG(dbg_s);
- LWIP_ERROR("lwip_recvfrom_udp_raw: invalid arguments", (msg->msg_iov != NULL) || (msg->msg_iovlen <= 0), return ERR_ARG;);
- if (flags & MSG_DONTWAIT) {
- apiflags = NETCONN_DONTBLOCK;
- } else {
- apiflags = 0;
- }
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recvfrom_udp_raw[UDP/RAW]: top sock->lastdata=%p\n", (void *)sock->lastdata.netbuf));
- /* Check if there is data left from the last recv operation. */
- buf = sock->lastdata.netbuf;
- if (buf == NULL) {
- /* No data was left from the previous operation, so we try to get
- some from the network. */
- err = netconn_recv_udp_raw_netbuf_flags(sock->conn, &buf, apiflags);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recvfrom_udp_raw[UDP/RAW]: netconn_recv err=%d, netbuf=%p\n",
- err, (void *)buf));
- if (err != ERR_OK) {
- return err;
- }
- LWIP_ASSERT("buf != NULL", buf != NULL);
- sock->lastdata.netbuf = buf;
- }
- buflen = buf->p->tot_len;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recvfrom_udp_raw: buflen=%"U16_F"\n", buflen));
- copied = 0;
- /* copy the pbuf payload into the iovs */
- for (i = 0; (i < msg->msg_iovlen) && (copied < buflen); i++) {
- u16_t len_left = (u16_t)(buflen - copied);
- if (msg->msg_iov[i].iov_len > len_left) {
- copylen = len_left;
- } else {
- copylen = (u16_t)msg->msg_iov[i].iov_len;
- }
- /* copy the contents of the received buffer into
- the supplied memory buffer */
- pbuf_copy_partial(buf->p, (u8_t *)msg->msg_iov[i].iov_base, copylen, copied);
- copied = (u16_t)(copied + copylen);
- }
- /* Check to see from where the data was.*/
- #if !SOCKETS_DEBUG
- if (msg->msg_name && msg->msg_namelen)
- #endif /* !SOCKETS_DEBUG */
- {
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recvfrom_udp_raw(%d): addr=", dbg_s));
- ip_addr_debug_print_val(SOCKETS_DEBUG, *netbuf_fromaddr(buf));
- LWIP_DEBUGF(SOCKETS_DEBUG, (" port=%"U16_F" len=%d\n", netbuf_fromport(buf), copied));
- if (msg->msg_name && msg->msg_namelen) {
- lwip_sock_make_addr(sock->conn, netbuf_fromaddr(buf), netbuf_fromport(buf),
- (struct sockaddr *)msg->msg_name, &msg->msg_namelen);
- }
- }
- /* Initialize flag output */
- msg->msg_flags = 0;
- if (msg->msg_control) {
- u8_t wrote_msg = 0;
- #if LWIP_NETBUF_RECVINFO
- /* Check if packet info was recorded */
- if (buf->flags & NETBUF_FLAG_DESTADDR) {
- if (IP_IS_V4(&buf->toaddr)) {
- #if LWIP_IPV4
- if (msg->msg_controllen >= CMSG_SPACE(sizeof(struct in_pktinfo))) {
- struct cmsghdr *chdr = CMSG_FIRSTHDR(msg); /* This will always return a header!! */
- struct in_pktinfo *pkti = (struct in_pktinfo *)CMSG_DATA(chdr);
- chdr->cmsg_level = IPPROTO_IP;
- chdr->cmsg_type = IP_PKTINFO;
- chdr->cmsg_len = CMSG_LEN(sizeof(struct in_pktinfo));
- pkti->ipi_ifindex = buf->p->if_idx;
- inet_addr_from_ip4addr(&pkti->ipi_addr, ip_2_ip4(netbuf_destaddr(buf)));
- msg->msg_controllen = CMSG_SPACE(sizeof(struct in_pktinfo));
- wrote_msg = 1;
- } else {
- msg->msg_flags |= MSG_CTRUNC;
- }
- #endif /* LWIP_IPV4 */
- }
- }
- #endif /* LWIP_NETBUF_RECVINFO */
- if (!wrote_msg) {
- msg->msg_controllen = 0;
- }
- }
- /* If we don't peek the incoming message: zero lastdata pointer and free the netbuf */
- if ((flags & MSG_PEEK) == 0) {
- sock->lastdata.netbuf = NULL;
- netbuf_delete(buf);
- }
- if (datagram_len) {
- *datagram_len = buflen;
- }
- return ERR_OK;
- }
- ssize_t
- lwip_recvfrom(int s, void *mem, size_t len, int flags,
- struct sockaddr *from, socklen_t *fromlen)
- {
- struct lwip_sock *sock;
- ssize_t ret;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recvfrom(%d, %p, %"SZT_F", 0x%x, ..)\n", s, mem, len, flags));
- sock = get_socket(s);
- if (!sock) {
- return -1;
- }
- #if LWIP_TCP
- if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_TCP) {
- ret = lwip_recv_tcp(sock, mem, len, flags);
- lwip_recv_tcp_from(sock, from, fromlen, "lwip_recvfrom", s, ret);
- done_socket(sock);
- return ret;
- } else
- #endif
- {
- u16_t datagram_len = 0;
- struct iovec vec;
- struct msghdr msg;
- err_t err;
- vec.iov_base = mem;
- vec.iov_len = len;
- msg.msg_control = NULL;
- msg.msg_controllen = 0;
- msg.msg_flags = 0;
- msg.msg_iov = &vec;
- msg.msg_iovlen = 1;
- msg.msg_name = from;
- msg.msg_namelen = (fromlen ? *fromlen : 0);
- err = lwip_recvfrom_udp_raw(sock, flags, &msg, &datagram_len, s);
- if (err != ERR_OK) {
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recvfrom[UDP/RAW](%d): buf == NULL, error is \"%s\"!\n",
- s, lwip_strerr(err)));
- sock_set_errno(sock, err_to_errno(err));
- done_socket(sock);
- return -1;
- }
- ret = (ssize_t)LWIP_MIN(LWIP_MIN(len, datagram_len), SSIZE_MAX);
- if (fromlen) {
- *fromlen = msg.msg_namelen;
- }
- }
- sock_set_errno(sock, 0);
- done_socket(sock);
- return ret;
- }
- ssize_t
- lwip_read(int s, void *mem, size_t len)
- {
- return lwip_recvfrom(s, mem, len, 0, NULL, NULL);
- }
- ssize_t
- lwip_readv(int s, const struct iovec *iov, int iovcnt)
- {
- struct msghdr msg;
- msg.msg_name = NULL;
- msg.msg_namelen = 0;
- /* Hack: we have to cast via number to cast from 'const' pointer to non-const.
- Blame the opengroup standard for this inconsistency. */
- msg.msg_iov = LWIP_CONST_CAST(struct iovec *, iov);
- msg.msg_iovlen = iovcnt;
- msg.msg_control = NULL;
- msg.msg_controllen = 0;
- msg.msg_flags = 0;
- return lwip_recvmsg(s, &msg, 0);
- }
- ssize_t
- lwip_recv(int s, void *mem, size_t len, int flags)
- {
- return lwip_recvfrom(s, mem, len, flags, NULL, NULL);
- }
- ssize_t
- lwip_recvmsg(int s, struct msghdr *message, int flags)
- {
- struct lwip_sock *sock;
- int i;
- ssize_t buflen;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recvmsg(%d, message=%p, flags=0x%x)\n", s, (void *)message, flags));
- LWIP_ERROR("lwip_recvmsg: invalid message pointer", message != NULL, return ERR_ARG;);
- LWIP_ERROR("lwip_recvmsg: unsupported flags", (flags & ~(MSG_PEEK|MSG_DONTWAIT)) == 0,
- set_errno(EOPNOTSUPP); return -1;);
- if ((message->msg_iovlen <= 0) || (message->msg_iovlen > IOV_MAX)) {
- set_errno(EMSGSIZE);
- return -1;
- }
- sock = get_socket(s);
- if (!sock) {
- return -1;
- }
- /* check for valid vectors */
- buflen = 0;
- for (i = 0; i < message->msg_iovlen; i++) {
- if ((message->msg_iov[i].iov_base == NULL) || ((ssize_t)message->msg_iov[i].iov_len <= 0) ||
- ((size_t)(ssize_t)message->msg_iov[i].iov_len != message->msg_iov[i].iov_len) ||
- ((ssize_t)(buflen + (ssize_t)message->msg_iov[i].iov_len) <= 0)) {
- sock_set_errno(sock, err_to_errno(ERR_VAL));
- done_socket(sock);
- return -1;
- }
- buflen = (ssize_t)(buflen + (ssize_t)message->msg_iov[i].iov_len);
- }
- if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_TCP) {
- #if LWIP_TCP
- int recv_flags = flags;
- message->msg_flags = 0;
- /* recv the data */
- buflen = 0;
- for (i = 0; i < message->msg_iovlen; i++) {
- /* try to receive into this vector's buffer */
- ssize_t recvd_local = lwip_recv_tcp(sock, message->msg_iov[i].iov_base, message->msg_iov[i].iov_len, recv_flags);
- if (recvd_local > 0) {
- /* sum up received bytes */
- buflen += recvd_local;
- }
- if ((recvd_local < 0) || (recvd_local < (int)message->msg_iov[i].iov_len) ||
- (flags & MSG_PEEK)) {
- /* returned prematurely (or peeking, which might actually be limitated to the first iov) */
- if (buflen <= 0) {
- /* nothing received at all, propagate the error */
- buflen = recvd_local;
- }
- break;
- }
- /* pass MSG_DONTWAIT to lwip_recv_tcp() to prevent waiting for more data */
- recv_flags |= MSG_DONTWAIT;
- }
- if (buflen > 0) {
- /* reset socket error since we have received something */
- sock_set_errno(sock, 0);
- }
- /* " If the socket is connected, the msg_name and msg_namelen members shall be ignored." */
- done_socket(sock);
- return buflen;
- #else /* LWIP_TCP */
- sock_set_errno(sock, err_to_errno(ERR_ARG));
- done_socket(sock);
- return -1;
- #endif /* LWIP_TCP */
- }
- /* else, UDP and RAW NETCONNs */
- #if LWIP_UDP || LWIP_RAW
- {
- u16_t datagram_len = 0;
- err_t err;
- err = lwip_recvfrom_udp_raw(sock, flags, message, &datagram_len, s);
- if (err != ERR_OK) {
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_recvmsg[UDP/RAW](%d): buf == NULL, error is \"%s\"!\n",
- s, lwip_strerr(err)));
- sock_set_errno(sock, err_to_errno(err));
- done_socket(sock);
- return -1;
- }
- if (datagram_len > buflen) {
- message->msg_flags |= MSG_TRUNC;
- }
- sock_set_errno(sock, 0);
- done_socket(sock);
- return (int)datagram_len;
- }
- #else /* LWIP_UDP || LWIP_RAW */
- sock_set_errno(sock, err_to_errno(ERR_ARG));
- done_socket(sock);
- return -1;
- #endif /* LWIP_UDP || LWIP_RAW */
- }
- ssize_t
- lwip_send(int s, const void *data, size_t size, int flags)
- {
- struct lwip_sock *sock;
- err_t err;
- u8_t write_flags;
- size_t written;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_send(%d, data=%p, size=%"SZT_F", flags=0x%x)\n",
- s, data, size, flags));
- sock = get_socket(s);
- if (!sock) {
- return -1;
- }
- if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) != NETCONN_TCP) {
- #if (LWIP_UDP || LWIP_RAW)
- done_socket(sock);
- return lwip_sendto(s, data, size, flags, NULL, 0);
- #else /* (LWIP_UDP || LWIP_RAW) */
- sock_set_errno(sock, err_to_errno(ERR_ARG));
- done_socket(sock);
- return -1;
- #endif /* (LWIP_UDP || LWIP_RAW) */
- }
- write_flags = (u8_t)(NETCONN_COPY |
- ((flags & MSG_MORE) ? NETCONN_MORE : 0) |
- ((flags & MSG_DONTWAIT) ? NETCONN_DONTBLOCK : 0));
- written = 0;
- err = netconn_write_partly(sock->conn, data, size, write_flags, &written);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_send(%d) err=%d written=%"SZT_F"\n", s, err, written));
- sock_set_errno(sock, err_to_errno(err));
- done_socket(sock);
- /* casting 'written' to ssize_t is OK here since the netconn API limits it to SSIZE_MAX */
- return (err == ERR_OK ? (ssize_t)written : -1);
- }
- ssize_t
- lwip_sendmsg(int s, const struct msghdr *msg, int flags)
- {
- struct lwip_sock *sock;
- #if LWIP_TCP
- u8_t write_flags;
- size_t written;
- #endif
- err_t err = ERR_OK;
- sock = get_socket(s);
- if (!sock) {
- return -1;
- }
- LWIP_ERROR("lwip_sendmsg: invalid msghdr", msg != NULL,
- sock_set_errno(sock, err_to_errno(ERR_ARG)); done_socket(sock); return -1;);
- LWIP_ERROR("lwip_sendmsg: invalid msghdr iov", msg->msg_iov != NULL,
- sock_set_errno(sock, err_to_errno(ERR_ARG)); done_socket(sock); return -1;);
- LWIP_ERROR("lwip_sendmsg: maximum iovs exceeded", (msg->msg_iovlen > 0) && (msg->msg_iovlen <= IOV_MAX),
- sock_set_errno(sock, EMSGSIZE); done_socket(sock); return -1;);
- LWIP_ERROR("lwip_sendmsg: unsupported flags", (flags & ~(MSG_DONTWAIT | MSG_MORE)) == 0,
- sock_set_errno(sock, EOPNOTSUPP); done_socket(sock); return -1;);
- LWIP_UNUSED_ARG(msg->msg_control);
- LWIP_UNUSED_ARG(msg->msg_controllen);
- LWIP_UNUSED_ARG(msg->msg_flags);
- if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_TCP) {
- #if LWIP_TCP
- write_flags = (u8_t)(NETCONN_COPY |
- ((flags & MSG_MORE) ? NETCONN_MORE : 0) |
- ((flags & MSG_DONTWAIT) ? NETCONN_DONTBLOCK : 0));
- written = 0;
- err = netconn_write_vectors_partly(sock->conn, (struct netvector *)msg->msg_iov, (u16_t)msg->msg_iovlen, write_flags, &written);
- sock_set_errno(sock, err_to_errno(err));
- done_socket(sock);
- /* casting 'written' to ssize_t is OK here since the netconn API limits it to SSIZE_MAX */
- return (err == ERR_OK ? (ssize_t)written : -1);
- #else /* LWIP_TCP */
- sock_set_errno(sock, err_to_errno(ERR_ARG));
- done_socket(sock);
- return -1;
- #endif /* LWIP_TCP */
- }
- /* else, UDP and RAW NETCONNs */
- #if LWIP_UDP || LWIP_RAW
- {
- struct netbuf chain_buf;
- int i;
- ssize_t size = 0;
- LWIP_UNUSED_ARG(flags);
- LWIP_ERROR("lwip_sendmsg: invalid msghdr name", (((msg->msg_name == NULL) && (msg->msg_namelen == 0)) ||
- IS_SOCK_ADDR_LEN_VALID(msg->msg_namelen)),
- sock_set_errno(sock, err_to_errno(ERR_ARG)); done_socket(sock); return -1;);
- /* initialize chain buffer with destination */
- memset(&chain_buf, 0, sizeof(struct netbuf));
- if (msg->msg_name) {
- u16_t remote_port;
- SOCKADDR_TO_IPADDR_PORT((const struct sockaddr *)msg->msg_name, &chain_buf.addr, remote_port);
- netbuf_fromport(&chain_buf) = remote_port;
- }
- #if LWIP_NETIF_TX_SINGLE_PBUF
- for (i = 0; i < msg->msg_iovlen; i++) {
- size += msg->msg_iov[i].iov_len;
- if ((msg->msg_iov[i].iov_len > INT_MAX) || (size < (int)msg->msg_iov[i].iov_len)) {
- /* overflow */
- goto sendmsg_emsgsize;
- }
- }
- if (size > 0xFFFF) {
- /* overflow */
- goto sendmsg_emsgsize;
- }
- /* Allocate a new netbuf and copy the data into it. */
- if (netbuf_alloc(&chain_buf, (u16_t)size) == NULL) {
- err = ERR_MEM;
- } else {
- /* flatten the IO vectors */
- size_t offset = 0;
- for (i = 0; i < msg->msg_iovlen; i++) {
- MEMCPY(&((u8_t *)chain_buf.p->payload)[offset], msg->msg_iov[i].iov_base, msg->msg_iov[i].iov_len);
- offset += msg->msg_iov[i].iov_len;
- }
- #if LWIP_CHECKSUM_ON_COPY
- {
- /* This can be improved by using LWIP_CHKSUM_COPY() and aggregating the checksum for each IO vector */
- u16_t chksum = ~inet_chksum_pbuf(chain_buf.p);
- netbuf_set_chksum(&chain_buf, chksum);
- }
- #endif /* LWIP_CHECKSUM_ON_COPY */
- err = ERR_OK;
- }
- #else /* LWIP_NETIF_TX_SINGLE_PBUF */
- /* create a chained netbuf from the IO vectors. NOTE: we assemble a pbuf chain
- manually to avoid having to allocate, chain, and delete a netbuf for each iov */
- for (i = 0; i < msg->msg_iovlen; i++) {
- struct pbuf *p;
- if (msg->msg_iov[i].iov_len > 0xFFFF) {
- /* overflow */
- goto sendmsg_emsgsize;
- }
- p = pbuf_alloc(PBUF_TRANSPORT, 0, PBUF_REF);
- if (p == NULL) {
- err = ERR_MEM; /* let netbuf_delete() cleanup chain_buf */
- break;
- }
- p->payload = msg->msg_iov[i].iov_base;
- p->len = p->tot_len = (u16_t)msg->msg_iov[i].iov_len;
- /* netbuf empty, add new pbuf */
- if (chain_buf.p == NULL) {
- chain_buf.p = chain_buf.ptr = p;
- /* add pbuf to existing pbuf chain */
- } else {
- if (chain_buf.p->tot_len + p->len > 0xffff) {
- /* overflow */
- pbuf_free(p);
- goto sendmsg_emsgsize;
- }
- pbuf_cat(chain_buf.p, p);
- }
- }
- /* save size of total chain */
- if (err == ERR_OK) {
- size = netbuf_len(&chain_buf);
- }
- #endif /* LWIP_NETIF_TX_SINGLE_PBUF */
- if (err == ERR_OK) {
- #if LWIP_IPV4 && LWIP_IPV6
- /* Dual-stack: Unmap IPv4 mapped IPv6 addresses */
- if (IP_IS_V6_VAL(chain_buf.addr) && ip6_addr_isipv4mappedipv6(ip_2_ip6(&chain_buf.addr))) {
- unmap_ipv4_mapped_ipv6(ip_2_ip4(&chain_buf.addr), ip_2_ip6(&chain_buf.addr));
- IP_SET_TYPE_VAL(chain_buf.addr, IPADDR_TYPE_V4);
- }
- #endif /* LWIP_IPV4 && LWIP_IPV6 */
- /* send the data */
- err = netconn_send(sock->conn, &chain_buf);
- }
- /* deallocated the buffer */
- netbuf_free(&chain_buf);
- sock_set_errno(sock, err_to_errno(err));
- done_socket(sock);
- return (err == ERR_OK ? size : -1);
- sendmsg_emsgsize:
- sock_set_errno(sock, EMSGSIZE);
- netbuf_free(&chain_buf);
- done_socket(sock);
- return -1;
- }
- #else /* LWIP_UDP || LWIP_RAW */
- sock_set_errno(sock, err_to_errno(ERR_ARG));
- done_socket(sock);
- return -1;
- #endif /* LWIP_UDP || LWIP_RAW */
- }
- ssize_t
- lwip_sendto(int s, const void *data, size_t size, int flags,
- const struct sockaddr *to, socklen_t tolen)
- {
- struct lwip_sock *sock;
- err_t err;
- u16_t short_size;
- u16_t remote_port;
- struct netbuf buf;
- sock = get_socket(s);
- if (!sock) {
- return -1;
- }
- if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_TCP) {
- #if LWIP_TCP
- done_socket(sock);
- return lwip_send(s, data, size, flags);
- #else /* LWIP_TCP */
- LWIP_UNUSED_ARG(flags);
- sock_set_errno(sock, err_to_errno(ERR_ARG));
- done_socket(sock);
- return -1;
- #endif /* LWIP_TCP */
- }
- if (size > LWIP_MIN(0xFFFF, SSIZE_MAX)) {
- /* cannot fit into one datagram (at least for us) */
- sock_set_errno(sock, EMSGSIZE);
- done_socket(sock);
- return -1;
- }
- short_size = (u16_t)size;
- LWIP_ERROR("lwip_sendto: invalid address", (((to == NULL) && (tolen == 0)) ||
- (IS_SOCK_ADDR_LEN_VALID(tolen) &&
- ((to != NULL) && (IS_SOCK_ADDR_TYPE_VALID(to) && IS_SOCK_ADDR_ALIGNED(to))))),
- sock_set_errno(sock, err_to_errno(ERR_ARG)); done_socket(sock); return -1;);
- LWIP_UNUSED_ARG(tolen);
- /* initialize a buffer */
- buf.p = buf.ptr = NULL;
- #if LWIP_CHECKSUM_ON_COPY
- buf.flags = 0;
- #endif /* LWIP_CHECKSUM_ON_COPY */
- if (to) {
- SOCKADDR_TO_IPADDR_PORT(to, &buf.addr, remote_port);
- } else {
- remote_port = 0;
- ip_addr_set_any(NETCONNTYPE_ISIPV6(netconn_type(sock->conn)), &buf.addr);
- }
- netbuf_fromport(&buf) = remote_port;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_sendto(%d, data=%p, short_size=%"U16_F", flags=0x%x to=",
- s, data, short_size, flags));
- ip_addr_debug_print_val(SOCKETS_DEBUG, buf.addr);
- LWIP_DEBUGF(SOCKETS_DEBUG, (" port=%"U16_F"\n", remote_port));
- /* make the buffer point to the data that should be sent */
- #if LWIP_NETIF_TX_SINGLE_PBUF
- /* Allocate a new netbuf and copy the data into it. */
- if (netbuf_alloc(&buf, short_size) == NULL) {
- err = ERR_MEM;
- } else {
- #if LWIP_CHECKSUM_ON_COPY
- if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) != NETCONN_RAW) {
- u16_t chksum = LWIP_CHKSUM_COPY(buf.p->payload, data, short_size);
- netbuf_set_chksum(&buf, chksum);
- } else
- #endif /* LWIP_CHECKSUM_ON_COPY */
- {
- MEMCPY(buf.p->payload, data, short_size);
- }
- err = ERR_OK;
- }
- #else /* LWIP_NETIF_TX_SINGLE_PBUF */
- err = netbuf_ref(&buf, data, short_size);
- #endif /* LWIP_NETIF_TX_SINGLE_PBUF */
- if (err == ERR_OK) {
- #if LWIP_IPV4 && LWIP_IPV6
- /* Dual-stack: Unmap IPv4 mapped IPv6 addresses */
- if (IP_IS_V6_VAL(buf.addr) && ip6_addr_isipv4mappedipv6(ip_2_ip6(&buf.addr))) {
- unmap_ipv4_mapped_ipv6(ip_2_ip4(&buf.addr), ip_2_ip6(&buf.addr));
- IP_SET_TYPE_VAL(buf.addr, IPADDR_TYPE_V4);
- }
- #endif /* LWIP_IPV4 && LWIP_IPV6 */
- /* send the data */
- err = netconn_send(sock->conn, &buf);
- }
- /* deallocated the buffer */
- netbuf_free(&buf);
- sock_set_errno(sock, err_to_errno(err));
- done_socket(sock);
- return (err == ERR_OK ? short_size : -1);
- }
- int
- lwip_socket(int domain, int type, int protocol)
- {
- struct netconn *conn;
- int i;
- LWIP_UNUSED_ARG(domain); /* @todo: check this */
- /* create a netconn */
- switch (type) {
- case SOCK_RAW:
- conn = netconn_new_with_proto_and_callback(DOMAIN_TO_NETCONN_TYPE(domain, NETCONN_RAW),
- (u8_t)protocol, DEFAULT_SOCKET_EVENTCB);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_socket(%s, SOCK_RAW, %d) = ",
- domain == PF_INET ? "PF_INET" : "UNKNOWN", protocol));
- break;
- case SOCK_DGRAM:
- conn = netconn_new_with_callback(DOMAIN_TO_NETCONN_TYPE(domain,
- ((protocol == IPPROTO_UDPLITE) ? NETCONN_UDPLITE : NETCONN_UDP)),
- DEFAULT_SOCKET_EVENTCB);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_socket(%s, SOCK_DGRAM, %d) = ",
- domain == PF_INET ? "PF_INET" : "UNKNOWN", protocol));
- #if LWIP_NETBUF_RECVINFO
- if (conn) {
- /* netconn layer enables pktinfo by default, sockets default to off */
- conn->flags &= ~NETCONN_FLAG_PKTINFO;
- }
- #endif /* LWIP_NETBUF_RECVINFO */
- break;
- case SOCK_STREAM:
- conn = netconn_new_with_callback(DOMAIN_TO_NETCONN_TYPE(domain, NETCONN_TCP), DEFAULT_SOCKET_EVENTCB);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_socket(%s, SOCK_STREAM, %d) = ",
- domain == PF_INET ? "PF_INET" : "UNKNOWN", protocol));
- break;
- default:
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_socket(%d, %d/UNKNOWN, %d) = -1\n",
- domain, type, protocol));
- set_errno(EINVAL);
- return -1;
- }
- if (!conn) {
- LWIP_DEBUGF(SOCKETS_DEBUG, ("-1 / ENOBUFS (could not create netconn)\n"));
- set_errno(ENOBUFS);
- return -1;
- }
- i = alloc_socket(conn, 0);
- if (i == -1) {
- netconn_delete(conn);
- set_errno(ENFILE);
- return -1;
- }
- conn->socket = i;
- done_socket(&sockets[i - LWIP_SOCKET_OFFSET]);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("%d\n", i));
- set_errno(0);
- return i;
- }
- ssize_t
- lwip_write(int s, const void *data, size_t size)
- {
- return lwip_send(s, data, size, 0);
- }
- ssize_t
- lwip_writev(int s, const struct iovec *iov, int iovcnt)
- {
- struct msghdr msg;
- msg.msg_name = NULL;
- msg.msg_namelen = 0;
- /* Hack: we have to cast via number to cast from 'const' pointer to non-const.
- Blame the opengroup standard for this inconsistency. */
- msg.msg_iov = LWIP_CONST_CAST(struct iovec *, iov);
- msg.msg_iovlen = iovcnt;
- msg.msg_control = NULL;
- msg.msg_controllen = 0;
- msg.msg_flags = 0;
- return lwip_sendmsg(s, &msg, 0);
- }
- #if LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL
- /* Add select_cb to select_cb_list. */
- static void
- lwip_link_select_cb(struct lwip_select_cb *select_cb)
- {
- LWIP_SOCKET_SELECT_DECL_PROTECT(lev);
- /* Protect the select_cb_list */
- LWIP_SOCKET_SELECT_PROTECT(lev);
- /* Put this select_cb on top of list */
- select_cb->next = select_cb_list;
- if (select_cb_list != NULL) {
- select_cb_list->prev = select_cb;
- }
- select_cb_list = select_cb;
- #if !LWIP_TCPIP_CORE_LOCKING
- /* Increasing this counter tells select_check_waiters that the list has changed. */
- select_cb_ctr++;
- #endif
- /* Now we can safely unprotect */
- LWIP_SOCKET_SELECT_UNPROTECT(lev);
- }
- /* Remove select_cb from select_cb_list. */
- static void
- lwip_unlink_select_cb(struct lwip_select_cb *select_cb)
- {
- LWIP_SOCKET_SELECT_DECL_PROTECT(lev);
- /* Take us off the list */
- LWIP_SOCKET_SELECT_PROTECT(lev);
- if (select_cb->next != NULL) {
- select_cb->next->prev = select_cb->prev;
- }
- if (select_cb_list == select_cb) {
- LWIP_ASSERT("select_cb->prev == NULL", select_cb->prev == NULL);
- select_cb_list = select_cb->next;
- } else {
- LWIP_ASSERT("select_cb->prev != NULL", select_cb->prev != NULL);
- select_cb->prev->next = select_cb->next;
- }
- #if !LWIP_TCPIP_CORE_LOCKING
- /* Increasing this counter tells select_check_waiters that the list has changed. */
- select_cb_ctr++;
- #endif
- LWIP_SOCKET_SELECT_UNPROTECT(lev);
- }
- #endif /* LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL */
- #if LWIP_SOCKET_SELECT
- /**
- * Go through the readset and writeset lists and see which socket of the sockets
- * set in the sets has events. On return, readset, writeset and exceptset have
- * the sockets enabled that had events.
- *
- * @param maxfdp1 the highest socket index in the sets
- * @param readset_in set of sockets to check for read events
- * @param writeset_in set of sockets to check for write events
- * @param exceptset_in set of sockets to check for error events
- * @param readset_out set of sockets that had read events
- * @param writeset_out set of sockets that had write events
- * @param exceptset_out set os sockets that had error events
- * @return number of sockets that had events (read/write/exception) (>= 0)
- */
- static int
- lwip_selscan(int maxfdp1, fd_set *readset_in, fd_set *writeset_in, fd_set *exceptset_in,
- fd_set *readset_out, fd_set *writeset_out, fd_set *exceptset_out)
- {
- int i, nready = 0;
- fd_set lreadset, lwriteset, lexceptset;
- struct lwip_sock *sock;
- SYS_ARCH_DECL_PROTECT(lev);
- FD_ZERO(&lreadset);
- FD_ZERO(&lwriteset);
- FD_ZERO(&lexceptset);
- /* Go through each socket in each list to count number of sockets which
- currently match */
- for (i = LWIP_SOCKET_OFFSET; i < maxfdp1; i++) {
- /* if this FD is not in the set, continue */
- if (!(readset_in && FD_ISSET(i, readset_in)) &&
- !(writeset_in && FD_ISSET(i, writeset_in)) &&
- !(exceptset_in && FD_ISSET(i, exceptset_in))) {
- continue;
- }
- /* First get the socket's status (protected)... */
- SYS_ARCH_PROTECT(lev);
- sock = tryget_socket_unconn_locked(i);
- if (sock != NULL) {
- void *lastdata = sock->lastdata.pbuf;
- s16_t rcvevent = sock->rcvevent;
- u16_t sendevent = sock->sendevent;
- u16_t errevent = sock->errevent;
- SYS_ARCH_UNPROTECT(lev);
- /* ... then examine it: */
- /* See if netconn of this socket is ready for read */
- if (readset_in && FD_ISSET(i, readset_in) && ((lastdata != NULL) || (rcvevent > 0))) {
- FD_SET(i, &lreadset);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_selscan: fd=%d ready for reading\n", i));
- nready++;
- }
- /* See if netconn of this socket is ready for write */
- if (writeset_in && FD_ISSET(i, writeset_in) && (sendevent != 0)) {
- FD_SET(i, &lwriteset);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_selscan: fd=%d ready for writing\n", i));
- nready++;
- }
- /* See if netconn of this socket had an error */
- if (exceptset_in && FD_ISSET(i, exceptset_in) && (errevent != 0)) {
- FD_SET(i, &lexceptset);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_selscan: fd=%d ready for exception\n", i));
- nready++;
- }
- done_socket(sock);
- } else {
- SYS_ARCH_UNPROTECT(lev);
- /* no a valid open socket */
- return -1;
- }
- }
- /* copy local sets to the ones provided as arguments */
- *readset_out = lreadset;
- *writeset_out = lwriteset;
- *exceptset_out = lexceptset;
- LWIP_ASSERT("nready >= 0", nready >= 0);
- return nready;
- }
- #if LWIP_NETCONN_FULLDUPLEX
- /* Mark all of the set sockets in one of the three fdsets passed to select as used.
- * All sockets are marked (and later unmarked), whether they are open or not.
- * This is OK as lwip_selscan aborts select when non-open sockets are found.
- */
- static void
- lwip_select_inc_sockets_used_set(int maxfdp, fd_set *fdset, fd_set *used_sockets)
- {
- SYS_ARCH_DECL_PROTECT(lev);
- if (fdset) {
- int i;
- for (i = LWIP_SOCKET_OFFSET; i < maxfdp; i++) {
- /* if this FD is in the set, lock it (unless already done) */
- if (FD_ISSET(i, fdset) && !FD_ISSET(i, used_sockets)) {
- struct lwip_sock *sock;
- SYS_ARCH_PROTECT(lev);
- sock = tryget_socket_unconn_locked(i);
- if (sock != NULL) {
- /* leave the socket used until released by lwip_select_dec_sockets_used */
- FD_SET(i, used_sockets);
- }
- SYS_ARCH_UNPROTECT(lev);
- }
- }
- }
- }
- /* Mark all sockets passed to select as used to prevent them from being freed
- * from other threads while select is running.
- * Marked sockets are added to 'used_sockets' to mark them only once an be able
- * to unmark them correctly.
- */
- static void
- lwip_select_inc_sockets_used(int maxfdp, fd_set *fdset1, fd_set *fdset2, fd_set *fdset3, fd_set *used_sockets)
- {
- FD_ZERO(used_sockets);
- lwip_select_inc_sockets_used_set(maxfdp, fdset1, used_sockets);
- lwip_select_inc_sockets_used_set(maxfdp, fdset2, used_sockets);
- lwip_select_inc_sockets_used_set(maxfdp, fdset3, used_sockets);
- }
- /* Let go all sockets that were marked as used when starting select */
- static void
- lwip_select_dec_sockets_used(int maxfdp, fd_set *used_sockets)
- {
- int i;
- for (i = LWIP_SOCKET_OFFSET; i < maxfdp; i++) {
- /* if this FD is not in the set, continue */
- if (FD_ISSET(i, used_sockets)) {
- struct lwip_sock *sock = tryget_socket_unconn_nouse(i);
- LWIP_ASSERT("socket gone at the end of select", sock != NULL);
- if (sock != NULL) {
- done_socket(sock);
- }
- }
- }
- }
- #else /* LWIP_NETCONN_FULLDUPLEX */
- #define lwip_select_inc_sockets_used(maxfdp1, readset, writeset, exceptset, used_sockets)
- #define lwip_select_dec_sockets_used(maxfdp1, used_sockets)
- #endif /* LWIP_NETCONN_FULLDUPLEX */
- int
- lwip_select(int maxfdp1, fd_set *readset, fd_set *writeset, fd_set *exceptset,
- struct timeval *timeout)
- {
- u32_t waitres = 0;
- int nready;
- fd_set lreadset, lwriteset, lexceptset;
- u32_t msectimeout;
- int i;
- int maxfdp2;
- #if LWIP_NETCONN_SEM_PER_THREAD
- int waited = 0;
- #endif
- #if LWIP_NETCONN_FULLDUPLEX
- fd_set used_sockets;
- #endif
- SYS_ARCH_DECL_PROTECT(lev);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_select(%d, %p, %p, %p, tvsec=%"S32_F" tvusec=%"S32_F")\n",
- maxfdp1, (void *)readset, (void *) writeset, (void *) exceptset,
- timeout ? (s32_t)timeout->tv_sec : (s32_t) - 1,
- timeout ? (s32_t)timeout->tv_usec : (s32_t) - 1));
- if ((maxfdp1 < 0) || (maxfdp1 > LWIP_SELECT_MAXNFDS)) {
- set_errno(EINVAL);
- return -1;
- }
- lwip_select_inc_sockets_used(maxfdp1, readset, writeset, exceptset, &used_sockets);
- /* Go through each socket in each list to count number of sockets which
- currently match */
- nready = lwip_selscan(maxfdp1, readset, writeset, exceptset, &lreadset, &lwriteset, &lexceptset);
- if (nready < 0) {
- /* one of the sockets in one of the fd_sets was invalid */
- set_errno(EBADF);
- lwip_select_dec_sockets_used(maxfdp1, &used_sockets);
- return -1;
- } else if (nready > 0) {
- /* one or more sockets are set, no need to wait */
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_select: nready=%d\n", nready));
- } else {
- /* If we don't have any current events, then suspend if we are supposed to */
- if (timeout && timeout->tv_sec == 0 && timeout->tv_usec == 0) {
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_select: no timeout, returning 0\n"));
- /* This is OK as the local fdsets are empty and nready is zero,
- or we would have returned earlier. */
- } else {
- /* None ready: add our semaphore to list:
- We don't actually need any dynamic memory. Our entry on the
- list is only valid while we are in this function, so it's ok
- to use local variables (unless we're running in MPU compatible
- mode). */
- API_SELECT_CB_VAR_DECLARE(select_cb);
- API_SELECT_CB_VAR_ALLOC(select_cb, set_errno(ENOMEM); lwip_select_dec_sockets_used(maxfdp1, &used_sockets); return -1);
- memset(&API_SELECT_CB_VAR_REF(select_cb), 0, sizeof(struct lwip_select_cb));
- API_SELECT_CB_VAR_REF(select_cb).readset = readset;
- API_SELECT_CB_VAR_REF(select_cb).writeset = writeset;
- API_SELECT_CB_VAR_REF(select_cb).exceptset = exceptset;
- #if LWIP_NETCONN_SEM_PER_THREAD
- API_SELECT_CB_VAR_REF(select_cb).sem = LWIP_NETCONN_THREAD_SEM_GET();
- #else /* LWIP_NETCONN_SEM_PER_THREAD */
- if (sys_sem_new(&API_SELECT_CB_VAR_REF(select_cb).sem, 0) != ERR_OK) {
- /* failed to create semaphore */
- set_errno(ENOMEM);
- lwip_select_dec_sockets_used(maxfdp1, &used_sockets);
- API_SELECT_CB_VAR_FREE(select_cb);
- return -1;
- }
- #endif /* LWIP_NETCONN_SEM_PER_THREAD */
- lwip_link_select_cb(&API_SELECT_CB_VAR_REF(select_cb));
- /* Increase select_waiting for each socket we are interested in */
- maxfdp2 = maxfdp1;
- for (i = LWIP_SOCKET_OFFSET; i < maxfdp1; i++) {
- if ((readset && FD_ISSET(i, readset)) ||
- (writeset && FD_ISSET(i, writeset)) ||
- (exceptset && FD_ISSET(i, exceptset))) {
- struct lwip_sock *sock;
- SYS_ARCH_PROTECT(lev);
- sock = tryget_socket_unconn_locked(i);
- if (sock != NULL) {
- sock->select_waiting++;
- if (sock->select_waiting == 0) {
- /* overflow - too many threads waiting */
- sock->select_waiting--;
- nready = -1;
- maxfdp2 = i;
- SYS_ARCH_UNPROTECT(lev);
- done_socket(sock);
- set_errno(EBUSY);
- break;
- }
- SYS_ARCH_UNPROTECT(lev);
- done_socket(sock);
- } else {
- /* Not a valid socket */
- nready = -1;
- maxfdp2 = i;
- SYS_ARCH_UNPROTECT(lev);
- set_errno(EBADF);
- break;
- }
- }
- }
- if (nready >= 0) {
- /* Call lwip_selscan again: there could have been events between
- the last scan (without us on the list) and putting us on the list! */
- nready = lwip_selscan(maxfdp1, readset, writeset, exceptset, &lreadset, &lwriteset, &lexceptset);
- if (!nready) {
- /* Still none ready, just wait to be woken */
- if (timeout == 0) {
- /* Wait forever */
- msectimeout = 0;
- } else {
- long msecs_long = ((timeout->tv_sec * 1000) + ((timeout->tv_usec + 500) / 1000));
- if (msecs_long <= 0) {
- /* Wait 1ms at least (0 means wait forever) */
- msectimeout = 1;
- } else {
- msectimeout = (u32_t)msecs_long;
- }
- }
- waitres = sys_arch_sem_wait(SELECT_SEM_PTR(API_SELECT_CB_VAR_REF(select_cb).sem), msectimeout);
- #if LWIP_NETCONN_SEM_PER_THREAD
- waited = 1;
- #endif
- }
- }
- /* Decrease select_waiting for each socket we are interested in */
- for (i = LWIP_SOCKET_OFFSET; i < maxfdp2; i++) {
- if ((readset && FD_ISSET(i, readset)) ||
- (writeset && FD_ISSET(i, writeset)) ||
- (exceptset && FD_ISSET(i, exceptset))) {
- struct lwip_sock *sock;
- SYS_ARCH_PROTECT(lev);
- sock = tryget_socket_unconn_locked(i);
- if (sock != NULL) {
- /* for now, handle select_waiting==0... */
- LWIP_ASSERT("sock->select_waiting > 0", sock->select_waiting > 0);
- if (sock->select_waiting > 0) {
- sock->select_waiting--;
- }
- SYS_ARCH_UNPROTECT(lev);
- done_socket(sock);
- } else {
- SYS_ARCH_UNPROTECT(lev);
- /* Not a valid socket */
- nready = -1;
- set_errno(EBADF);
- }
- }
- }
- lwip_unlink_select_cb(&API_SELECT_CB_VAR_REF(select_cb));
- #if LWIP_NETCONN_SEM_PER_THREAD
- if (API_SELECT_CB_VAR_REF(select_cb).sem_signalled && (!waited || (waitres == SYS_ARCH_TIMEOUT))) {
- /* don't leave the thread-local semaphore signalled */
- sys_arch_sem_wait(API_SELECT_CB_VAR_REF(select_cb).sem, 1);
- }
- #else /* LWIP_NETCONN_SEM_PER_THREAD */
- sys_sem_free(&API_SELECT_CB_VAR_REF(select_cb).sem);
- #endif /* LWIP_NETCONN_SEM_PER_THREAD */
- API_SELECT_CB_VAR_FREE(select_cb);
- if (nready < 0) {
- /* This happens when a socket got closed while waiting */
- lwip_select_dec_sockets_used(maxfdp1, &used_sockets);
- return -1;
- }
- if (waitres == SYS_ARCH_TIMEOUT) {
- /* Timeout */
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_select: timeout expired\n"));
- /* This is OK as the local fdsets are empty and nready is zero,
- or we would have returned earlier. */
- } else {
- /* See what's set now after waiting */
- nready = lwip_selscan(maxfdp1, readset, writeset, exceptset, &lreadset, &lwriteset, &lexceptset);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_select: nready=%d\n", nready));
- }
- }
- }
- lwip_select_dec_sockets_used(maxfdp1, &used_sockets);
- set_errno(0);
- if (readset) {
- *readset = lreadset;
- }
- if (writeset) {
- *writeset = lwriteset;
- }
- if (exceptset) {
- *exceptset = lexceptset;
- }
- return nready;
- }
- #endif /* LWIP_SOCKET_SELECT */
- #if LWIP_SOCKET_POLL
- /** Options for the lwip_pollscan function. */
- enum lwip_pollscan_opts
- {
- /** Clear revents in each struct pollfd. */
- LWIP_POLLSCAN_CLEAR = 1,
- /** Increment select_waiting in each struct lwip_sock. */
- LWIP_POLLSCAN_INC_WAIT = 2,
- /** Decrement select_waiting in each struct lwip_sock. */
- LWIP_POLLSCAN_DEC_WAIT = 4
- };
- /**
- * Update revents in each struct pollfd.
- * Optionally update select_waiting in struct lwip_sock.
- *
- * @param fds array of structures to update
- * @param nfds number of structures in fds
- * @param opts what to update and how
- * @return number of structures that have revents != 0
- */
- static int
- lwip_pollscan(struct pollfd *fds, nfds_t nfds, enum lwip_pollscan_opts opts)
- {
- int nready = 0;
- nfds_t fdi;
- struct lwip_sock *sock;
- SYS_ARCH_DECL_PROTECT(lev);
- /* Go through each struct pollfd in the array. */
- for (fdi = 0; fdi < nfds; fdi++) {
- if ((opts & LWIP_POLLSCAN_CLEAR) != 0) {
- fds[fdi].revents = 0;
- }
- /* Negative fd means the caller wants us to ignore this struct.
- POLLNVAL means we already detected that the fd is invalid;
- if another thread has since opened a new socket with that fd,
- we must not use that socket. */
- if (fds[fdi].fd >= 0 && (fds[fdi].revents & POLLNVAL) == 0) {
- /* First get the socket's status (protected)... */
- SYS_ARCH_PROTECT(lev);
- sock = tryget_socket_unconn_locked(fds[fdi].fd);
- if (sock != NULL) {
- void* lastdata = sock->lastdata.pbuf;
- s16_t rcvevent = sock->rcvevent;
- u16_t sendevent = sock->sendevent;
- u16_t errevent = sock->errevent;
- if ((opts & LWIP_POLLSCAN_INC_WAIT) != 0) {
- sock->select_waiting++;
- if (sock->select_waiting == 0) {
- /* overflow - too many threads waiting */
- sock->select_waiting--;
- nready = -1;
- SYS_ARCH_UNPROTECT(lev);
- done_socket(sock);
- break;
- }
- } else if ((opts & LWIP_POLLSCAN_DEC_WAIT) != 0) {
- /* for now, handle select_waiting==0... */
- LWIP_ASSERT("sock->select_waiting > 0", sock->select_waiting > 0);
- if (sock->select_waiting > 0) {
- sock->select_waiting--;
- }
- }
- SYS_ARCH_UNPROTECT(lev);
- done_socket(sock);
- /* ... then examine it: */
- /* See if netconn of this socket is ready for read */
- if ((fds[fdi].events & POLLIN) != 0 && ((lastdata != NULL) || (rcvevent > 0))) {
- fds[fdi].revents |= POLLIN;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_pollscan: fd=%d ready for reading\n", fds[fdi].fd));
- }
- /* See if netconn of this socket is ready for write */
- if ((fds[fdi].events & POLLOUT) != 0 && (sendevent != 0)) {
- fds[fdi].revents |= POLLOUT;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_pollscan: fd=%d ready for writing\n", fds[fdi].fd));
- }
- /* See if netconn of this socket had an error */
- if (errevent != 0) {
- /* POLLERR is output only. */
- fds[fdi].revents |= POLLERR;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_pollscan: fd=%d ready for exception\n", fds[fdi].fd));
- }
- } else {
- /* Not a valid socket */
- SYS_ARCH_UNPROTECT(lev);
- /* POLLNVAL is output only. */
- fds[fdi].revents |= POLLNVAL;
- return -1;
- }
- }
- /* Will return the number of structures that have events,
- not the number of events. */
- if (fds[fdi].revents != 0) {
- nready++;
- }
- }
- LWIP_ASSERT("nready >= 0", nready >= 0);
- return nready;
- }
- #if LWIP_NETCONN_FULLDUPLEX
- /* Mark all sockets as used.
- *
- * All sockets are marked (and later unmarked), whether they are open or not.
- * This is OK as lwip_pollscan aborts select when non-open sockets are found.
- */
- static void
- lwip_poll_inc_sockets_used(struct pollfd *fds, nfds_t nfds)
- {
- nfds_t fdi;
- if(fds) {
- /* Go through each struct pollfd in the array. */
- for (fdi = 0; fdi < nfds; fdi++) {
- /* Increase the reference counter */
- tryget_socket_unconn(fds[fdi].fd);
- }
- }
- }
- /* Let go all sockets that were marked as used when starting poll */
- static void
- lwip_poll_dec_sockets_used(struct pollfd *fds, nfds_t nfds)
- {
- nfds_t fdi;
- if(fds) {
- /* Go through each struct pollfd in the array. */
- for (fdi = 0; fdi < nfds; fdi++) {
- struct lwip_sock *sock = tryget_socket_unconn_nouse(fds[fdi].fd);
- if (sock != NULL) {
- done_socket(sock);
- }
- }
- }
- }
- #else /* LWIP_NETCONN_FULLDUPLEX */
- #define lwip_poll_inc_sockets_used(fds, nfds)
- #define lwip_poll_dec_sockets_used(fds, nfds)
- #endif /* LWIP_NETCONN_FULLDUPLEX */
- int
- lwip_poll(struct pollfd *fds, nfds_t nfds, int timeout)
- {
- u32_t waitres = 0;
- int nready;
- u32_t msectimeout;
- #if LWIP_NETCONN_SEM_PER_THREAD
- int waited = 0;
- #endif
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_poll(%p, %d, %d)\n",
- (void*)fds, (int)nfds, timeout));
- LWIP_ERROR("lwip_poll: invalid fds", ((fds != NULL && nfds > 0) || (fds == NULL && nfds == 0)),
- set_errno(EINVAL); return -1;);
- lwip_poll_inc_sockets_used(fds, nfds);
- /* Go through each struct pollfd to count number of structures
- which currently match */
- nready = lwip_pollscan(fds, nfds, LWIP_POLLSCAN_CLEAR);
- if (nready < 0) {
- lwip_poll_dec_sockets_used(fds, nfds);
- return -1;
- }
- /* If we don't have any current events, then suspend if we are supposed to */
- if (!nready) {
- API_SELECT_CB_VAR_DECLARE(select_cb);
- if (timeout == 0) {
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_poll: no timeout, returning 0\n"));
- goto return_success;
- }
- API_SELECT_CB_VAR_ALLOC(select_cb, set_errno(EAGAIN); lwip_poll_dec_sockets_used(fds, nfds); return -1);
- memset(&API_SELECT_CB_VAR_REF(select_cb), 0, sizeof(struct lwip_select_cb));
- /* None ready: add our semaphore to list:
- We don't actually need any dynamic memory. Our entry on the
- list is only valid while we are in this function, so it's ok
- to use local variables. */
- API_SELECT_CB_VAR_REF(select_cb).poll_fds = fds;
- API_SELECT_CB_VAR_REF(select_cb).poll_nfds = nfds;
- #if LWIP_NETCONN_SEM_PER_THREAD
- API_SELECT_CB_VAR_REF(select_cb).sem = LWIP_NETCONN_THREAD_SEM_GET();
- #else /* LWIP_NETCONN_SEM_PER_THREAD */
- if (sys_sem_new(&API_SELECT_CB_VAR_REF(select_cb).sem, 0) != ERR_OK) {
- /* failed to create semaphore */
- set_errno(EAGAIN);
- lwip_poll_dec_sockets_used(fds, nfds);
- API_SELECT_CB_VAR_FREE(select_cb);
- return -1;
- }
- #endif /* LWIP_NETCONN_SEM_PER_THREAD */
- lwip_link_select_cb(&API_SELECT_CB_VAR_REF(select_cb));
- /* Increase select_waiting for each socket we are interested in.
- Also, check for events again: there could have been events between
- the last scan (without us on the list) and putting us on the list! */
- nready = lwip_pollscan(fds, nfds, LWIP_POLLSCAN_INC_WAIT);
- if (!nready) {
- /* Still none ready, just wait to be woken */
- if (timeout < 0) {
- /* Wait forever */
- msectimeout = 0;
- } else {
- /* timeout == 0 would have been handled earlier. */
- LWIP_ASSERT("timeout > 0", timeout > 0);
- msectimeout = timeout;
- }
- waitres = sys_arch_sem_wait(SELECT_SEM_PTR(API_SELECT_CB_VAR_REF(select_cb).sem), msectimeout);
- #if LWIP_NETCONN_SEM_PER_THREAD
- waited = 1;
- #endif
- }
- /* Decrease select_waiting for each socket we are interested in,
- and check which events occurred while we waited. */
- nready = lwip_pollscan(fds, nfds, LWIP_POLLSCAN_DEC_WAIT);
- lwip_unlink_select_cb(&API_SELECT_CB_VAR_REF(select_cb));
- #if LWIP_NETCONN_SEM_PER_THREAD
- if (select_cb.sem_signalled && (!waited || (waitres == SYS_ARCH_TIMEOUT))) {
- /* don't leave the thread-local semaphore signalled */
- sys_arch_sem_wait(API_SELECT_CB_VAR_REF(select_cb).sem, 1);
- }
- #else /* LWIP_NETCONN_SEM_PER_THREAD */
- sys_sem_free(&API_SELECT_CB_VAR_REF(select_cb).sem);
- #endif /* LWIP_NETCONN_SEM_PER_THREAD */
- API_SELECT_CB_VAR_FREE(select_cb);
- if (nready < 0) {
- /* This happens when a socket got closed while waiting */
- lwip_poll_dec_sockets_used(fds, nfds);
- return -1;
- }
- if (waitres == SYS_ARCH_TIMEOUT) {
- /* Timeout */
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_poll: timeout expired\n"));
- goto return_success;
- }
- }
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_poll: nready=%d\n", nready));
- return_success:
- lwip_poll_dec_sockets_used(fds, nfds);
- set_errno(0);
- return nready;
- }
- /**
- * Check whether event_callback should wake up a thread waiting in
- * lwip_poll.
- */
- static int
- lwip_poll_should_wake(const struct lwip_select_cb *scb, int fd, int has_recvevent, int has_sendevent, int has_errevent)
- {
- nfds_t fdi;
- for (fdi = 0; fdi < scb->poll_nfds; fdi++) {
- const struct pollfd *pollfd = &scb->poll_fds[fdi];
- if (pollfd->fd == fd) {
- /* Do not update pollfd->revents right here;
- that would be a data race because lwip_pollscan
- accesses revents without protecting. */
- if (has_recvevent && (pollfd->events & POLLIN) != 0) {
- return 1;
- }
- if (has_sendevent && (pollfd->events & POLLOUT) != 0) {
- return 1;
- }
- if (has_errevent) {
- /* POLLERR is output only. */
- return 1;
- }
- }
- }
- return 0;
- }
- #endif /* LWIP_SOCKET_POLL */
- #if LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL
- /**
- * Callback registered in the netconn layer for each socket-netconn.
- * Processes recvevent (data available) and wakes up tasks waiting for select.
- *
- * @note for LWIP_TCPIP_CORE_LOCKING any caller of this function
- * must have the core lock held when signaling the following events
- * as they might cause select_list_cb to be checked:
- * NETCONN_EVT_RCVPLUS
- * NETCONN_EVT_SENDPLUS
- * NETCONN_EVT_ERROR
- * This requirement will be asserted in select_check_waiters()
- */
- static void
- event_callback(struct netconn *conn, enum netconn_evt evt, u16_t len)
- {
- int s, check_waiters;
- struct lwip_sock *sock;
- SYS_ARCH_DECL_PROTECT(lev);
- LWIP_UNUSED_ARG(len);
- /* Get socket */
- if (conn) {
- s = conn->socket;
- if (s < 0) {
- /* Data comes in right away after an accept, even though
- * the server task might not have created a new socket yet.
- * Just count down (or up) if that's the case and we
- * will use the data later. Note that only receive events
- * can happen before the new socket is set up. */
- SYS_ARCH_PROTECT(lev);
- if (conn->socket < 0) {
- if (evt == NETCONN_EVT_RCVPLUS) {
- /* conn->socket is -1 on initialization
- lwip_accept adjusts sock->recvevent if conn->socket < -1 */
- conn->socket--;
- }
- SYS_ARCH_UNPROTECT(lev);
- return;
- }
- s = conn->socket;
- SYS_ARCH_UNPROTECT(lev);
- }
- sock = get_socket(s);
- if (!sock) {
- return;
- }
- } else {
- return;
- }
- check_waiters = 1;
- SYS_ARCH_PROTECT(lev);
- /* Set event as required */
- switch (evt) {
- case NETCONN_EVT_RCVPLUS:
- sock->rcvevent++;
- if (sock->rcvevent > 1) {
- check_waiters = 0;
- }
- break;
- case NETCONN_EVT_RCVMINUS:
- sock->rcvevent--;
- check_waiters = 0;
- break;
- case NETCONN_EVT_SENDPLUS:
- if (sock->sendevent) {
- check_waiters = 0;
- }
- sock->sendevent = 1;
- break;
- case NETCONN_EVT_SENDMINUS:
- sock->sendevent = 0;
- check_waiters = 0;
- break;
- case NETCONN_EVT_ERROR:
- sock->errevent = 1;
- break;
- default:
- LWIP_ASSERT("unknown event", 0);
- break;
- }
- if (sock->select_waiting && check_waiters) {
- /* Save which events are active */
- int has_recvevent, has_sendevent, has_errevent;
- has_recvevent = sock->rcvevent > 0;
- has_sendevent = sock->sendevent != 0;
- has_errevent = sock->errevent != 0;
- SYS_ARCH_UNPROTECT(lev);
- /* Check any select calls waiting on this socket */
- select_check_waiters(s, has_recvevent, has_sendevent, has_errevent);
- } else {
- SYS_ARCH_UNPROTECT(lev);
- }
- done_socket(sock);
- }
- /**
- * Check if any select waiters are waiting on this socket and its events
- *
- * @note on synchronization of select_cb_list:
- * LWIP_TCPIP_CORE_LOCKING: the select_cb_list must only be accessed while holding
- * the core lock. We do a single pass through the list and signal any waiters.
- * Core lock should already be held when calling here!!!!
- * !LWIP_TCPIP_CORE_LOCKING: we use SYS_ARCH_PROTECT but unlock on each iteration
- * of the loop, thus creating a possibility where a thread could modify the
- * select_cb_list during our UNPROTECT/PROTECT. We use a generational counter to
- * detect this change and restart the list walk. The list is expected to be small
- */
- static void select_check_waiters(int s, int has_recvevent, int has_sendevent, int has_errevent)
- {
- struct lwip_select_cb *scb;
- #if !LWIP_TCPIP_CORE_LOCKING
- int last_select_cb_ctr;
- SYS_ARCH_DECL_PROTECT(lev);
- #endif /* !LWIP_TCPIP_CORE_LOCKING */
- LWIP_ASSERT_CORE_LOCKED();
- #if !LWIP_TCPIP_CORE_LOCKING
- SYS_ARCH_PROTECT(lev);
- again:
- /* remember the state of select_cb_list to detect changes */
- last_select_cb_ctr = select_cb_ctr;
- #endif /* !LWIP_TCPIP_CORE_LOCKING */
- for (scb = select_cb_list; scb != NULL; scb = scb->next) {
- if (scb->sem_signalled == 0) {
- /* semaphore not signalled yet */
- int do_signal = 0;
- #if LWIP_SOCKET_POLL
- if (scb->poll_fds != NULL) {
- do_signal = lwip_poll_should_wake(scb, s, has_recvevent, has_sendevent, has_errevent);
- }
- #endif /* LWIP_SOCKET_POLL */
- #if LWIP_SOCKET_SELECT && LWIP_SOCKET_POLL
- else
- #endif /* LWIP_SOCKET_SELECT && LWIP_SOCKET_POLL */
- #if LWIP_SOCKET_SELECT
- {
- /* Test this select call for our socket */
- if (has_recvevent) {
- if (scb->readset && FD_ISSET(s, scb->readset)) {
- do_signal = 1;
- }
- }
- if (has_sendevent) {
- if (!do_signal && scb->writeset && FD_ISSET(s, scb->writeset)) {
- do_signal = 1;
- }
- }
- if (has_errevent) {
- if (!do_signal && scb->exceptset && FD_ISSET(s, scb->exceptset)) {
- do_signal = 1;
- }
- }
- }
- #endif /* LWIP_SOCKET_SELECT */
- if (do_signal) {
- scb->sem_signalled = 1;
- /* For !LWIP_TCPIP_CORE_LOCKING, we don't call SYS_ARCH_UNPROTECT() before signaling
- the semaphore, as this might lead to the select thread taking itself off the list,
- invalidating the semaphore. */
- sys_sem_signal(SELECT_SEM_PTR(scb->sem));
- }
- }
- #if LWIP_TCPIP_CORE_LOCKING
- }
- #else
- /* unlock interrupts with each step */
- SYS_ARCH_UNPROTECT(lev);
- /* this makes sure interrupt protection time is short */
- SYS_ARCH_PROTECT(lev);
- if (last_select_cb_ctr != select_cb_ctr) {
- /* someone has changed select_cb_list, restart at the beginning */
- goto again;
- }
- /* remember the state of select_cb_list to detect changes */
- last_select_cb_ctr = select_cb_ctr;
- }
- SYS_ARCH_UNPROTECT(lev);
- #endif
- }
- #endif /* LWIP_SOCKET_SELECT || LWIP_SOCKET_POLL */
- /**
- * Close one end of a full-duplex connection.
- */
- int
- lwip_shutdown(int s, int how)
- {
- struct lwip_sock *sock;
- err_t err;
- u8_t shut_rx = 0, shut_tx = 0;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_shutdown(%d, how=%d)\n", s, how));
- sock = get_socket(s);
- if (!sock) {
- return -1;
- }
- if (sock->conn != NULL) {
- if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) != NETCONN_TCP) {
- sock_set_errno(sock, EOPNOTSUPP);
- done_socket(sock);
- return -1;
- }
- } else {
- sock_set_errno(sock, ENOTCONN);
- done_socket(sock);
- return -1;
- }
- if (how == SHUT_RD) {
- shut_rx = 1;
- } else if (how == SHUT_WR) {
- shut_tx = 1;
- } else if (how == SHUT_RDWR) {
- shut_rx = 1;
- shut_tx = 1;
- } else {
- sock_set_errno(sock, EINVAL);
- done_socket(sock);
- return -1;
- }
- err = netconn_shutdown(sock->conn, shut_rx, shut_tx);
- sock_set_errno(sock, err_to_errno(err));
- done_socket(sock);
- return (err == ERR_OK ? 0 : -1);
- }
- static int
- lwip_getaddrname(int s, struct sockaddr *name, socklen_t *namelen, u8_t local)
- {
- struct lwip_sock *sock;
- union sockaddr_aligned saddr;
- ip_addr_t naddr;
- u16_t port;
- err_t err;
- sock = get_socket(s);
- if (!sock) {
- return -1;
- }
- /* get the IP address and port */
- err = netconn_getaddr(sock->conn, &naddr, &port, local);
- if (err != ERR_OK) {
- sock_set_errno(sock, err_to_errno(err));
- done_socket(sock);
- return -1;
- }
- #if LWIP_IPV4 && LWIP_IPV6
- /* Dual-stack: Map IPv4 addresses to IPv4 mapped IPv6 */
- if (NETCONNTYPE_ISIPV6(netconn_type(sock->conn)) &&
- IP_IS_V4_VAL(naddr)) {
- ip4_2_ipv4_mapped_ipv6(ip_2_ip6(&naddr), ip_2_ip4(&naddr));
- IP_SET_TYPE_VAL(naddr, IPADDR_TYPE_V6);
- }
- #endif /* LWIP_IPV4 && LWIP_IPV6 */
- IPADDR_PORT_TO_SOCKADDR(&saddr, &naddr, port);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getaddrname(%d, addr=", s));
- ip_addr_debug_print_val(SOCKETS_DEBUG, naddr);
- LWIP_DEBUGF(SOCKETS_DEBUG, (" port=%"U16_F")\n", port));
- if (*namelen > saddr.sa.sa_len) {
- *namelen = saddr.sa.sa_len;
- }
- MEMCPY(name, &saddr, *namelen);
- sock_set_errno(sock, 0);
- done_socket(sock);
- return 0;
- }
- int
- lwip_getpeername(int s, struct sockaddr *name, socklen_t *namelen)
- {
- return lwip_getaddrname(s, name, namelen, 0);
- }
- int
- lwip_getsockname(int s, struct sockaddr *name, socklen_t *namelen)
- {
- return lwip_getaddrname(s, name, namelen, 1);
- }
- int
- lwip_getsockopt(int s, int level, int optname, void *optval, socklen_t *optlen)
- {
- int err;
- struct lwip_sock *sock = get_socket(s);
- #if !LWIP_TCPIP_CORE_LOCKING
- err_t cberr;
- LWIP_SETGETSOCKOPT_DATA_VAR_DECLARE(data);
- #endif /* !LWIP_TCPIP_CORE_LOCKING */
- if (!sock) {
- return -1;
- }
- if ((NULL == optval) || (NULL == optlen)) {
- sock_set_errno(sock, EFAULT);
- done_socket(sock);
- return -1;
- }
- #if LWIP_TCPIP_CORE_LOCKING
- /* core-locking can just call the -impl function */
- LOCK_TCPIP_CORE();
- err = lwip_getsockopt_impl(s, level, optname, optval, optlen);
- UNLOCK_TCPIP_CORE();
- #else /* LWIP_TCPIP_CORE_LOCKING */
- #if LWIP_MPU_COMPATIBLE
- /* MPU_COMPATIBLE copies the optval data, so check for max size here */
- if (*optlen > LWIP_SETGETSOCKOPT_MAXOPTLEN) {
- sock_set_errno(sock, ENOBUFS);
- done_socket(sock);
- return -1;
- }
- #endif /* LWIP_MPU_COMPATIBLE */
- LWIP_SETGETSOCKOPT_DATA_VAR_ALLOC(data, sock);
- LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).s = s;
- LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).level = level;
- LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optname = optname;
- LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optlen = *optlen;
- #if !LWIP_MPU_COMPATIBLE
- LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optval.p = optval;
- #endif /* !LWIP_MPU_COMPATIBLE */
- LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).err = 0;
- #if LWIP_NETCONN_SEM_PER_THREAD
- LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).completed_sem = LWIP_NETCONN_THREAD_SEM_GET();
- #else
- LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).completed_sem = &sock->conn->op_completed;
- #endif
- cberr = tcpip_callback(lwip_getsockopt_callback, &LWIP_SETGETSOCKOPT_DATA_VAR_REF(data));
- if (cberr != ERR_OK) {
- LWIP_SETGETSOCKOPT_DATA_VAR_FREE(data);
- sock_set_errno(sock, err_to_errno(cberr));
- done_socket(sock);
- return -1;
- }
- sys_arch_sem_wait((sys_sem_t *)(LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).completed_sem), 0);
- /* write back optlen and optval */
- *optlen = LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optlen;
- #if LWIP_MPU_COMPATIBLE
- MEMCPY(optval, LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optval,
- LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optlen);
- #endif /* LWIP_MPU_COMPATIBLE */
- /* maybe lwip_getsockopt_internal has changed err */
- err = LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).err;
- LWIP_SETGETSOCKOPT_DATA_VAR_FREE(data);
- #endif /* LWIP_TCPIP_CORE_LOCKING */
- sock_set_errno(sock, err);
- done_socket(sock);
- return err ? -1 : 0;
- }
- #if !LWIP_TCPIP_CORE_LOCKING
- /** lwip_getsockopt_callback: only used without CORE_LOCKING
- * to get into the tcpip_thread
- */
- static void
- lwip_getsockopt_callback(void *arg)
- {
- struct lwip_setgetsockopt_data *data;
- LWIP_ASSERT("arg != NULL", arg != NULL);
- data = (struct lwip_setgetsockopt_data *)arg;
- data->err = lwip_getsockopt_impl(data->s, data->level, data->optname,
- #if LWIP_MPU_COMPATIBLE
- data->optval,
- #else /* LWIP_MPU_COMPATIBLE */
- data->optval.p,
- #endif /* LWIP_MPU_COMPATIBLE */
- &data->optlen);
- sys_sem_signal((sys_sem_t *)(data->completed_sem));
- }
- #endif /* LWIP_TCPIP_CORE_LOCKING */
- static int
- lwip_sockopt_to_ipopt(int optname)
- {
- /* Map SO_* values to our internal SOF_* values
- * We should not rely on #defines in socket.h
- * being in sync with ip.h.
- */
- switch (optname) {
- case SO_BROADCAST:
- return SOF_BROADCAST;
- case SO_KEEPALIVE:
- return SOF_KEEPALIVE;
- case SO_REUSEADDR:
- return SOF_REUSEADDR;
- default:
- LWIP_ASSERT("Unknown socket option", 0);
- return 0;
- }
- }
- /** lwip_getsockopt_impl: the actual implementation of getsockopt:
- * same argument as lwip_getsockopt, either called directly or through callback
- */
- static int
- lwip_getsockopt_impl(int s, int level, int optname, void *optval, socklen_t *optlen)
- {
- int err = 0;
- struct lwip_sock *sock = tryget_socket(s);
- if (!sock) {
- return EBADF;
- }
- #ifdef LWIP_HOOK_SOCKETS_GETSOCKOPT
- if (LWIP_HOOK_SOCKETS_GETSOCKOPT(s, sock, level, optname, optval, optlen, &err)) {
- return err;
- }
- #endif
- switch (level) {
- /* Level: SOL_SOCKET */
- case SOL_SOCKET:
- switch (optname) {
- #if LWIP_TCP
- case SO_ACCEPTCONN:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, *optlen, int);
- if (NETCONNTYPE_GROUP(sock->conn->type) != NETCONN_TCP) {
- done_socket(sock);
- return ENOPROTOOPT;
- }
- if ((sock->conn->pcb.tcp != NULL) && (sock->conn->pcb.tcp->state == LISTEN)) {
- *(int *)optval = 1;
- } else {
- *(int *)optval = 0;
- }
- break;
- #endif /* LWIP_TCP */
- /* The option flags */
- case SO_BROADCAST:
- case SO_KEEPALIVE:
- #if SO_REUSE
- case SO_REUSEADDR:
- #endif /* SO_REUSE */
- if ((optname == SO_BROADCAST) &&
- (NETCONNTYPE_GROUP(sock->conn->type) != NETCONN_UDP)) {
- done_socket(sock);
- return ENOPROTOOPT;
- }
- optname = lwip_sockopt_to_ipopt(optname);
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, *optlen, int);
- *(int *)optval = ip_get_option(sock->conn->pcb.ip, optname);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, SOL_SOCKET, optname=0x%x, ..) = %s\n",
- s, optname, (*(int *)optval ? "on" : "off")));
- break;
- case SO_TYPE:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, *optlen, int);
- switch (NETCONNTYPE_GROUP(netconn_type(sock->conn))) {
- case NETCONN_RAW:
- *(int *)optval = SOCK_RAW;
- break;
- case NETCONN_TCP:
- *(int *)optval = SOCK_STREAM;
- break;
- case NETCONN_UDP:
- *(int *)optval = SOCK_DGRAM;
- break;
- default: /* unrecognized socket type */
- *(int *)optval = netconn_type(sock->conn);
- LWIP_DEBUGF(SOCKETS_DEBUG,
- ("lwip_getsockopt(%d, SOL_SOCKET, SO_TYPE): unrecognized socket type %d\n",
- s, *(int *)optval));
- } /* switch (netconn_type(sock->conn)) */
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, SOL_SOCKET, SO_TYPE) = %d\n",
- s, *(int *)optval));
- break;
- case SO_ERROR:
- LWIP_SOCKOPT_CHECK_OPTLEN(sock, *optlen, int);
- *(int *)optval = err_to_errno(netconn_err(sock->conn));
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, SOL_SOCKET, SO_ERROR) = %d\n",
- s, *(int *)optval));
- break;
- #if LWIP_SO_SNDTIMEO
- case SO_SNDTIMEO:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, *optlen, LWIP_SO_SNDRCVTIMEO_OPTTYPE);
- LWIP_SO_SNDRCVTIMEO_SET(optval, netconn_get_sendtimeout(sock->conn));
- break;
- #endif /* LWIP_SO_SNDTIMEO */
- #if LWIP_SO_RCVTIMEO
- case SO_RCVTIMEO:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, *optlen, LWIP_SO_SNDRCVTIMEO_OPTTYPE);
- LWIP_SO_SNDRCVTIMEO_SET(optval, netconn_get_recvtimeout(sock->conn));
- break;
- #endif /* LWIP_SO_RCVTIMEO */
- #if LWIP_SO_RCVBUF
- case SO_RCVBUF:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, *optlen, int);
- *(int *)optval = netconn_get_recvbufsize(sock->conn);
- break;
- #endif /* LWIP_SO_RCVBUF */
- #if LWIP_SO_LINGER
- case SO_LINGER: {
- s16_t conn_linger;
- struct linger *linger = (struct linger *)optval;
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, *optlen, struct linger);
- conn_linger = sock->conn->linger;
- if (conn_linger >= 0) {
- linger->l_onoff = 1;
- linger->l_linger = (int)conn_linger;
- } else {
- linger->l_onoff = 0;
- linger->l_linger = 0;
- }
- }
- break;
- #endif /* LWIP_SO_LINGER */
- #if LWIP_UDP
- case SO_NO_CHECK:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, *optlen, int, NETCONN_UDP);
- #if LWIP_UDPLITE
- if (udp_is_flag_set(sock->conn->pcb.udp, UDP_FLAGS_UDPLITE)) {
- /* this flag is only available for UDP, not for UDP lite */
- done_socket(sock);
- return EAFNOSUPPORT;
- }
- #endif /* LWIP_UDPLITE */
- *(int *)optval = udp_is_flag_set(sock->conn->pcb.udp, UDP_FLAGS_NOCHKSUM) ? 1 : 0;
- break;
- #endif /* LWIP_UDP*/
- default:
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, SOL_SOCKET, UNIMPL: optname=0x%x, ..)\n",
- s, optname));
- err = ENOPROTOOPT;
- break;
- } /* switch (optname) */
- break;
- /* Level: IPPROTO_IP */
- case IPPROTO_IP:
- switch (optname) {
- case IP_TTL:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, *optlen, int);
- *(int *)optval = sock->conn->pcb.ip->ttl;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_IP, IP_TTL) = %d\n",
- s, *(int *)optval));
- break;
- case IP_TOS:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, *optlen, int);
- *(int *)optval = sock->conn->pcb.ip->tos;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_IP, IP_TOS) = %d\n",
- s, *(int *)optval));
- break;
- #if LWIP_IPV4 && LWIP_MULTICAST_TX_OPTIONS && LWIP_UDP
- case IP_MULTICAST_TTL:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, *optlen, u8_t);
- if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) != NETCONN_UDP) {
- done_socket(sock);
- return ENOPROTOOPT;
- }
- *(u8_t *)optval = udp_get_multicast_ttl(sock->conn->pcb.udp);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_IP, IP_MULTICAST_TTL) = %d\n",
- s, *(int *)optval));
- break;
- case IP_MULTICAST_IF:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, *optlen, struct in_addr);
- if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) != NETCONN_UDP) {
- done_socket(sock);
- return ENOPROTOOPT;
- }
- inet_addr_from_ip4addr((struct in_addr *)optval, udp_get_multicast_netif_addr(sock->conn->pcb.udp));
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_IP, IP_MULTICAST_IF) = 0x%"X32_F"\n",
- s, *(u32_t *)optval));
- break;
- case IP_MULTICAST_LOOP:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, *optlen, u8_t);
- if ((sock->conn->pcb.udp->flags & UDP_FLAGS_MULTICAST_LOOP) != 0) {
- *(u8_t *)optval = 1;
- } else {
- *(u8_t *)optval = 0;
- }
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_IP, IP_MULTICAST_LOOP) = %d\n",
- s, *(int *)optval));
- break;
- #endif /* LWIP_IPV4 && LWIP_MULTICAST_TX_OPTIONS && LWIP_UDP */
- default:
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_IP, UNIMPL: optname=0x%x, ..)\n",
- s, optname));
- err = ENOPROTOOPT;
- break;
- } /* switch (optname) */
- break;
- #if LWIP_TCP
- /* Level: IPPROTO_TCP */
- case IPPROTO_TCP:
- /* Special case: all IPPROTO_TCP option take an int */
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, *optlen, int, NETCONN_TCP);
- if (sock->conn->pcb.tcp->state == LISTEN) {
- done_socket(sock);
- return EINVAL;
- }
- switch (optname) {
- case TCP_NODELAY:
- *(int *)optval = tcp_nagle_disabled(sock->conn->pcb.tcp);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_TCP, TCP_NODELAY) = %s\n",
- s, (*(int *)optval) ? "on" : "off") );
- break;
- case TCP_KEEPALIVE:
- *(int *)optval = (int)sock->conn->pcb.tcp->keep_idle;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_TCP, TCP_KEEPALIVE) = %d\n",
- s, *(int *)optval));
- break;
- #if LWIP_TCP_KEEPALIVE
- case TCP_KEEPIDLE:
- *(int *)optval = (int)(sock->conn->pcb.tcp->keep_idle / 1000);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_TCP, TCP_KEEPIDLE) = %d\n",
- s, *(int *)optval));
- break;
- case TCP_KEEPINTVL:
- *(int *)optval = (int)(sock->conn->pcb.tcp->keep_intvl / 1000);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_TCP, TCP_KEEPINTVL) = %d\n",
- s, *(int *)optval));
- break;
- case TCP_KEEPCNT:
- *(int *)optval = (int)sock->conn->pcb.tcp->keep_cnt;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_TCP, TCP_KEEPCNT) = %d\n",
- s, *(int *)optval));
- break;
- #endif /* LWIP_TCP_KEEPALIVE */
- default:
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_TCP, UNIMPL: optname=0x%x, ..)\n",
- s, optname));
- err = ENOPROTOOPT;
- break;
- } /* switch (optname) */
- break;
- #endif /* LWIP_TCP */
- #if LWIP_IPV6
- /* Level: IPPROTO_IPV6 */
- case IPPROTO_IPV6:
- switch (optname) {
- case IPV6_V6ONLY:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, *optlen, int);
- *(int *)optval = (netconn_get_ipv6only(sock->conn) ? 1 : 0);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_IPV6, IPV6_V6ONLY) = %d\n",
- s, *(int *)optval));
- break;
- default:
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_IPV6, UNIMPL: optname=0x%x, ..)\n",
- s, optname));
- err = ENOPROTOOPT;
- break;
- } /* switch (optname) */
- break;
- #endif /* LWIP_IPV6 */
- #if LWIP_UDP && LWIP_UDPLITE
- /* Level: IPPROTO_UDPLITE */
- case IPPROTO_UDPLITE:
- /* Special case: all IPPROTO_UDPLITE option take an int */
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, *optlen, int);
- /* If this is no UDP lite socket, ignore any options. */
- if (!NETCONNTYPE_ISUDPLITE(netconn_type(sock->conn))) {
- done_socket(sock);
- return ENOPROTOOPT;
- }
- switch (optname) {
- case UDPLITE_SEND_CSCOV:
- *(int *)optval = sock->conn->pcb.udp->chksum_len_tx;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_UDPLITE, UDPLITE_SEND_CSCOV) = %d\n",
- s, (*(int *)optval)) );
- break;
- case UDPLITE_RECV_CSCOV:
- *(int *)optval = sock->conn->pcb.udp->chksum_len_rx;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_UDPLITE, UDPLITE_RECV_CSCOV) = %d\n",
- s, (*(int *)optval)) );
- break;
- default:
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_UDPLITE, UNIMPL: optname=0x%x, ..)\n",
- s, optname));
- err = ENOPROTOOPT;
- break;
- } /* switch (optname) */
- break;
- #endif /* LWIP_UDP */
- /* Level: IPPROTO_RAW */
- case IPPROTO_RAW:
- switch (optname) {
- #if LWIP_IPV6 && LWIP_RAW
- case IPV6_CHECKSUM:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, *optlen, int, NETCONN_RAW);
- if (sock->conn->pcb.raw->chksum_reqd == 0) {
- *(int *)optval = -1;
- } else {
- *(int *)optval = sock->conn->pcb.raw->chksum_offset;
- }
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_RAW, IPV6_CHECKSUM) = %d\n",
- s, (*(int *)optval)) );
- break;
- #endif /* LWIP_IPV6 && LWIP_RAW */
- default:
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, IPPROTO_RAW, UNIMPL: optname=0x%x, ..)\n",
- s, optname));
- err = ENOPROTOOPT;
- break;
- } /* switch (optname) */
- break;
- default:
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_getsockopt(%d, level=0x%x, UNIMPL: optname=0x%x, ..)\n",
- s, level, optname));
- err = ENOPROTOOPT;
- break;
- } /* switch (level) */
- done_socket(sock);
- return err;
- }
- int
- lwip_setsockopt(int s, int level, int optname, const void *optval, socklen_t optlen)
- {
- int err = 0;
- struct lwip_sock *sock = get_socket(s);
- #if !LWIP_TCPIP_CORE_LOCKING
- err_t cberr;
- LWIP_SETGETSOCKOPT_DATA_VAR_DECLARE(data);
- #endif /* !LWIP_TCPIP_CORE_LOCKING */
- if (!sock) {
- return -1;
- }
- if (NULL == optval) {
- sock_set_errno(sock, EFAULT);
- done_socket(sock);
- return -1;
- }
- #if LWIP_TCPIP_CORE_LOCKING
- /* core-locking can just call the -impl function */
- LOCK_TCPIP_CORE();
- err = lwip_setsockopt_impl(s, level, optname, optval, optlen);
- UNLOCK_TCPIP_CORE();
- #else /* LWIP_TCPIP_CORE_LOCKING */
- #if LWIP_MPU_COMPATIBLE
- /* MPU_COMPATIBLE copies the optval data, so check for max size here */
- if (optlen > LWIP_SETGETSOCKOPT_MAXOPTLEN) {
- sock_set_errno(sock, ENOBUFS);
- done_socket(sock);
- return -1;
- }
- #endif /* LWIP_MPU_COMPATIBLE */
- LWIP_SETGETSOCKOPT_DATA_VAR_ALLOC(data, sock);
- LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).s = s;
- LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).level = level;
- LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optname = optname;
- LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optlen = optlen;
- #if LWIP_MPU_COMPATIBLE
- MEMCPY(LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optval, optval, optlen);
- #else /* LWIP_MPU_COMPATIBLE */
- LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).optval.pc = (const void *)optval;
- #endif /* LWIP_MPU_COMPATIBLE */
- LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).err = 0;
- #if LWIP_NETCONN_SEM_PER_THREAD
- LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).completed_sem = LWIP_NETCONN_THREAD_SEM_GET();
- #else
- LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).completed_sem = &sock->conn->op_completed;
- #endif
- cberr = tcpip_callback(lwip_setsockopt_callback, &LWIP_SETGETSOCKOPT_DATA_VAR_REF(data));
- if (cberr != ERR_OK) {
- LWIP_SETGETSOCKOPT_DATA_VAR_FREE(data);
- sock_set_errno(sock, err_to_errno(cberr));
- done_socket(sock);
- return -1;
- }
- sys_arch_sem_wait((sys_sem_t *)(LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).completed_sem), 0);
- /* maybe lwip_getsockopt_internal has changed err */
- err = LWIP_SETGETSOCKOPT_DATA_VAR_REF(data).err;
- LWIP_SETGETSOCKOPT_DATA_VAR_FREE(data);
- #endif /* LWIP_TCPIP_CORE_LOCKING */
- sock_set_errno(sock, err);
- done_socket(sock);
- return err ? -1 : 0;
- }
- #if !LWIP_TCPIP_CORE_LOCKING
- /** lwip_setsockopt_callback: only used without CORE_LOCKING
- * to get into the tcpip_thread
- */
- static void
- lwip_setsockopt_callback(void *arg)
- {
- struct lwip_setgetsockopt_data *data;
- LWIP_ASSERT("arg != NULL", arg != NULL);
- data = (struct lwip_setgetsockopt_data *)arg;
- data->err = lwip_setsockopt_impl(data->s, data->level, data->optname,
- #if LWIP_MPU_COMPATIBLE
- data->optval,
- #else /* LWIP_MPU_COMPATIBLE */
- data->optval.pc,
- #endif /* LWIP_MPU_COMPATIBLE */
- data->optlen);
- sys_sem_signal((sys_sem_t *)(data->completed_sem));
- }
- #endif /* LWIP_TCPIP_CORE_LOCKING */
- /** lwip_setsockopt_impl: the actual implementation of setsockopt:
- * same argument as lwip_setsockopt, either called directly or through callback
- */
- static int
- lwip_setsockopt_impl(int s, int level, int optname, const void *optval, socklen_t optlen)
- {
- int err = 0;
- struct lwip_sock *sock = tryget_socket(s);
- if (!sock) {
- return EBADF;
- }
- #ifdef LWIP_HOOK_SOCKETS_SETSOCKOPT
- if (LWIP_HOOK_SOCKETS_SETSOCKOPT(s, sock, level, optname, optval, optlen, &err)) {
- return err;
- }
- #endif
- switch (level) {
- /* Level: SOL_SOCKET */
- case SOL_SOCKET:
- switch (optname) {
- /* SO_ACCEPTCONN is get-only */
- /* The option flags */
- case SO_BROADCAST:
- case SO_KEEPALIVE:
- #if SO_REUSE
- case SO_REUSEADDR:
- #endif /* SO_REUSE */
- if ((optname == SO_BROADCAST) &&
- (NETCONNTYPE_GROUP(sock->conn->type) != NETCONN_UDP)) {
- done_socket(sock);
- return ENOPROTOOPT;
- }
- optname = lwip_sockopt_to_ipopt(optname);
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, optlen, int);
- if (*(const int *)optval) {
- ip_set_option(sock->conn->pcb.ip, optname);
- } else {
- ip_reset_option(sock->conn->pcb.ip, optname);
- }
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, SOL_SOCKET, optname=0x%x, ..) -> %s\n",
- s, optname, (*(const int *)optval ? "on" : "off")));
- break;
- /* SO_TYPE is get-only */
- /* SO_ERROR is get-only */
- #if LWIP_SO_SNDTIMEO
- case SO_SNDTIMEO: {
- long ms_long;
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, optlen, LWIP_SO_SNDRCVTIMEO_OPTTYPE);
- ms_long = LWIP_SO_SNDRCVTIMEO_GET_MS(optval);
- if (ms_long < 0) {
- done_socket(sock);
- return EINVAL;
- }
- netconn_set_sendtimeout(sock->conn, ms_long);
- break;
- }
- #endif /* LWIP_SO_SNDTIMEO */
- #if LWIP_SO_RCVTIMEO
- case SO_RCVTIMEO: {
- long ms_long;
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, optlen, LWIP_SO_SNDRCVTIMEO_OPTTYPE);
- ms_long = LWIP_SO_SNDRCVTIMEO_GET_MS(optval);
- if (ms_long < 0) {
- done_socket(sock);
- return EINVAL;
- }
- netconn_set_recvtimeout(sock->conn, (u32_t)ms_long);
- break;
- }
- #endif /* LWIP_SO_RCVTIMEO */
- #if LWIP_SO_RCVBUF
- case SO_RCVBUF:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, optlen, int);
- netconn_set_recvbufsize(sock->conn, *(const int *)optval);
- break;
- #endif /* LWIP_SO_RCVBUF */
- #if LWIP_SO_LINGER
- case SO_LINGER: {
- const struct linger *linger = (const struct linger *)optval;
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, optlen, struct linger);
- if (linger->l_onoff) {
- int lingersec = linger->l_linger;
- if (lingersec < 0) {
- done_socket(sock);
- return EINVAL;
- }
- if (lingersec > 0xFFFF) {
- lingersec = 0xFFFF;
- }
- sock->conn->linger = (s16_t)lingersec;
- } else {
- sock->conn->linger = -1;
- }
- }
- break;
- #endif /* LWIP_SO_LINGER */
- #if LWIP_UDP
- case SO_NO_CHECK:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, int, NETCONN_UDP);
- #if LWIP_UDPLITE
- if (udp_is_flag_set(sock->conn->pcb.udp, UDP_FLAGS_UDPLITE)) {
- /* this flag is only available for UDP, not for UDP lite */
- done_socket(sock);
- return EAFNOSUPPORT;
- }
- #endif /* LWIP_UDPLITE */
- if (*(const int *)optval) {
- udp_set_flags(sock->conn->pcb.udp, UDP_FLAGS_NOCHKSUM);
- } else {
- udp_clear_flags(sock->conn->pcb.udp, UDP_FLAGS_NOCHKSUM);
- }
- break;
- #endif /* LWIP_UDP */
- case SO_BINDTODEVICE: {
- const struct ifreq *iface;
- struct netif *n = NULL;
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN(sock, optlen, struct ifreq);
- iface = (const struct ifreq *)optval;
- if (iface->ifr_name[0] != 0) {
- n = netif_find(iface->ifr_name);
- if (n == NULL) {
- done_socket(sock);
- return ENODEV;
- }
- }
- switch (NETCONNTYPE_GROUP(netconn_type(sock->conn))) {
- #if LWIP_TCP
- case NETCONN_TCP:
- tcp_bind_netif(sock->conn->pcb.tcp, n);
- break;
- #endif
- #if LWIP_UDP
- case NETCONN_UDP:
- udp_bind_netif(sock->conn->pcb.udp, n);
- break;
- #endif
- #if LWIP_RAW
- case NETCONN_RAW:
- raw_bind_netif(sock->conn->pcb.raw, n);
- break;
- #endif
- default:
- LWIP_ASSERT("Unhandled netconn type in SO_BINDTODEVICE", 0);
- break;
- }
- }
- break;
- default:
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, SOL_SOCKET, UNIMPL: optname=0x%x, ..)\n",
- s, optname));
- err = ENOPROTOOPT;
- break;
- } /* switch (optname) */
- break;
- /* Level: IPPROTO_IP */
- case IPPROTO_IP:
- switch (optname) {
- case IP_TTL:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, optlen, int);
- sock->conn->pcb.ip->ttl = (u8_t)(*(const int *)optval);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_IP, IP_TTL, ..) -> %d\n",
- s, sock->conn->pcb.ip->ttl));
- break;
- case IP_TOS:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, optlen, int);
- sock->conn->pcb.ip->tos = (u8_t)(*(const int *)optval);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_IP, IP_TOS, ..)-> %d\n",
- s, sock->conn->pcb.ip->tos));
- break;
- #if LWIP_NETBUF_RECVINFO
- case IP_PKTINFO:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, int, NETCONN_UDP);
- if (*(const int *)optval) {
- sock->conn->flags |= NETCONN_FLAG_PKTINFO;
- } else {
- sock->conn->flags &= ~NETCONN_FLAG_PKTINFO;
- }
- break;
- #endif /* LWIP_NETBUF_RECVINFO */
- #if LWIP_IPV4 && LWIP_MULTICAST_TX_OPTIONS && LWIP_UDP
- case IP_MULTICAST_TTL:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, u8_t, NETCONN_UDP);
- udp_set_multicast_ttl(sock->conn->pcb.udp, (u8_t)(*(const u8_t *)optval));
- break;
- case IP_MULTICAST_IF: {
- ip4_addr_t if_addr;
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, struct in_addr, NETCONN_UDP);
- inet_addr_to_ip4addr(&if_addr, (const struct in_addr *)optval);
- udp_set_multicast_netif_addr(sock->conn->pcb.udp, &if_addr);
- }
- break;
- case IP_MULTICAST_LOOP:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, u8_t, NETCONN_UDP);
- if (*(const u8_t *)optval) {
- udp_set_flags(sock->conn->pcb.udp, UDP_FLAGS_MULTICAST_LOOP);
- } else {
- udp_clear_flags(sock->conn->pcb.udp, UDP_FLAGS_MULTICAST_LOOP);
- }
- break;
- #endif /* LWIP_IPV4 && LWIP_MULTICAST_TX_OPTIONS && LWIP_UDP */
- #if LWIP_IGMP
- case IP_ADD_MEMBERSHIP:
- case IP_DROP_MEMBERSHIP: {
- /* If this is a TCP or a RAW socket, ignore these options. */
- err_t igmp_err;
- const struct ip_mreq *imr = (const struct ip_mreq *)optval;
- ip4_addr_t if_addr;
- ip4_addr_t multi_addr;
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, struct ip_mreq, NETCONN_UDP);
- inet_addr_to_ip4addr(&if_addr, &imr->imr_interface);
- inet_addr_to_ip4addr(&multi_addr, &imr->imr_multiaddr);
- if (optname == IP_ADD_MEMBERSHIP) {
- if (!lwip_socket_register_membership(s, &if_addr, &multi_addr)) {
- /* cannot track membership (out of memory) */
- err = ENOMEM;
- igmp_err = ERR_OK;
- } else {
- igmp_err = igmp_joingroup(&if_addr, &multi_addr);
- }
- } else {
- igmp_err = igmp_leavegroup(&if_addr, &multi_addr);
- lwip_socket_unregister_membership(s, &if_addr, &multi_addr);
- }
- if (igmp_err != ERR_OK) {
- err = EADDRNOTAVAIL;
- }
- }
- break;
- #endif /* LWIP_IGMP */
- default:
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_IP, UNIMPL: optname=0x%x, ..)\n",
- s, optname));
- err = ENOPROTOOPT;
- break;
- } /* switch (optname) */
- break;
- #if LWIP_TCP
- /* Level: IPPROTO_TCP */
- case IPPROTO_TCP:
- /* Special case: all IPPROTO_TCP option take an int */
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, int, NETCONN_TCP);
- if (sock->conn->pcb.tcp->state == LISTEN) {
- done_socket(sock);
- return EINVAL;
- }
- switch (optname) {
- case TCP_NODELAY:
- if (*(const int *)optval) {
- tcp_nagle_disable(sock->conn->pcb.tcp);
- } else {
- tcp_nagle_enable(sock->conn->pcb.tcp);
- }
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_TCP, TCP_NODELAY) -> %s\n",
- s, (*(const int *)optval) ? "on" : "off") );
- break;
- case TCP_KEEPALIVE:
- sock->conn->pcb.tcp->keep_idle = (u32_t)(*(const int *)optval);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_TCP, TCP_KEEPALIVE) -> %"U32_F"\n",
- s, sock->conn->pcb.tcp->keep_idle));
- break;
- #if LWIP_TCP_KEEPALIVE
- case TCP_KEEPIDLE:
- sock->conn->pcb.tcp->keep_idle = 1000 * (u32_t)(*(const int *)optval);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_TCP, TCP_KEEPIDLE) -> %"U32_F"\n",
- s, sock->conn->pcb.tcp->keep_idle));
- break;
- case TCP_KEEPINTVL:
- sock->conn->pcb.tcp->keep_intvl = 1000 * (u32_t)(*(const int *)optval);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_TCP, TCP_KEEPINTVL) -> %"U32_F"\n",
- s, sock->conn->pcb.tcp->keep_intvl));
- break;
- case TCP_KEEPCNT:
- sock->conn->pcb.tcp->keep_cnt = (u32_t)(*(const int *)optval);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_TCP, TCP_KEEPCNT) -> %"U32_F"\n",
- s, sock->conn->pcb.tcp->keep_cnt));
- break;
- #endif /* LWIP_TCP_KEEPALIVE */
- default:
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_TCP, UNIMPL: optname=0x%x, ..)\n",
- s, optname));
- err = ENOPROTOOPT;
- break;
- } /* switch (optname) */
- break;
- #endif /* LWIP_TCP*/
- #if LWIP_IPV6
- /* Level: IPPROTO_IPV6 */
- case IPPROTO_IPV6:
- switch (optname) {
- case IPV6_V6ONLY:
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, optlen, int);
- if (*(const int *)optval) {
- netconn_set_ipv6only(sock->conn, 1);
- } else {
- netconn_set_ipv6only(sock->conn, 0);
- }
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_IPV6, IPV6_V6ONLY, ..) -> %d\n",
- s, (netconn_get_ipv6only(sock->conn) ? 1 : 0)));
- break;
- #if LWIP_IPV6_MLD
- case IPV6_JOIN_GROUP:
- case IPV6_LEAVE_GROUP: {
- /* If this is a TCP or a RAW socket, ignore these options. */
- err_t mld6_err;
- struct netif *netif;
- ip6_addr_t multi_addr;
- const struct ipv6_mreq *imr = (const struct ipv6_mreq *)optval;
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, struct ipv6_mreq, NETCONN_UDP);
- inet6_addr_to_ip6addr(&multi_addr, &imr->ipv6mr_multiaddr);
- LWIP_ASSERT("Invalid netif index", imr->ipv6mr_interface <= 0xFFu);
- netif = netif_get_by_index((u8_t)imr->ipv6mr_interface);
- if (netif == NULL) {
- err = EADDRNOTAVAIL;
- break;
- }
- if (optname == IPV6_JOIN_GROUP) {
- if (!lwip_socket_register_mld6_membership(s, imr->ipv6mr_interface, &multi_addr)) {
- /* cannot track membership (out of memory) */
- err = ENOMEM;
- mld6_err = ERR_OK;
- } else {
- mld6_err = mld6_joingroup_netif(netif, &multi_addr);
- }
- } else {
- mld6_err = mld6_leavegroup_netif(netif, &multi_addr);
- lwip_socket_unregister_mld6_membership(s, imr->ipv6mr_interface, &multi_addr);
- }
- if (mld6_err != ERR_OK) {
- err = EADDRNOTAVAIL;
- }
- }
- break;
- #endif /* LWIP_IPV6_MLD */
- default:
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_IPV6, UNIMPL: optname=0x%x, ..)\n",
- s, optname));
- err = ENOPROTOOPT;
- break;
- } /* switch (optname) */
- break;
- #endif /* LWIP_IPV6 */
- #if LWIP_UDP && LWIP_UDPLITE
- /* Level: IPPROTO_UDPLITE */
- case IPPROTO_UDPLITE:
- /* Special case: all IPPROTO_UDPLITE option take an int */
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB(sock, optlen, int);
- /* If this is no UDP lite socket, ignore any options. */
- if (!NETCONNTYPE_ISUDPLITE(netconn_type(sock->conn))) {
- done_socket(sock);
- return ENOPROTOOPT;
- }
- switch (optname) {
- case UDPLITE_SEND_CSCOV:
- if ((*(const int *)optval != 0) && ((*(const int *)optval < 8) || (*(const int *)optval > 0xffff))) {
- /* don't allow illegal values! */
- sock->conn->pcb.udp->chksum_len_tx = 8;
- } else {
- sock->conn->pcb.udp->chksum_len_tx = (u16_t) * (const int *)optval;
- }
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_UDPLITE, UDPLITE_SEND_CSCOV) -> %d\n",
- s, (*(const int *)optval)) );
- break;
- case UDPLITE_RECV_CSCOV:
- if ((*(const int *)optval != 0) && ((*(const int *)optval < 8) || (*(const int *)optval > 0xffff))) {
- /* don't allow illegal values! */
- sock->conn->pcb.udp->chksum_len_rx = 8;
- } else {
- sock->conn->pcb.udp->chksum_len_rx = (u16_t) * (const int *)optval;
- }
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_UDPLITE, UDPLITE_RECV_CSCOV) -> %d\n",
- s, (*(const int *)optval)) );
- break;
- default:
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_UDPLITE, UNIMPL: optname=0x%x, ..)\n",
- s, optname));
- err = ENOPROTOOPT;
- break;
- } /* switch (optname) */
- break;
- #endif /* LWIP_UDP */
- /* Level: IPPROTO_RAW */
- case IPPROTO_RAW:
- switch (optname) {
- #if LWIP_IPV6 && LWIP_RAW
- case IPV6_CHECKSUM:
- /* It should not be possible to disable the checksum generation with ICMPv6
- * as per RFC 3542 chapter 3.1 */
- if (sock->conn->pcb.raw->protocol == IPPROTO_ICMPV6) {
- done_socket(sock);
- return EINVAL;
- }
- LWIP_SOCKOPT_CHECK_OPTLEN_CONN_PCB_TYPE(sock, optlen, int, NETCONN_RAW);
- if (*(const int *)optval < 0) {
- sock->conn->pcb.raw->chksum_reqd = 0;
- } else if (*(const int *)optval & 1) {
- /* Per RFC3542, odd offsets are not allowed */
- done_socket(sock);
- return EINVAL;
- } else {
- sock->conn->pcb.raw->chksum_reqd = 1;
- sock->conn->pcb.raw->chksum_offset = (u16_t) * (const int *)optval;
- }
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_RAW, IPV6_CHECKSUM, ..) -> %d\n",
- s, sock->conn->pcb.raw->chksum_reqd));
- break;
- #endif /* LWIP_IPV6 && LWIP_RAW */
- default:
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, IPPROTO_RAW, UNIMPL: optname=0x%x, ..)\n",
- s, optname));
- err = ENOPROTOOPT;
- break;
- } /* switch (optname) */
- break;
- default:
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_setsockopt(%d, level=0x%x, UNIMPL: optname=0x%x, ..)\n",
- s, level, optname));
- err = ENOPROTOOPT;
- break;
- } /* switch (level) */
- done_socket(sock);
- return err;
- }
- int
- lwip_ioctl(int s, long cmd, void *argp)
- {
- struct lwip_sock *sock = get_socket(s);
- u8_t val;
- #if LWIP_SO_RCVBUF
- int recv_avail;
- #endif /* LWIP_SO_RCVBUF */
- if (!sock) {
- return -1;
- }
- switch (cmd) {
- #if LWIP_SO_RCVBUF || LWIP_FIONREAD_LINUXMODE
- case FIONREAD:
- if (!argp) {
- sock_set_errno(sock, EINVAL);
- done_socket(sock);
- return -1;
- }
- #if LWIP_FIONREAD_LINUXMODE
- if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) != NETCONN_TCP) {
- struct netbuf *nb;
- if (sock->lastdata.netbuf) {
- nb = sock->lastdata.netbuf;
- *((int *)argp) = nb->p->tot_len;
- } else {
- struct netbuf *rxbuf;
- err_t err = netconn_recv_udp_raw_netbuf_flags(sock->conn, &rxbuf, NETCONN_DONTBLOCK);
- if (err != ERR_OK) {
- *((int *)argp) = 0;
- } else {
- sock->lastdata.netbuf = rxbuf;
- *((int *)argp) = rxbuf->p->tot_len;
- }
- }
- done_socket(sock);
- return 0;
- }
- #endif /* LWIP_FIONREAD_LINUXMODE */
- #if LWIP_SO_RCVBUF
- /* we come here if either LWIP_FIONREAD_LINUXMODE==0 or this is a TCP socket */
- SYS_ARCH_GET(sock->conn->recv_avail, recv_avail);
- if (recv_avail < 0) {
- recv_avail = 0;
- }
- /* Check if there is data left from the last recv operation. /maq 041215 */
- if (sock->lastdata.netbuf) {
- if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_TCP) {
- recv_avail += sock->lastdata.pbuf->tot_len;
- } else {
- recv_avail += sock->lastdata.netbuf->p->tot_len;
- }
- }
- *((int *)argp) = recv_avail;
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_ioctl(%d, FIONREAD, %p) = %"U16_F"\n", s, argp, *((u16_t *)argp)));
- sock_set_errno(sock, 0);
- done_socket(sock);
- return 0;
- #else /* LWIP_SO_RCVBUF */
- break;
- #endif /* LWIP_SO_RCVBUF */
- #endif /* LWIP_SO_RCVBUF || LWIP_FIONREAD_LINUXMODE */
- case (long)FIONBIO:
- val = 0;
- if (argp && *(int *)argp) {
- val = 1;
- }
- netconn_set_nonblocking(sock->conn, val);
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_ioctl(%d, FIONBIO, %d)\n", s, val));
- sock_set_errno(sock, 0);
- done_socket(sock);
- return 0;
- default:
- break;
- } /* switch (cmd) */
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_ioctl(%d, UNIMPL: 0x%lx, %p)\n", s, cmd, argp));
- sock_set_errno(sock, ENOSYS); /* not yet implemented */
- done_socket(sock);
- return -1;
- }
- /** A minimal implementation of fcntl.
- * Currently only the commands F_GETFL and F_SETFL are implemented.
- * The flag O_NONBLOCK and access modes are supported for F_GETFL, only
- * the flag O_NONBLOCK is implemented for F_SETFL.
- */
- int
- lwip_fcntl(int s, int cmd, int val)
- {
- struct lwip_sock *sock = get_socket(s);
- int ret = -1;
- int op_mode = 0;
- if (!sock) {
- return -1;
- }
- switch (cmd) {
- case F_GETFL:
- ret = netconn_is_nonblocking(sock->conn) ? O_NONBLOCK : 0;
- sock_set_errno(sock, 0);
- if (NETCONNTYPE_GROUP(netconn_type(sock->conn)) == NETCONN_TCP) {
- #if LWIP_TCPIP_CORE_LOCKING
- LOCK_TCPIP_CORE();
- #else
- SYS_ARCH_DECL_PROTECT(lev);
- /* the proper thing to do here would be to get into the tcpip_thread,
- but locking should be OK as well since we only *read* some flags */
- SYS_ARCH_PROTECT(lev);
- #endif
- #if LWIP_TCP
- if (sock->conn->pcb.tcp) {
- if (!(sock->conn->pcb.tcp->flags & TF_RXCLOSED)) {
- op_mode |= O_RDONLY;
- }
- if (!(sock->conn->pcb.tcp->flags & TF_FIN)) {
- op_mode |= O_WRONLY;
- }
- }
- #endif
- #if LWIP_TCPIP_CORE_LOCKING
- UNLOCK_TCPIP_CORE();
- #else
- SYS_ARCH_UNPROTECT(lev);
- #endif
- } else {
- op_mode |= O_RDWR;
- }
- /* ensure O_RDWR for (O_RDONLY|O_WRONLY) != O_RDWR cases */
- ret |= (op_mode == (O_RDONLY | O_WRONLY)) ? O_RDWR : op_mode;
- break;
- case F_SETFL:
- /* Bits corresponding to the file access mode and the file creation flags [..] that are set in arg shall be ignored */
- val &= ~(O_RDONLY | O_WRONLY | O_RDWR);
- if ((val & ~O_NONBLOCK) == 0) {
- /* only O_NONBLOCK, all other bits are zero */
- netconn_set_nonblocking(sock->conn, val & O_NONBLOCK);
- ret = 0;
- sock_set_errno(sock, 0);
- } else {
- sock_set_errno(sock, ENOSYS); /* not yet implemented */
- }
- break;
- default:
- LWIP_DEBUGF(SOCKETS_DEBUG, ("lwip_fcntl(%d, UNIMPL: %d, %d)\n", s, cmd, val));
- sock_set_errno(sock, ENOSYS); /* not yet implemented */
- break;
- }
- done_socket(sock);
- return ret;
- }
- #if LWIP_COMPAT_SOCKETS == 2 && LWIP_POSIX_SOCKETS_IO_NAMES
- int
- fcntl(int s, int cmd, ...)
- {
- va_list ap;
- int val;
- va_start(ap, cmd);
- val = va_arg(ap, int);
- va_end(ap);
- return lwip_fcntl(s, cmd, val);
- }
- #endif
- const char *
- lwip_inet_ntop(int af, const void *src, char *dst, socklen_t size)
- {
- const char *ret = NULL;
- int size_int = (int)size;
- if (size_int < 0) {
- set_errno(ENOSPC);
- return NULL;
- }
- switch (af) {
- #if LWIP_IPV4
- case AF_INET:
- ret = ip4addr_ntoa_r((const ip4_addr_t *)src, dst, size_int);
- if (ret == NULL) {
- set_errno(ENOSPC);
- }
- break;
- #endif
- #if LWIP_IPV6
- case AF_INET6:
- ret = ip6addr_ntoa_r((const ip6_addr_t *)src, dst, size_int);
- if (ret == NULL) {
- set_errno(ENOSPC);
- }
- break;
- #endif
- default:
- set_errno(EAFNOSUPPORT);
- break;
- }
- return ret;
- }
- int
- lwip_inet_pton(int af, const char *src, void *dst)
- {
- int err;
- switch (af) {
- #if LWIP_IPV4
- case AF_INET:
- err = ip4addr_aton(src, (ip4_addr_t *)dst);
- break;
- #endif
- #if LWIP_IPV6
- case AF_INET6: {
- /* convert into temporary variable since ip6_addr_t might be larger
- than in6_addr when scopes are enabled */
- ip6_addr_t addr;
- err = ip6addr_aton(src, &addr);
- if (err) {
- memcpy(dst, &addr.addr, sizeof(addr.addr));
- }
- break;
- }
- #endif
- default:
- err = -1;
- set_errno(EAFNOSUPPORT);
- break;
- }
- return err;
- }
- #if LWIP_IGMP
- /** Register a new IGMP membership. On socket close, the membership is dropped automatically.
- *
- * ATTENTION: this function is called from tcpip_thread (or under CORE_LOCK).
- *
- * @return 1 on success, 0 on failure
- */
- static int
- lwip_socket_register_membership(int s, const ip4_addr_t *if_addr, const ip4_addr_t *multi_addr)
- {
- struct lwip_sock *sock = get_socket(s);
- int i;
- if (!sock) {
- return 0;
- }
- for (i = 0; i < LWIP_SOCKET_MAX_MEMBERSHIPS; i++) {
- if (socket_ipv4_multicast_memberships[i].sock == NULL) {
- socket_ipv4_multicast_memberships[i].sock = sock;
- ip4_addr_copy(socket_ipv4_multicast_memberships[i].if_addr, *if_addr);
- ip4_addr_copy(socket_ipv4_multicast_memberships[i].multi_addr, *multi_addr);
- done_socket(sock);
- return 1;
- }
- }
- done_socket(sock);
- return 0;
- }
- /** Unregister a previously registered membership. This prevents dropping the membership
- * on socket close.
- *
- * ATTENTION: this function is called from tcpip_thread (or under CORE_LOCK).
- */
- static void
- lwip_socket_unregister_membership(int s, const ip4_addr_t *if_addr, const ip4_addr_t *multi_addr)
- {
- struct lwip_sock *sock = get_socket(s);
- int i;
- if (!sock) {
- return;
- }
- for (i = 0; i < LWIP_SOCKET_MAX_MEMBERSHIPS; i++) {
- if ((socket_ipv4_multicast_memberships[i].sock == sock) &&
- ip4_addr_cmp(&socket_ipv4_multicast_memberships[i].if_addr, if_addr) &&
- ip4_addr_cmp(&socket_ipv4_multicast_memberships[i].multi_addr, multi_addr)) {
- socket_ipv4_multicast_memberships[i].sock = NULL;
- ip4_addr_set_zero(&socket_ipv4_multicast_memberships[i].if_addr);
- ip4_addr_set_zero(&socket_ipv4_multicast_memberships[i].multi_addr);
- break;
- }
- }
- done_socket(sock);
- }
- /** Drop all memberships of a socket that were not dropped explicitly via setsockopt.
- *
- * ATTENTION: this function is NOT called from tcpip_thread (or under CORE_LOCK).
- */
- static void
- lwip_socket_drop_registered_memberships(int s)
- {
- struct lwip_sock *sock = get_socket(s);
- int i;
- if (!sock) {
- return;
- }
- for (i = 0; i < LWIP_SOCKET_MAX_MEMBERSHIPS; i++) {
- if (socket_ipv4_multicast_memberships[i].sock == sock) {
- ip_addr_t multi_addr, if_addr;
- ip_addr_copy_from_ip4(multi_addr, socket_ipv4_multicast_memberships[i].multi_addr);
- ip_addr_copy_from_ip4(if_addr, socket_ipv4_multicast_memberships[i].if_addr);
- socket_ipv4_multicast_memberships[i].sock = NULL;
- ip4_addr_set_zero(&socket_ipv4_multicast_memberships[i].if_addr);
- ip4_addr_set_zero(&socket_ipv4_multicast_memberships[i].multi_addr);
- netconn_join_leave_group(sock->conn, &multi_addr, &if_addr, NETCONN_LEAVE);
- }
- }
- done_socket(sock);
- }
- #endif /* LWIP_IGMP */
- #if LWIP_IPV6_MLD
- /** Register a new MLD6 membership. On socket close, the membership is dropped automatically.
- *
- * ATTENTION: this function is called from tcpip_thread (or under CORE_LOCK).
- *
- * @return 1 on success, 0 on failure
- */
- static int
- lwip_socket_register_mld6_membership(int s, unsigned int if_idx, const ip6_addr_t *multi_addr)
- {
- struct lwip_sock *sock = get_socket(s);
- int i;
- if (!sock) {
- return 0;
- }
- for (i = 0; i < LWIP_SOCKET_MAX_MEMBERSHIPS; i++) {
- if (socket_ipv6_multicast_memberships[i].sock == NULL) {
- socket_ipv6_multicast_memberships[i].sock = sock;
- socket_ipv6_multicast_memberships[i].if_idx = (u8_t)if_idx;
- ip6_addr_copy(socket_ipv6_multicast_memberships[i].multi_addr, *multi_addr);
- done_socket(sock);
- return 1;
- }
- }
- done_socket(sock);
- return 0;
- }
- /** Unregister a previously registered MLD6 membership. This prevents dropping the membership
- * on socket close.
- *
- * ATTENTION: this function is called from tcpip_thread (or under CORE_LOCK).
- */
- static void
- lwip_socket_unregister_mld6_membership(int s, unsigned int if_idx, const ip6_addr_t *multi_addr)
- {
- struct lwip_sock *sock = get_socket(s);
- int i;
- if (!sock) {
- return;
- }
- for (i = 0; i < LWIP_SOCKET_MAX_MEMBERSHIPS; i++) {
- if ((socket_ipv6_multicast_memberships[i].sock == sock) &&
- (socket_ipv6_multicast_memberships[i].if_idx == if_idx) &&
- ip6_addr_cmp(&socket_ipv6_multicast_memberships[i].multi_addr, multi_addr)) {
- socket_ipv6_multicast_memberships[i].sock = NULL;
- socket_ipv6_multicast_memberships[i].if_idx = NETIF_NO_INDEX;
- ip6_addr_set_zero(&socket_ipv6_multicast_memberships[i].multi_addr);
- break;
- }
- }
- done_socket(sock);
- }
- /** Drop all MLD6 memberships of a socket that were not dropped explicitly via setsockopt.
- *
- * ATTENTION: this function is NOT called from tcpip_thread (or under CORE_LOCK).
- */
- static void
- lwip_socket_drop_registered_mld6_memberships(int s)
- {
- struct lwip_sock *sock = get_socket(s);
- int i;
- if (!sock) {
- return;
- }
- for (i = 0; i < LWIP_SOCKET_MAX_MEMBERSHIPS; i++) {
- if (socket_ipv6_multicast_memberships[i].sock == sock) {
- ip_addr_t multi_addr;
- u8_t if_idx;
- ip_addr_copy_from_ip6(multi_addr, socket_ipv6_multicast_memberships[i].multi_addr);
- if_idx = socket_ipv6_multicast_memberships[i].if_idx;
- socket_ipv6_multicast_memberships[i].sock = NULL;
- socket_ipv6_multicast_memberships[i].if_idx = NETIF_NO_INDEX;
- ip6_addr_set_zero(&socket_ipv6_multicast_memberships[i].multi_addr);
- netconn_join_leave_group_netif(sock->conn, &multi_addr, if_idx, NETCONN_LEAVE);
- }
- }
- done_socket(sock);
- }
- #endif /* LWIP_IPV6_MLD */
- #endif /* LWIP_SOCKET */
|