| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425442644274428442944304431443244334434443544364437443844394440444144424443444444454446444744484449445044514452445344544455445644574458445944604461446244634464446544664467446844694470447144724473447444754476447744784479448044814482448344844485448644874488448944904491449244934494449544964497449844994500450145024503450445054506450745084509451045114512451345144515451645174518 |
- #include "luat_base.h"
- #include "luat_malloc.h"
- #include "luat_crypto.h"
- #include "luat_rtos.h"
- #ifdef LUAT_USE_NETWORK
- #include "luat_rtos.h"
- #include "platform_def.h"
- #include "ctype.h"
- #include "luat_network_adapter.h"
- #define LUAT_LOG_TAG "adapter"
- #include "luat_log.h"
- #ifndef LWIP_NUM_SOCKETS
- #define LWIP_NUM_SOCKETS 8
- #endif
- typedef struct
- {
- #ifdef LUAT_USE_LWIP
- network_ctrl_t lwip_ctrl_table[LWIP_NUM_SOCKETS];
- // HANDLE network_mutex;
- #endif
- int last_adapter_index;
- int default_adapter_index;
- llist_head dns_cache_head;
- #ifdef LUAT_USE_LWIP
- uint8_t lwip_ctrl_busy[LWIP_NUM_SOCKETS];
- #endif
- uint8_t is_init;
- }network_info_t;
- typedef struct
- {
- network_adapter_info *opt;
- void *user_data;
- uint8_t *ctrl_busy;
- network_ctrl_t *ctrl_table;
- uint16_t port;
- }network_adapter_t;
- static network_adapter_t prv_adapter_table[NW_ADAPTER_QTY];
- static network_info_t prv_network = {
- .last_adapter_index = -1,
- .default_adapter_index = -1,
- .is_init = 0,
- };
- static const char *prv_network_event_id_string[] =
- {
- "适配器复位",
- "LINK状态变更",
- "超时",
- "DNS结果",
- "发送成功",
- "有新的数据",
- "接收缓存满了",
- "断开成功",
- "对端关闭",
- "连接成功",
- "连接异常",
- "开始监听",
- "新的客户端来了",
- "唤醒",
- "未知",
- };
- static const char *prv_network_ctrl_state_string[] =
- {
- "硬件离线",
- "离线",
- "等待DNS",
- "正在连接",
- "正在TLS握手",
- "在线",
- "在监听",
- "正在离线",
- "未知"
- };
- static const char *prv_network_ctrl_wait_state_string[] =
- {
- "无等待",
- "等待硬件上线",
- "等待连接完成",
- "等待发送完成",
- "等待离线完成",
- "等待任意网络变化",
- "未知",
- };
- static const char *prv_network_ctrl_callback_event_string[] =
- {
- "硬件状态回调",
- "连接状态回调",
- "离线状态回调",
- "发送状态回调",
- "任意网络变化回调",
- };
- const char *network_ctrl_event_id_string(uint32_t event)
- {
- if (event > EV_NW_END || event < EV_NW_RESET)
- {
- return prv_network_event_id_string[EV_NW_END - EV_NW_RESET];
- }
- return prv_network_event_id_string[event - EV_NW_RESET];
- }
- const char *network_ctrl_state_string(uint8_t state)
- {
- if (state > NW_STATE_DISCONNECTING)
- {
- return prv_network_ctrl_state_string[NW_STATE_DISCONNECTING + 1];
- }
- return prv_network_ctrl_state_string[state];
- }
- const char *network_ctrl_wait_state_string(uint8_t state)
- {
- if (state > NW_WAIT_EVENT)
- {
- return prv_network_ctrl_wait_state_string[NW_WAIT_EVENT + 1];
- }
- return prv_network_ctrl_wait_state_string[state];
- }
- const char *network_ctrl_callback_event_string(uint32_t event)
- {
- if (event > EV_NW_RESULT_EVENT || event < EV_NW_RESULT_LINK)
- {
- return prv_network_ctrl_callback_event_string[event - EV_NW_RESULT_EVENT + 1];
- }
- return prv_network_ctrl_callback_event_string[event - EV_NW_RESULT_EVENT];
- }
- network_adapter_info* network_adapter_fetch(int id, void** userdata) {
- if (id >= 0 && id < NW_ADAPTER_QTY) {
- if (prv_adapter_table[id].opt) {
- *userdata = prv_adapter_table[id].user_data;
- return prv_adapter_table[id].opt;
- }
- }
- return NULL;
- }
- #ifdef LUAT_USE_LWIP
- #include "../lwip/port/net_lwip.h"
- extern void DBG_Printf(const char* format, ...);
- extern void DBG_HexPrintf(void *Data, unsigned int len);
- //#define DBG(x,y...) DBG_Printf("%s %d:"x"\r\n", __FUNCTION__,__LINE__,##y)
- //#define DBG_ERR(x,y...) DBG_Printf("%s %d:"x"\r\n", __FUNCTION__,__LINE__,##y)
- static int tls_random( void *p_rng,
- unsigned char *output, size_t output_len );
- #define __NW_DEBUG_ENABLE__
- #ifdef __NW_DEBUG_ENABLE__
- #ifdef LUAT_LOG_NO_NEWLINE
- #define DBG(x,y...) do {if (ctrl->is_debug) {DBG_Printf("%s %d:"x, __FUNCTION__,__LINE__,##y);}} while(0)
- #define DBG_ERR(x,y...) DBG_Printf("%s %d:"x, __FUNCTION__,__LINE__,##y)
- #else
- #define DBG(x,y...) do {if (ctrl->is_debug) {DBG_Printf("%s %d:"x"\r\n", __FUNCTION__,__LINE__,##y);}} while(0)
- #define DBG_ERR(x,y...) DBG_Printf("%s %d:"x"\r\n", __FUNCTION__,__LINE__,##y)
- #endif
- #else
- #define DBG(x,y...)
- #define DBG_ERR(x,y...)
- #endif
- #define NW_LOCK platform_lock_mutex(ctrl->mutex)
- #define NW_UNLOCK platform_unlock_mutex(ctrl->mutex)
- #define SOL_SOCKET 0xfff /* options for socket level */
- #define SO_REUSEADDR 0x0004 /* Allow local address reuse */
- #define SO_KEEPALIVE 0x0008 /* keep connections alive */
- #define IPPROTO_ICMP 1
- #define IPPROTO_TCP 6
- #define TCP_NODELAY 0x01 /* don't delay send to coalesce packets */
- #define TCP_KEEPALIVE 0x02 /* send KEEPALIVE probes when idle for pcb->keep_idle milliseconds */
- #define TCP_KEEPIDLE 0x03 /* set pcb->keep_idle - Same as TCP_KEEPALIVE, but use seconds for get/setsockopt */
- #define TCP_KEEPINTVL 0x04 /* set pcb->keep_intvl - Use seconds for get/setsockopt */
- #define TCP_KEEPCNT 0x05 /* set pcb->keep_cnt - Use number of probes sent for get/setsockopt */
- static uint8_t network_check_ip_same(luat_ip_addr_t *ip1, luat_ip_addr_t *ip2)
- {
- #if defined ENABLE_PSIF
- return ip_addr_cmp(ip1, ip2);
- #else
- return ip_addr_cmp_zoneless(ip1, ip2);
- #endif
- }
- static int network_base_tx(network_ctrl_t *ctrl, const uint8_t *data, uint32_t len, int flags, luat_ip_addr_t *remote_ip, uint16_t remote_port)
- {
- int result = -1;
- if (ctrl->is_tcp)
- {
- result = network_socket_send(ctrl, data, len, flags, NULL, 0);
- }
- else
- {
- if (remote_ip)
- {
- result = network_socket_send(ctrl, data, len, flags, remote_ip, remote_port);
- }
- else
- {
- result = network_socket_send(ctrl, data, len, flags, ctrl->online_ip, ctrl->remote_port);
- }
- }
- if (result >= 0)
- {
- ctrl->tx_size += len;
- }
- else
- {
- ctrl->need_close = 1;
- }
- return result;
- }
- static LUAT_RT_RET_TYPE tls_shorttimeout(LUAT_RT_CB_PARAM)
- {
- network_ctrl_t *ctrl = (network_ctrl_t *)param;
- if (!ctrl->tls_mode)
- {
- platform_stop_timer(ctrl->tls_long_timer);
- return LUAT_RT_RET;
- }
- if (0 == ctrl->tls_timer_state)
- {
- ctrl->tls_timer_state = 1;
- }
- return LUAT_RT_RET;
- }
- #ifdef LUAT_USE_TLS
- static LUAT_RT_RET_TYPE tls_longtimeout(LUAT_RT_CB_PARAM)
- {
- network_ctrl_t *ctrl = (network_ctrl_t *)param;
- platform_stop_timer(ctrl->tls_short_timer);
- if (!ctrl->tls_mode)
- {
- return LUAT_RT_RET;
- }
- ctrl->tls_timer_state = 2;
- return LUAT_RT_RET;
- }
- static void tls_settimer( void *data, uint32_t int_ms, uint32_t fin_ms )
- {
- network_ctrl_t *ctrl = (network_ctrl_t *)data;
- if (!ctrl->tls_mode)
- {
- return;
- }
- if (!fin_ms)
- {
- platform_stop_timer(ctrl->tls_short_timer);
- platform_stop_timer(ctrl->tls_long_timer);
- ctrl->tls_timer_state = -1;
- return ;
- }
- platform_start_timer(ctrl->tls_short_timer, int_ms, 0);
- platform_start_timer(ctrl->tls_long_timer, fin_ms, 0);
- ctrl->tls_timer_state = 0;
- }
- static int tls_gettimer( void *data )
- {
- network_ctrl_t *ctrl = (network_ctrl_t *)data;
- if (!ctrl->tls_mode)
- {
- return -ERROR_PARAM_INVALID;
- }
- #if MBEDTLS_VERSION_NUMBER >= 0x03000000
- if(!mbedtls_ssl_is_handshake_over(ctrl->ssl))
- #else
- if (ctrl->ssl->state != MBEDTLS_SSL_HANDSHAKE_OVER)
- #endif
- {
- return ctrl->tls_timer_state;
- }
- else
- {
- return 0;
- }
- }
- static void tls_dbg(void *data, int level,
- const char *file, int line,
- const char *str)
- {
- DBG_Printf("%s %d:%s", file, line, str);
- }
- static int tls_send(void *ctx, const unsigned char *buf, size_t len )
- {
- network_ctrl_t *ctrl = (network_ctrl_t *)ctx;
- if (!ctrl->tls_mode)
- {
- return -ERROR_PERMISSION_DENIED;
- }
- if (network_base_tx(ctrl, buf, len, 0, NULL, 0) != len)
- {
- return -0x004E;
- }
- else
- {
- return len;
- }
- }
- #endif
- static int tls_recv(void *ctx, unsigned char *buf, size_t len )
- {
- #ifdef LUAT_USE_TLS
- network_ctrl_t *ctrl = (network_ctrl_t *)ctx;
- luat_ip_addr_t remote_ip;
- uint16_t remote_port;
- int result = -1;
- if (!ctrl->tls_mode)
- {
- return -1;
- }
- TLS_RECV:
- result = network_socket_receive(ctrl, buf, len, 0, &remote_ip, &remote_port);
- if (result < 0)
- {
- return -0x004C;
- }
- if (result > 0)
- {
- if (!ctrl->is_tcp)
- {
- if ((remote_port == ctrl->remote_port) && network_check_ip_same(&remote_ip, ctrl->online_ip))
- {
- goto TLS_RECV;
- }
- }
- return result;
- }
- return MBEDTLS_ERR_SSL_WANT_READ;
- #else
- return -1;
- #endif
- }
- static int network_get_host_by_name(network_ctrl_t *ctrl)
- {
- ctrl->remote_ip.type = 0xff;
- if (ipaddr_aton(ctrl->domain_name, &ctrl->remote_ip))
- {
- return 0;
- }
- ctrl->remote_ip.type = 0xff;
- return -1;
- }
- static void network_update_dns_cache(network_ctrl_t *ctrl)
- {
- }
- static void network_get_dns_cache(network_ctrl_t *ctrl)
- {
- }
- static int network_base_connect(network_ctrl_t *ctrl, luat_ip_addr_t *remote_ip)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- if (ctrl->socket_id >= 0)
- {
- network_force_close_socket(ctrl);
- }
- if (remote_ip)
- {
- if (network_create_soceket(ctrl, IPADDR_TYPE_V6 == remote_ip->type) < 0)
- {
- network_clean_invaild_socket(ctrl->adapter_index);
- if (network_create_soceket(ctrl, IPADDR_TYPE_V6 == remote_ip->type) < 0)
- {
- return -1;
- }
- }
- if (adapter->opt->is_posix)
- {
- volatile uint32_t val;
- val = ctrl->tcp_keep_alive;
- network_setsockopt(ctrl, SOL_SOCKET, SO_KEEPALIVE, (void *)&val, sizeof(val));
- if (ctrl->tcp_keep_alive)
- {
- val = ctrl->tcp_keep_idle;
- network_setsockopt(ctrl, IPPROTO_TCP, TCP_KEEPIDLE, (void*)&val, sizeof(val));
- val = ctrl->tcp_keep_interval;
- network_setsockopt(ctrl, IPPROTO_TCP, TCP_KEEPINTVL, (void *)&val, sizeof(val));
- val = ctrl->tcp_keep_cnt;
- network_setsockopt(ctrl, IPPROTO_TCP, TCP_KEEPCNT, (void *)&val, sizeof(val));
- }
- }
- else
- {
- network_user_cmd(ctrl, NW_CMD_AUTO_HEART_TIME, ctrl->tcp_keep_idle);
- }
- return network_socket_connect(ctrl, remote_ip);
- }
- else
- {
- if (network_create_soceket(ctrl, 0) < 0)
- {
- network_clean_invaild_socket(ctrl->adapter_index);
- if (network_create_soceket(ctrl, 0) < 0)
- {
- return -1;
- }
- }
- return network_socket_listen(ctrl);
- }
- }
- static int network_prepare_connect(network_ctrl_t *ctrl)
- {
- if (ctrl->remote_ip.type != 0xff)
- {
- ;
- }
- else if (ctrl->domain_name)
- {
- if (network_get_host_by_name(ctrl))
- {
- if (network_dns(ctrl))
- {
- network_socket_force_close(ctrl);
- return -1;
- }
- ctrl->state = NW_STATE_WAIT_DNS;
- return 0;
- }
- }
- else
- {
- return -1;
- }
- if (network_base_connect(ctrl, &ctrl->remote_ip))
- {
- network_socket_force_close(ctrl);
- return -1;
- }
- ctrl->state = NW_STATE_CONNECTING;
- return 0;
- }
- static int network_state_link_off(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- if (EV_NW_STATE == event->ID)
- {
- if (event->Param2)
- {
- ctrl->state = NW_STATE_OFF_LINE;
- if (NW_WAIT_LINK_UP == ctrl->wait_target_state)
- {
- return 0;
- }
- else if (NW_WAIT_ON_LINE == ctrl->wait_target_state)
- {
- if (ctrl->is_server_mode)
- {
- if (network_base_connect(ctrl, NULL))
- {
- return -1;
- }
- ctrl->state = NW_STATE_CONNECTING;
- }
- else
- {
- if (network_prepare_connect(ctrl))
- {
- return -1;
- }
- }
- return 1;
- }
- }
- }
- return 1;
- }
- static int network_state_off_line(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- return 1;
- }
- static int network_state_wait_dns(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- if ((ctrl->need_close) || ctrl->wait_target_state != NW_WAIT_ON_LINE) return -1;
- int i;
- switch(event->ID)
- {
- case EV_NW_RESET:
- case EV_NW_SOCKET_ERROR:
- return -1;
- case EV_NW_STATE:
- if (!event->Param2)
- {
- return -1;
- }
- break;
- case EV_NW_DNS_RESULT:
- if (event->Param1)
- {
- //更新dns cache
- ctrl->dns_ip = event->Param2;
- ctrl->dns_ip_nums = event->Param1;
- for(i = 0; i < ctrl->dns_ip_nums; i++)
- {
- DBG("dns ip%d, ttl %u, %s", i, ctrl->dns_ip[i].ttl_end, ipaddr_ntoa(&ctrl->dns_ip[i].ip));
- }
- network_update_dns_cache(ctrl);
- }
- else
- {
- ctrl->dns_ip_nums = 0;
- network_get_dns_cache(ctrl);
- if (!ctrl->dns_ip_nums)
- {
- return -1;
- }
- }
- ctrl->dns_ip_cnt = 0;
- if (network_base_connect(ctrl, &ctrl->dns_ip[ctrl->dns_ip_cnt].ip))
- {
- network_socket_force_close(ctrl);
- return -1;
- }
- else
- {
- ctrl->state = NW_STATE_CONNECTING;
- return 1;
- }
- default:
- return 1;
- }
- return -1;
- }
- static int network_state_connecting(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- if ((ctrl->need_close) || (ctrl->wait_target_state != NW_WAIT_ON_LINE)) return -1;
- switch(event->ID)
- {
- case EV_NW_RESET:
- return -1;
- case EV_NW_SOCKET_ERROR:
- case EV_NW_SOCKET_REMOTE_CLOSE:
- case EV_NW_SOCKET_CLOSE_OK:
- if (ctrl->remote_ip.type != 0xff)
- {
- return -1;
- }
- DBG("dns ip %d no connect!", ctrl->dns_ip_cnt);
- ctrl->dns_ip_cnt++;
- if (ctrl->dns_ip_cnt >= ctrl->dns_ip_nums)
- {
- DBG("all ip try connect failed");
- return -1;
- }
- if (network_base_connect(ctrl, &ctrl->dns_ip[ctrl->dns_ip_cnt].ip))
- {
- network_socket_force_close(ctrl);
- return -1;
- }
- else
- {
- ctrl->state = NW_STATE_CONNECTING;
- return 1;
- }
- break;
- case EV_NW_STATE:
- if (!event->Param2)
- {
- return -1;
- }
- break;
- case EV_NW_SOCKET_LISTEN:
- if (ctrl->is_server_mode)
- {
- ctrl->state = NW_STATE_LISTEN;
- return 1;
- }
- break;
- case EV_NW_SOCKET_CONNECT_OK:
- #ifdef LUAT_USE_TLS
- if (ctrl->tls_mode)
- {
- mbedtls_ssl_free(ctrl->ssl);
- memset(ctrl->ssl, 0, sizeof(mbedtls_ssl_context));
- mbedtls_ssl_setup(ctrl->ssl, ctrl->config);
- // ctrl->ssl->f_set_timer = tls_settimer;
- // ctrl->ssl->f_get_timer = tls_gettimer;
- // ctrl->ssl->p_timer = ctrl;
- mbedtls_ssl_set_timer_cb(ctrl->ssl, ctrl, tls_settimer, tls_gettimer);
- // ctrl->ssl->p_bio = ctrl;
- // ctrl->ssl->f_send = tls_send;
- // ctrl->ssl->f_recv = tls_recv;
- mbedtls_ssl_set_bio(ctrl->ssl, ctrl, tls_send, tls_recv, NULL);
- // add by wendal
- // cloudflare的https需要设置hostname才能访问
- if (ctrl->domain_name_len > 0 && ctrl->domain_name_len < 256) {
- char host[257] = {0};
- memcpy(host, ctrl->domain_name, ctrl->domain_name_len);
- mbedtls_ssl_set_hostname(ctrl->ssl, host);
- //LLOGD("CALL mbedtls_ssl_set_hostname(%s)", host);
- }
- else {
- //LLOGD("skip mbedtls_ssl_set_hostname");
- }
- ctrl->state = NW_STATE_SHAKEHAND;
- do
- {
- int result = mbedtls_ssl_handshake_step( ctrl->ssl );
- switch(result)
- {
- case MBEDTLS_ERR_SSL_WANT_READ:
- return 1;
- case 0:
- break;
- default:
- #if MBEDTLS_VERSION_NUMBER >= 0x03000000
- #else
- DBG_ERR("0x%x, %d", -result, ctrl->ssl->state);
- #endif
- return -1;
- }
- #if MBEDTLS_VERSION_NUMBER >= 0x03000000
- }while(!mbedtls_ssl_is_handshake_over(ctrl->ssl));
- #else
- }while(ctrl->ssl->state != MBEDTLS_SSL_HANDSHAKE_OVER);
- #endif
- return 0;
- }
- else
- #endif
- {
- ctrl->state = NW_STATE_ONLINE;
- return 0;
- }
- default:
- return 1;
- }
- return -1;
- }
- static int network_state_shakehand(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- if ((ctrl->need_close) || ((ctrl->wait_target_state != NW_WAIT_ON_LINE) && (ctrl->wait_target_state != NW_WAIT_TX_OK))) return -1;
- switch(event->ID)
- {
- case EV_NW_RESET:
- case EV_NW_SOCKET_ERROR:
- case EV_NW_SOCKET_REMOTE_CLOSE:
- case EV_NW_SOCKET_CLOSE_OK:
- ctrl->need_close = 1;
- return -1;
- case EV_NW_STATE:
- if (!event->Param2)
- {
- ctrl->need_close = 1;
- return -1;
- }
- break;
- case EV_NW_SOCKET_TX_OK:
- ctrl->ack_size += event->Param2;
- if (ctrl->is_debug)
- {
- DBG("%llu,%llu",ctrl->tx_size, ctrl->ack_size);
- }
- break;
- #ifdef LUAT_USE_TLS
- case EV_NW_SOCKET_RX_NEW:
- do
- {
- int result = mbedtls_ssl_handshake_step( ctrl->ssl );
- switch(result)
- {
- case MBEDTLS_ERR_SSL_WANT_READ:
- return 1;
- case 0:
- break;
- default:
- #if MBEDTLS_VERSION_NUMBER >= 0x03000000
- #else
- DBG_ERR("0x%x, %d", -result, ctrl->ssl->state);
- #endif
- ctrl->need_close = 1;
- return -1;
- }
- #if MBEDTLS_VERSION_NUMBER >= 0x03000000
- }while(!mbedtls_ssl_is_handshake_over(ctrl->ssl));
- #else
- }while(ctrl->ssl->state != MBEDTLS_SSL_HANDSHAKE_OVER);
- #endif
- ctrl->state = NW_STATE_ONLINE;
- if (NW_WAIT_TX_OK == ctrl->wait_target_state)
- {
- if (!ctrl->cache_data)
- {
- ctrl->need_close = 1;
- return -1;
- }
- int result = mbedtls_ssl_write(ctrl->ssl, ctrl->cache_data, ctrl->cache_len);
- free(ctrl->cache_data);
- ctrl->cache_data = NULL;
- ctrl->cache_len = 0;
- if (result < 0)
- {
- DBG("%08x", -result);
- ctrl->need_close = 1;
- return -1;
- }
- return 1;
- }
- return 0;
- #endif
- case EV_NW_SOCKET_CONNECT_OK:
- DBG_ERR("!");
- return 1;
- default:
- return 1;
- }
- return 1;
- }
- static int network_state_on_line(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- if ((ctrl->need_close) || NW_WAIT_OFF_LINE == ctrl->wait_target_state)
- {
- return -1;
- }
- switch(event->ID)
- {
- case EV_NW_RESET:
- case EV_NW_SOCKET_ERROR:
- case EV_NW_SOCKET_REMOTE_CLOSE:
- case EV_NW_SOCKET_CLOSE_OK:
- ctrl->need_close = 1;
- return -1;
- case EV_NW_STATE:
- if (!event->Param2)
- {
- ctrl->need_close = 1;
- return -1;
- }
- break;
- case EV_NW_SOCKET_TX_OK:
- ctrl->ack_size += event->Param2;
- if (NW_WAIT_TX_OK == ctrl->wait_target_state)
- {
- if (ctrl->ack_size == ctrl->tx_size)
- {
- #ifdef LUAT_USE_LWIP
- if (ctrl->is_tcp)
- {
- return net_lwip_check_all_ack(ctrl->socket_id);
- }
- return 0;
- #else
- return 0;
- #endif
- }
- }
- break;
- case EV_NW_SOCKET_RX_NEW:
- ctrl->new_rx_flag = 1;
- if (NW_WAIT_TX_OK != ctrl->wait_target_state)
- {
- return 0;
- }
- break;
- default:
- return 1;
- }
- return 1;
- }
- static int network_state_listen(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- if ((ctrl->need_close) || NW_WAIT_OFF_LINE == ctrl->wait_target_state)
- {
- return -1;
- }
- switch(event->ID)
- {
- case EV_NW_RESET:
- case EV_NW_SOCKET_ERROR:
- case EV_NW_SOCKET_REMOTE_CLOSE:
- case EV_NW_SOCKET_CLOSE_OK:
- ctrl->need_close = 1;
- return -1;
- case EV_NW_STATE:
- if (!event->Param2)
- {
- ctrl->need_close = 1;
- return -1;
- }
- break;
- case EV_NW_SOCKET_NEW_CONNECT:
- case EV_NW_SOCKET_CONNECT_OK:
- ctrl->state = NW_STATE_ONLINE;
- return 0;
- default:
- return 1;
- }
- return 1;
- }
- static int network_state_disconnecting(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- if (ctrl->wait_target_state != NW_WAIT_OFF_LINE)
- {
- return -1;
- }
- switch(event->ID)
- {
- case EV_NW_RESET:
- case EV_NW_SOCKET_ERROR:
- case EV_NW_SOCKET_REMOTE_CLOSE:
- case EV_NW_SOCKET_CLOSE_OK:
- network_force_close_socket(ctrl);
- ctrl->state = NW_STATE_OFF_LINE;
- ctrl->socket_id = -1;
- return 0;
- case EV_NW_STATE:
- if (!event->Param2)
- {
- return -1;
- }
- else
- {
- network_force_close_socket(ctrl);
- ctrl->state = NW_STATE_OFF_LINE;
- ctrl->socket_id = -1;
- }
- break;
- default:
- return 1;
- }
- return -1;
- }
- typedef int (*network_state_fun)(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter);
- static network_state_fun network_state_fun_list[]=
- {
- network_state_link_off,
- network_state_off_line,
- network_state_wait_dns,
- network_state_connecting,
- network_state_shakehand,
- network_state_on_line,
- network_state_listen,
- network_state_disconnecting,
- };
- static void network_default_statemachine(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- int result = -1;
- NW_LOCK;
- if (ctrl->state > NW_STATE_DISCONNECTING)
- {
- ctrl->state = NW_STATE_LINK_OFF;
- event->Param1 = -1;
- network_force_close_socket(ctrl);
- event->ID = ctrl->wait_target_state + EV_NW_RESULT_BASE;
- }
- else
- {
- result = network_state_fun_list[ctrl->state](ctrl, event, adapter);
- if (result > 0)
- {
- NW_UNLOCK;
- return ;
- }
- event->ID = (ctrl->wait_target_state?ctrl->wait_target_state:NW_WAIT_EVENT) + EV_NW_RESULT_BASE;
- event->Param1 = result;
- }
- if ((ctrl->state != NW_STATE_LISTEN) || (result < 0))
- {
- ctrl->wait_target_state = NW_WAIT_NONE;
- }
- NW_UNLOCK;
- if (ctrl->task_handle)
- {
- platform_send_event(ctrl->task_handle, event->ID, event->Param1, event->Param2, event->Param3);
- }
- else if (ctrl->user_callback)
- {
- ctrl->user_callback(event, ctrl->user_data);
- }
- }
- static int32_t network_default_socket_callback(void *data, void *param)
- {
- OS_EVENT *event = (OS_EVENT *)data;
- OS_EVENT temp_event;
- luat_network_cb_param_t *cb_param = (luat_network_cb_param_t *)param;
- network_adapter_t *adapter =(network_adapter_t *)(cb_param->param);
- int i;
- network_ctrl_t *ctrl = (network_ctrl_t *)event->Param3;
- if (event->ID > EV_NW_TIMEOUT)
- {
- if (ctrl && ((ctrl->tag == cb_param->tag) || (event->ID == EV_NW_DNS_RESULT)))
- {
- if (ctrl->auto_mode)
- {
- DBG("socket %d,%s,%s,%s", ctrl->socket_id, network_ctrl_event_id_string(event->ID),
- network_ctrl_state_string(ctrl->state),
- network_ctrl_wait_state_string(ctrl->wait_target_state));
- network_default_statemachine(ctrl, event, adapter);
- DBG("%s,%s",network_ctrl_state_string(ctrl->state),network_ctrl_wait_state_string(ctrl->wait_target_state));
- }
- else if (ctrl->task_handle)
- {
- platform_send_event(ctrl->task_handle, event->ID, event->Param1, event->Param2, event->Param3);
- }
- else if (ctrl->user_callback)
- {
- ctrl->user_callback(event, ctrl->user_data);
- }
- }
- else
- {
- DBG_ERR("cb ctrl invaild %x", ctrl);
- DBG_HexPrintf(&ctrl->tag, 8);
- DBG_HexPrintf(&cb_param->tag, 8);
- }
- }
- else
- {
- for (i = 0; i < adapter->opt->max_socket_num; i++)
- {
- temp_event = *event;
- if (adapter->ctrl_busy[i])
- {
- ctrl = &adapter->ctrl_table[i];
- if (ctrl->adapter_index == (uint8_t)(event->Param3))
- {
- if (ctrl->auto_mode)
- {
- DBG("socket %d,%s,%s,%s", ctrl->socket_id, network_ctrl_event_id_string(event->ID),
- network_ctrl_state_string(ctrl->state),
- network_ctrl_wait_state_string(ctrl->wait_target_state));
- network_default_statemachine(ctrl, &temp_event, adapter);
- DBG("%s,%s",network_ctrl_state_string(ctrl->state), network_ctrl_wait_state_string(ctrl->wait_target_state));
- }
- else if (ctrl->task_handle)
- {
- platform_send_event(ctrl->task_handle, event->ID, event->Param1, event->Param2, event->Param3);
- }
- else if (ctrl->user_callback)
- {
- ctrl->user_callback(&temp_event, ctrl->user_data);
- }
- }
- }
- }
- }
- return 0;
- }
- static LUAT_RT_RET_TYPE network_default_timer_callback(LUAT_RT_CB_PARAM)
- {
- platform_send_event(param, EV_NW_TIMEOUT, 0, 0, 0);
- return LUAT_RT_RET;
- }
- int network_get_last_register_adapter(void)
- {
- if (prv_network.default_adapter_index != -1) return prv_network.default_adapter_index;
- return prv_network.last_adapter_index;
- }
- void network_register_set_default(uint8_t adapter_index)
- {
- prv_network.default_adapter_index = adapter_index;
- }
- int network_register_adapter(uint8_t adapter_index, network_adapter_info *info, void *user_data)
- {
- prv_adapter_table[adapter_index].opt = info;
- prv_adapter_table[adapter_index].user_data = user_data;
- info->socket_set_callback(network_default_socket_callback, &prv_adapter_table[adapter_index], user_data);
- if (adapter_index < NW_ADAPTER_INDEX_HW_PS_DEVICE)
- {
- prv_adapter_table[adapter_index].ctrl_table = prv_network.lwip_ctrl_table;
- prv_adapter_table[adapter_index].ctrl_busy = prv_network.lwip_ctrl_busy;
- }
- else
- {
- prv_adapter_table[adapter_index].ctrl_table = zalloc((info->max_socket_num) * sizeof(network_ctrl_t));
- prv_adapter_table[adapter_index].ctrl_busy = zalloc(info->max_socket_num);
- }
- prv_adapter_table[adapter_index].port = 60000;
- if (!prv_network.is_init)
- {
- //prv_network.network_mutex = platform_create_mutex();
- INIT_LLIST_HEAD(&prv_network.dns_cache_head);
- prv_network.is_init = 0;
- }
- prv_network.last_adapter_index = adapter_index;
- return 0;
- }
- void network_set_dns_server(uint8_t adapter_index, uint8_t server_index, luat_ip_addr_t *ip)
- {
- network_adapter_t *adapter = &prv_adapter_table[adapter_index];
- adapter->opt->set_dns_server(server_index, ip, adapter->user_data);
- }
- /*
- * 申请一个network_ctrl
- */
- network_ctrl_t *network_alloc_ctrl(uint8_t adapter_index)
- {
- int i;
- network_ctrl_t *ctrl = NULL;
- network_adapter_t *adapter = &prv_adapter_table[adapter_index];
- OS_LOCK;
- for (i = 0; i < adapter->opt->max_socket_num; i++)
- {
- if (!adapter->ctrl_busy[i])
- {
- adapter->ctrl_busy[i] = 1;
- ctrl = &adapter->ctrl_table[i];
- ctrl->adapter_index = adapter_index;
- ctrl->domain_ipv6 = 0;
- break;
- }
- }
- OS_UNLOCK;
- if (i >= adapter->opt->max_socket_num) {DBG_ERR("adapter no more ctrl!");}
- return ctrl;
- }
- /*
- * 归还一个network_ctrl
- */
- void network_release_ctrl(network_ctrl_t *ctrl)
- {
- int i;
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- NW_UNLOCK;
- OS_LOCK;
- for (i = 0; i < adapter->opt->max_socket_num; i++)
- {
- if (&adapter->ctrl_table[i] == ctrl)
- {
- network_deinit_tls(ctrl);
- if (ctrl->timer)
- {
- platform_stop_timer(ctrl->timer);
- platform_release_timer(ctrl->timer);
- ctrl->timer = NULL;
- }
- if (ctrl->cache_data)
- {
- free(ctrl->cache_data);
- ctrl->cache_data = NULL;
- }
- if (ctrl->dns_ip)
- {
- free(ctrl->dns_ip);
- ctrl->dns_ip = NULL;
- }
- if (ctrl->domain_name)
- {
- free(ctrl->domain_name);
- ctrl->domain_name = NULL;
- }
- adapter->ctrl_busy[i] = 0;
- platform_release_mutex(ctrl->mutex);
- ctrl->mutex = NULL;
- break;
- }
- }
- OS_UNLOCK;
- if (i >= adapter->opt->max_socket_num) {DBG_ERR("adapter index maybe error!, %d, %x", ctrl->adapter_index, ctrl);}
- }
- void network_init_ctrl(network_ctrl_t *ctrl, HANDLE task_handle, CBFuncEx_t callback, void *param)
- {
- uint8_t adapter_index = ctrl->adapter_index;
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- if (ctrl->dns_ip)
- {
- free(ctrl->dns_ip);
- }
- if (ctrl->cache_data)
- {
- free(ctrl->cache_data);
- }
- if (ctrl->domain_name)
- {
- free(ctrl->domain_name);
- }
- HANDLE sem = ctrl->mutex;
- memset(ctrl, 0, sizeof(network_ctrl_t));
- ctrl->adapter_index = adapter_index;
- ctrl->task_handle = task_handle;
- ctrl->user_callback = callback;
- ctrl->user_data = param;
- ctrl->socket_id = -1;
- ctrl->socket_param = ctrl;
- ctrl->remote_ip.type = 0xff;
- ctrl->mutex = sem;
- if (task_handle)
- {
- ctrl->timer = platform_create_timer(network_default_timer_callback, task_handle, NULL);
- }
- if (!ctrl->mutex)
- {
- ctrl->mutex = platform_create_mutex();
- }
- }
- void network_set_base_mode(network_ctrl_t *ctrl, uint8_t is_tcp, uint32_t tcp_timeout_ms, uint8_t keep_alive, uint32_t keep_idle, uint8_t keep_interval, uint8_t keep_cnt)
- {
- ctrl->is_tcp = is_tcp;
- ctrl->tcp_keep_alive = keep_alive;
- ctrl->tcp_keep_idle = keep_idle;
- ctrl->tcp_keep_interval = keep_interval;
- ctrl->tcp_keep_cnt = keep_cnt;
- ctrl->tcp_timeout_ms = tcp_timeout_ms;
- }
- void network_connect_ipv6_domain(network_ctrl_t *ctrl, uint8_t onoff)
- {
- ctrl->domain_ipv6 = onoff;
- }
- int network_set_local_port(network_ctrl_t *ctrl, uint16_t local_port)
- {
- int i;
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- if (local_port)
- {
- OS_LOCK;
- for (i = 0; i < adapter->opt->max_socket_num; i++)
- {
- if (&adapter->ctrl_table[i] != ctrl)
- {
- if (adapter->ctrl_table[i].local_port == local_port)
- {
- OS_UNLOCK;
- return -1;
- }
- }
- }
- ctrl->local_port = local_port;
- OS_UNLOCK;
- return 0;
- }
- else
- {
- // OS_LOCK;
- // adapter->port++;
- // if (adapter->port < 60000)
- // {
- // adapter->port = 60000;
- // }
- // ctrl->local_port = adapter->port;
- // OS_UNLOCK;
- ctrl->local_port = 0;
- return 0;
- }
- }
- uint8_t network_check_ready(network_ctrl_t *ctrl)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- return adapter->opt->check_ready(adapter->user_data);
- }
- int network_create_soceket(network_ctrl_t *ctrl, uint8_t is_ipv6)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- ctrl->socket_id = adapter->opt->create_soceket(ctrl->is_tcp, &ctrl->tag, ctrl->socket_param, is_ipv6, adapter->user_data);
- if (ctrl->socket_id < 0)
- {
- ctrl->tag = 0;
- return -1;
- }
- return 0;
- }
- int network_socket_connect(network_ctrl_t *ctrl, luat_ip_addr_t *remote_ip)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- ctrl->is_server_mode = 0;
- ctrl->online_ip = remote_ip;
- return adapter->opt->socket_connect(ctrl->socket_id, ctrl->tag, ctrl->local_port, remote_ip, ctrl->remote_port, adapter->user_data);
- }
- int network_socket_listen(network_ctrl_t *ctrl)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- ctrl->is_server_mode = 1;
- return adapter->opt->socket_listen(ctrl->socket_id, ctrl->tag, ctrl->local_port, adapter->user_data);
- }
- uint8_t network_accept_enable(network_ctrl_t *ctrl)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- return !(adapter->opt->no_accept);
- }
- //作为server接受一个client
- //成功返回0,失败 < 0
- int network_socket_accept(network_ctrl_t *ctrl, network_ctrl_t *accept_ctrl)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- if (adapter->opt->no_accept)
- {
- // DBG("%x,%d,%llu,%x,%x,%x",adapter->opt->socket_accept, ctrl->socket_id, ctrl->tag, &ctrl->remote_ip, &ctrl->remote_port, adapter->user_data);
- adapter->opt->socket_accept(ctrl->socket_id, ctrl->tag, &ctrl->remote_ip, &ctrl->remote_port, adapter->user_data);
- return 0;
- }
- accept_ctrl->socket_id = adapter->opt->socket_accept(ctrl->socket_id, ctrl->tag, &accept_ctrl->remote_ip, &accept_ctrl->remote_port, adapter->user_data);
- if (accept_ctrl->socket_id < 0)
- {
- return -1;
- }
- else
- {
- accept_ctrl->is_tcp = ctrl->is_tcp;
- accept_ctrl->tcp_keep_alive = ctrl->tcp_keep_alive;
- accept_ctrl->tcp_keep_idle = ctrl->tcp_keep_idle;
- accept_ctrl->tcp_keep_interval = ctrl->tcp_keep_interval;
- accept_ctrl->tcp_keep_cnt = ctrl->tcp_keep_cnt;
- accept_ctrl->tcp_timeout_ms = ctrl->tcp_timeout_ms;
- accept_ctrl->local_port = ctrl->local_port;
- accept_ctrl->state = NW_STATE_ONLINE;
- return 0;
- }
- }
- //主动断开一个tcp连接,需要走完整个tcp流程,用户需要接收到close ok回调才能确认彻底断开
- //成功返回0,失败 < 0
- int network_socket_disconnect(network_ctrl_t *ctrl)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- if (ctrl->socket_id >= 0)
- {
- return adapter->opt->socket_disconnect(ctrl->socket_id, ctrl->tag, adapter->user_data);
- }
- return 0;
- }
- //释放掉socket的控制权
- //成功返回0,失败 < 0
- int network_socket_close(network_ctrl_t *ctrl)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- if (ctrl->socket_id >= 0)
- {
- return adapter->opt->socket_close(ctrl->socket_id, ctrl->tag, adapter->user_data);
- }
- return 0;
- }
- //强行释放掉socket的控制权
- //成功返回0,失败 < 0
- int network_socket_force_close(network_ctrl_t *ctrl)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- if (ctrl->socket_id >= 0)
- {
- adapter->opt->socket_force_close(ctrl->socket_id, adapter->user_data);
- }
- ctrl->socket_id = -1;
- return 0;
- }
- //tcp时,不需要remote_ip和remote_port,如果buf为NULL,则返回当前缓存区的数据量,当返回值小于len时说明已经读完了
- //udp时,只返回1个block数据,需要多次读直到没有数据为止
- //成功返回实际读取的值,失败 < 0
- int network_socket_receive(network_ctrl_t *ctrl,uint8_t *buf, uint32_t len, int flags, luat_ip_addr_t *remote_ip, uint16_t *remote_port)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- return adapter->opt->socket_receive(ctrl->socket_id, ctrl->tag, buf, len, flags, remote_ip, remote_port, adapter->user_data);
- }
- //tcp时,不需要remote_ip和remote_port
- //成功返回0,失败 < 0
- int network_socket_send(network_ctrl_t *ctrl,const uint8_t *buf, uint32_t len, int flags, luat_ip_addr_t *remote_ip, uint16_t remote_port)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- return adapter->opt->socket_send(ctrl->socket_id, ctrl->tag, buf, len, flags, remote_ip, remote_port, adapter->user_data);
- }
- int network_getsockopt(network_ctrl_t *ctrl, int level, int optname, void *optval, uint32_t *optlen)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- return adapter->opt->getsockopt(ctrl->socket_id, ctrl->tag, level, optname, optval, optlen, adapter->user_data);
- }
- int network_setsockopt(network_ctrl_t *ctrl, int level, int optname, const void *optval, uint32_t optlen)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- return adapter->opt->setsockopt(ctrl->socket_id, ctrl->tag, level, optname, optval, optlen, adapter->user_data);
- }
- //非posix的socket,用这个根据实际硬件设置参数
- int network_user_cmd(network_ctrl_t *ctrl, uint32_t cmd, uint32_t value)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- return adapter->opt->user_cmd(ctrl->socket_id, ctrl->tag, cmd, value, adapter->user_data);
- }
- int network_dns(network_ctrl_t *ctrl)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- if (ctrl->domain_ipv6)
- {
- return adapter->opt->dns_ipv6(ctrl->domain_name, ctrl->domain_name_len, ctrl, adapter->user_data);
- }
- else
- {
- return adapter->opt->dns(ctrl->domain_name, ctrl->domain_name_len, ctrl, adapter->user_data);
- }
- }
- int network_set_mac(uint8_t adapter_index, uint8_t *mac)
- {
- network_adapter_t *adapter = &prv_adapter_table[adapter_index];
- return adapter->opt->set_mac(mac, adapter->user_data);
- }
- int network_set_static_ip_info(uint8_t adapter_index, luat_ip_addr_t *ip, luat_ip_addr_t *submask, luat_ip_addr_t *gateway, luat_ip_addr_t *ipv6)
- {
- network_adapter_t *adapter = &prv_adapter_table[adapter_index];
- return adapter->opt->set_static_ip(ip, submask, gateway, ipv6, adapter->user_data);
- }
- int network_get_local_ip_info(network_ctrl_t *ctrl, luat_ip_addr_t *ip, luat_ip_addr_t *submask, luat_ip_addr_t *gateway)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- return adapter->opt->get_local_ip_info(ip, submask, gateway, adapter->user_data);
- }
- int network_get_full_local_ip_info(network_ctrl_t *ctrl, uint8_t index, luat_ip_addr_t *ip, luat_ip_addr_t *submask, luat_ip_addr_t *gateway, luat_ip_addr_t *ipv6)
- {
- network_adapter_t *adapter;
- if (ctrl)
- {
- adapter = &prv_adapter_table[ctrl->adapter_index];
- }
- else
- {
- adapter = &prv_adapter_table[index];
- }
- return adapter->opt->get_full_ip_info(ip, submask, gateway, ipv6, adapter->user_data);
- }
- void network_force_close_socket(network_ctrl_t *ctrl)
- {
- #ifdef LUAT_USE_TLS
- if (ctrl->tls_mode)
- {
- mbedtls_ssl_free(ctrl->ssl);
- }
- #endif
- if (network_socket_close(ctrl))
- {
- network_clean_invaild_socket(ctrl->adapter_index);
- network_socket_force_close(ctrl);
- }
- ctrl->need_close = 0;
- ctrl->socket_id = -1;
- if (ctrl->dns_ip)
- {
- free(ctrl->dns_ip);
- ctrl->dns_ip = NULL;
- }
- if (ctrl->domain_name)
- {
- free(ctrl->domain_name);
- ctrl->domain_name = NULL;
- }
- ctrl->dns_ip_cnt = 0;
- ctrl->dns_ip_nums = 0;
- }
- void network_clean_invaild_socket(uint8_t adapter_index)
- {
- int i;
- int *list;
- network_adapter_t *adapter = &prv_adapter_table[adapter_index];
- network_ctrl_t *ctrl;
- list = malloc(adapter->opt->max_socket_num * sizeof(int));
- OS_LOCK;
- for (i = 0; i < adapter->opt->max_socket_num; i++)
- {
- ctrl = &adapter->ctrl_table[i];
- if (!adapter->opt->socket_check(ctrl->socket_id, ctrl->tag, adapter->user_data))
- {
- list[i] = ctrl->socket_id;
- }
- else
- {
- ctrl->socket_id = -1;
- list[i] = -1;
- }
- DBG("%d,%d", i, list[i]);
- }
- OS_UNLOCK;
- adapter->opt->socket_clean(list, adapter->opt->max_socket_num, adapter->user_data);
- free(list);
- }
- #ifdef LUAT_USE_TLS
- static int tls_verify(void *ctx, mbedtls_x509_crt *crt, int Index, uint32_t *result)
- {
- network_ctrl_t *ctrl = (network_ctrl_t *)ctx;
- DBG("%d, %08x", Index, *result);
- return 0;
- }
- #endif
- int network_set_psk_info(network_ctrl_t *ctrl,
- const unsigned char *psk, size_t psk_len,
- const unsigned char *psk_identity, size_t psk_identity_len)
- {
- #ifdef LUAT_USE_TLS
- if (!ctrl->tls_mode)
- {
- return -ERROR_PERMISSION_DENIED;
- }
- // DBG("%.*s, %.*s", psk_len, psk, psk_identity_len, psk_identity);
- int ret = mbedtls_ssl_conf_psk( ctrl->config,
- psk, psk_len, psk_identity, psk_identity_len );
- if (ret != 0)
- {
- DBG("0x%x", -ret);
- return -ERROR_OPERATION_FAILED;
- }
- return ERROR_NONE;
- #else
- return -1;
- #endif
- }
- int network_set_server_cert(network_ctrl_t *ctrl, const unsigned char *cert, size_t cert_len)
- {
- #ifdef LUAT_USE_TLS
- int ret;
- if (!ctrl->tls_mode)
- {
- return -ERROR_PERMISSION_DENIED;
- }
- ret = mbedtls_x509_crt_parse( ctrl->ca_cert, cert, cert_len);
- if (ret != 0)
- {
- DBG("%08x", -ret);
- return -ERROR_OPERATION_FAILED;
- }
- return ERROR_NONE;
- #else
- return -1;
- #endif
- }
- int network_set_client_cert(network_ctrl_t *ctrl,
- const unsigned char *cert, size_t certLen,
- const unsigned char *key, size_t keylen,
- const unsigned char *pwd, size_t pwdlen)
- {
- #ifdef LUAT_USE_TLS
- int ret;
- mbedtls_x509_crt *client_cert = NULL;
- mbedtls_pk_context *pkey = NULL;
- if (!ctrl->tls_mode)
- {
- return -ERROR_PERMISSION_DENIED;
- }
- client_cert = zalloc(sizeof(mbedtls_x509_crt));
- pkey = zalloc(sizeof(mbedtls_pk_context));
- if (!client_cert || !pkey)
- {
- goto ERROR_OUT;
- }
- ret = mbedtls_x509_crt_parse( client_cert, cert, certLen );
- if (ret != 0)
- {
- DBG("%08x", -ret);
- goto ERROR_OUT;
- }
- #if MBEDTLS_VERSION_NUMBER >= 0x03000000
- ret = mbedtls_pk_parse_key( pkey, key, keylen, pwd, pwdlen , tls_random, NULL);
- #else
- ret = mbedtls_pk_parse_key( pkey, key, keylen, pwd, pwdlen );
- #endif
- if (ret != 0)
- {
- DBG("%08x", -ret);
- goto ERROR_OUT;
- }
- ret = mbedtls_ssl_conf_own_cert( ctrl->config, client_cert, pkey );
- if (ret != 0)
- {
- DBG("%08x", -ret);
- goto ERROR_OUT;
- }
- return ERROR_NONE;
- ERROR_OUT:
- if (client_cert) free(client_cert);
- if (pkey) free(pkey);
- return -1;
- #else
- return -1;
- #endif
- }
- int network_cert_verify_result(network_ctrl_t *ctrl)
- {
- #ifdef LUAT_USE_TLS
- if (!ctrl->tls_mode)
- {
- return -1;
- }
- return mbedtls_ssl_get_verify_result(ctrl->ssl);
- #else
- return -1;
- #endif
- }
- static int tls_random( void *p_rng,
- unsigned char *output, size_t output_len )
- {
- platform_random((char*)output, output_len);
- return 0;
- }
- void network_init_tls(network_ctrl_t *ctrl, int verify_mode)
- {
- #ifdef LUAT_USE_TLS
- ctrl->tls_mode = 1;
- if (!ctrl->ssl)
- {
- ctrl->ssl = zalloc(sizeof(mbedtls_ssl_context));
- ctrl->ca_cert = zalloc(sizeof(mbedtls_x509_crt));
- ctrl->config = zalloc(sizeof(mbedtls_ssl_config));
- mbedtls_ssl_config_defaults( ctrl->config, MBEDTLS_SSL_IS_CLIENT, ctrl->is_tcp?MBEDTLS_SSL_TRANSPORT_STREAM:MBEDTLS_SSL_TRANSPORT_DATAGRAM, MBEDTLS_SSL_PRESET_DEFAULT);
- // ctrl->config->authmode = verify_mode;
- mbedtls_ssl_conf_authmode(ctrl->config, verify_mode);
- // ctrl->config->hs_timeout_min = 20000;
- #if defined(MBEDTLS_SSL_PROTO_DTLS)
- mbedtls_ssl_conf_handshake_timeout(ctrl->config, 2000, MBEDTLS_SSL_DTLS_TIMEOUT_DFL_MAX);
- #endif
- // ctrl->config->f_rng = tls_random;
- // ctrl->config->p_rng = NULL;
- mbedtls_ssl_conf_rng(ctrl->config, tls_random, NULL);
- // ctrl->config->f_dbg = tls_dbg;
- // ctrl->config->p_dbg = NULL;
- mbedtls_ssl_conf_dbg(ctrl->config, tls_dbg, NULL);
- // ctrl->config->f_vrfy = tls_verify;
- // ctrl->config->p_vrfy = ctrl;
- mbedtls_ssl_conf_verify(ctrl->config, tls_verify, ctrl);
- // ctrl->config->ca_chain = ctrl->ca_cert;
- mbedtls_ssl_conf_ca_chain(ctrl->config, ctrl->ca_cert, NULL);
- // ctrl->config->read_timeout = 20000;
- mbedtls_ssl_conf_read_timeout(ctrl->config, 20000);
- ctrl->tls_long_timer = platform_create_timer(tls_longtimeout, ctrl, NULL);
- ctrl->tls_short_timer = platform_create_timer(tls_shorttimeout, ctrl, NULL);
- }
- ctrl->tls_timer_state = -1;
- #endif
- }
- void network_deinit_tls(network_ctrl_t *ctrl)
- {
- #ifdef LUAT_USE_TLS
- if (ctrl->ssl)
- {
- mbedtls_ssl_free(ctrl->ssl);
- free(ctrl->ssl);
- ctrl->ssl = NULL;
- }
- if (ctrl->config)
- {
- mbedtls_ssl_config_free(ctrl->config);
- free(ctrl->config);
- ctrl->config = NULL;
- }
- if (ctrl->ca_cert)
- {
- mbedtls_x509_crt_free(ctrl->ca_cert);
- free(ctrl->ca_cert);
- ctrl->ca_cert = NULL;
- }
- ctrl->tls_mode = 0;
- ctrl->tls_timer_state = -1;
- if (ctrl->tls_short_timer)
- {
- platform_release_timer(ctrl->tls_short_timer);
- ctrl->tls_short_timer = NULL;
- }
- if (ctrl->tls_long_timer)
- {
- platform_release_timer(ctrl->tls_long_timer);
- ctrl->tls_long_timer = NULL;
- }
- #endif
- }
- int network_wait_link_up(network_ctrl_t *ctrl, uint32_t timeout_ms)
- {
- NW_LOCK;
- ctrl->auto_mode = 1;
- // network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- if (network_check_ready(ctrl))
- {
- ctrl->state = NW_STATE_OFF_LINE;
- ctrl->wait_target_state = NW_WAIT_NONE;
- NW_UNLOCK;
- return 0;
- }
- ctrl->state = NW_STATE_LINK_OFF;
- ctrl->wait_target_state = NW_WAIT_LINK_UP;
- NW_UNLOCK;
- if (!ctrl->task_handle || !timeout_ms)
- {
- return 1;
- }
- uint8_t finish = 0;
- OS_EVENT event;
- int result;
- //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
- platform_start_timer(ctrl->timer, timeout_ms, 0);
- while (!finish)
- {
- platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
- switch (event.ID)
- {
- case EV_NW_RESULT_LINK:
- result = (int)event.Param1;
- finish = 1;
- break;
- case EV_NW_TIMEOUT:
- result = -1;
- finish = 1;
- break;
- default:
- if (ctrl->user_callback)
- {
- ctrl->user_callback((void *)&event, ctrl->user_data);
- }
- break;
- }
- }
- platform_stop_timer(ctrl->timer);
- return result;
- }
- /*
- * 1.进行ready检测和等待ready
- * 2.有remote_ip则开始连接服务器并等待连接结果
- * 3.没有remote_ip则开始对url进行dns解析,解析完成后对所有ip进行尝试连接直到有个成功或者全部失败
- * 4.如果是加密模式,还要走握手环节,等到握手环节完成后才能返回结果
- * local_port如果为0则api内部自动生成一个
- */
- int network_connect(network_ctrl_t *ctrl, const char *domain_name, uint32_t domain_name_len, luat_ip_addr_t *remote_ip, uint16_t remote_port, uint32_t timeout_ms)
- {
- if (ctrl->socket_id >= 0)
- {
- return -1;
- }
- NW_LOCK;
- ctrl->is_server_mode = 0;
- ctrl->tx_size = 0;
- ctrl->ack_size = 0;
- if (ctrl->dns_ip)
- {
- free(ctrl->dns_ip);
- ctrl->dns_ip = NULL;
- }
- if (ctrl->cache_data)
- {
- free(ctrl->cache_data);
- ctrl->cache_data = NULL;
- }
- ctrl->need_close = 0;
- if (ctrl->domain_name)
- {
- free(ctrl->domain_name);
- }
- ctrl->domain_name = zalloc(domain_name_len + 1);
- memcpy(ctrl->domain_name, domain_name, domain_name_len);
- ctrl->domain_name_len = domain_name_len;
- if (remote_ip)
- {
- ctrl->remote_ip = *remote_ip;
- }
- else
- {
- ctrl->remote_ip.type = 0xff;
- }
- ctrl->auto_mode = 1;
- ctrl->remote_port = remote_port;
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- ctrl->wait_target_state = NW_WAIT_ON_LINE;
- if (!network_check_ready(ctrl))
- {
- ctrl->state = NW_STATE_LINK_OFF;
- goto NETWORK_CONNECT_WAIT;
- }
- if (network_prepare_connect(ctrl))
- {
- ctrl->state = NW_STATE_OFF_LINE;
- ctrl->wait_target_state = NW_WAIT_NONE;
- NW_UNLOCK;
- return -1;
- }
- NETWORK_CONNECT_WAIT:
- NW_UNLOCK;
- if (!ctrl->task_handle || !timeout_ms)
- {
- return 1;
- }
- uint8_t finish = 0;
- OS_EVENT event;
- int result;
- //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
- platform_start_timer(ctrl->timer, timeout_ms, 0);
- while (!finish)
- {
- platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
- switch (event.ID)
- {
- case EV_NW_RESULT_CONNECT:
- result = (int)event.Param1;
- finish = 1;
- break;
- case EV_NW_TIMEOUT:
- result = -1;
- finish = 1;
- break;
- default:
- if (ctrl->user_callback)
- {
- ctrl->user_callback((void *)&event, ctrl->user_data);
- }
- break;
- }
- }
- platform_stop_timer(ctrl->timer);
- return result;
- }
- int network_listen(network_ctrl_t *ctrl, uint32_t timeout_ms)
- {
- if (NW_STATE_LISTEN == ctrl->state)
- {
- DBG("socket %d is listen", ctrl->socket_id);
- return 0;
- }
- if (ctrl->socket_id >= 0)
- {
- return -1;
- }
- NW_LOCK;
- ctrl->is_server_mode = 1;
- ctrl->auto_mode = 1;
- ctrl->need_close = 0;
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- ctrl->wait_target_state = NW_WAIT_ON_LINE;
- if (!network_check_ready(ctrl))
- {
- ctrl->state = NW_STATE_LINK_OFF;
- goto NETWORK_LISTEN_WAIT;
- }
- if (network_base_connect(ctrl, NULL))
- {
- ctrl->state = NW_STATE_OFF_LINE;
- ctrl->wait_target_state = NW_WAIT_NONE;
- NW_UNLOCK;
- return -1;
- }
- ctrl->state = NW_STATE_CONNECTING;
- NETWORK_LISTEN_WAIT:
- NW_UNLOCK;
- if (!ctrl->task_handle || !timeout_ms)
- {
- return 1;
- }
- uint8_t finish = 0;
- OS_EVENT event;
- int result;
- //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
- if (timeout_ms != 0xffffffff)
- {
- platform_start_timer(ctrl->timer, timeout_ms, 0);
- }
- while (!finish)
- {
- platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
- switch (event.ID)
- {
- case EV_NW_RESULT_CONNECT:
- result = (int)event.Param1;
- finish = 1;
- break;
- case EV_NW_TIMEOUT:
- result = -1;
- finish = 1;
- break;
- default:
- if (ctrl->user_callback)
- {
- ctrl->user_callback((void *)&event, ctrl->user_data);
- }
- break;
- }
- }
- if (timeout_ms != 0xffffffff)
- {
- platform_stop_timer(ctrl->timer);
- }
- return result;
- }
- int network_close(network_ctrl_t *ctrl, uint32_t timeout_ms)
- {
- NW_LOCK;
- if (ctrl->cache_data)
- {
- free(ctrl->cache_data);
- ctrl->cache_data = NULL;
- }
- uint8_t old_state = ctrl->state;
- ctrl->auto_mode = 1;
- ctrl->need_close = 0;
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- #ifdef LUAT_USE_TLS
- if (ctrl->tls_mode)
- {
- mbedtls_ssl_free(ctrl->ssl);
- }
- #endif
- if (ctrl->socket_id < 0)
- {
- ctrl->state = NW_STATE_OFF_LINE;
- ctrl->wait_target_state = NW_WAIT_NONE;
- NW_UNLOCK;
- return 0;
- }
- ctrl->state = NW_STATE_DISCONNECTING;
- ctrl->wait_target_state = NW_WAIT_OFF_LINE;
- if ((NW_STATE_ONLINE == old_state) && ctrl->is_tcp)
- {
- if (network_socket_disconnect(ctrl))
- {
- network_force_close_socket(ctrl);
- ctrl->state = NW_STATE_OFF_LINE;
- ctrl->wait_target_state = NW_WAIT_NONE;
- NW_UNLOCK;
- return 0;
- }
- }
- else
- {
- network_force_close_socket(ctrl);
- ctrl->state = NW_STATE_OFF_LINE;
- ctrl->wait_target_state = NW_WAIT_NONE;
- NW_UNLOCK;
- return 0;
- }
- NW_UNLOCK;
- if (!ctrl->task_handle || !timeout_ms)
- {
- return 1;
- }
- uint8_t finish = 0;
- OS_EVENT event;
- int result;
- //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
- platform_start_timer(ctrl->timer, timeout_ms, 0);
- while (!finish)
- {
- platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
- switch (event.ID)
- {
- case EV_NW_RESULT_CLOSE:
- result = 0;
- finish = 1;
- break;
- case EV_NW_TIMEOUT:
- result = 0;
- finish = 1;
- break;
- default:
- if (ctrl->user_callback)
- {
- ctrl->user_callback((void *)&event, ctrl->user_data);
- }
- break;
- }
- }
- platform_stop_timer(ctrl->timer);
- network_socket_force_close(ctrl);
- return result;
- }
- /*
- * timeout_ms=0时,为非阻塞接口
- */
- int network_tx(network_ctrl_t *ctrl, const uint8_t *data, uint32_t len, int flags, luat_ip_addr_t *remote_ip, uint16_t remote_port, uint32_t *tx_len, uint32_t timeout_ms)
- {
- if ((ctrl->need_close) || (ctrl->socket_id < 0) || (ctrl->state != NW_STATE_ONLINE))
- {
- return -1;
- }
- NW_LOCK;
- int result;
- ctrl->auto_mode = 1;
- #ifdef LUAT_USE_TLS
- if (ctrl->tls_mode)
- {
- if (ctrl->tls_need_reshakehand)
- {
- ctrl->tls_need_reshakehand = 0;
- if (ctrl->cache_data)
- {
- free(ctrl->cache_data);
- ctrl->cache_data = NULL;
- }
- ctrl->cache_data = malloc(len);
- memcpy(ctrl->cache_data, data, len);
- ctrl->cache_len = len;
- mbedtls_ssl_session_reset(ctrl->ssl);
- do
- {
- result = mbedtls_ssl_handshake_step( ctrl->ssl );
- switch(result)
- {
- case MBEDTLS_ERR_SSL_WANT_READ:
- ctrl->state = NW_STATE_SHAKEHAND;
- goto NETWORK_TX_WAIT;
- case 0:
- break;
- default:
- #if MBEDTLS_VERSION_NUMBER >= 0x03000000
- #else
- DBG_ERR("0x%x, %d", -result, ctrl->ssl->state);
- #endif
- ctrl->need_close = 1;
- NW_UNLOCK;
- return -1;
- }
- #if MBEDTLS_VERSION_NUMBER >= 0x03000000
- }while(!mbedtls_ssl_is_handshake_over(ctrl->ssl));
- #else
- }while(ctrl->ssl->state != MBEDTLS_SSL_HANDSHAKE_OVER);
- #endif
- }
- result = mbedtls_ssl_write(ctrl->ssl, data, len);
- if (result < 0)
- {
- DBG("%08x", -result);
- ctrl->need_close = 1;
- NW_UNLOCK;
- return -1;
- }
- *tx_len = result;
- }
- else
- #endif
- {
- result = network_base_tx(ctrl, data, len, flags, remote_ip, remote_port);
- if (result < 0)
- {
- ctrl->need_close = 1;
- NW_UNLOCK;
- return -1;
- }
- *tx_len = result;
- if (!result && len)
- {
- NW_UNLOCK;
- return 0;
- }
- }
- #ifdef LUAT_USE_TLS
- NETWORK_TX_WAIT:
- #endif
- ctrl->wait_target_state = NW_WAIT_TX_OK;
- NW_UNLOCK;
- if (!ctrl->task_handle || !timeout_ms)
- {
- return 1;
- }
- uint8_t finish = 0;
- OS_EVENT event;
- //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
- platform_start_timer(ctrl->timer, timeout_ms, 0);
- while (!finish)
- {
- platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
- switch (event.ID)
- {
- case EV_NW_RESULT_TX:
- result = (int)event.Param1;
- finish = 1;
- break;
- case EV_NW_TIMEOUT:
- result = -1;
- finish = 1;
- break;
- default:
- if (ctrl->user_callback)
- {
- ctrl->user_callback((void *)&event, ctrl->user_data);
- }
- break;
- }
- }
- platform_stop_timer(ctrl->timer);
- return result;
- }
- /*
- * 实际读到的数据量在rx_len里,如果是UDP模式且为server时,需要看remote_ip和remote_port
- */
- int network_rx(network_ctrl_t *ctrl, uint8_t *data, uint32_t len, int flags, luat_ip_addr_t *remote_ip, uint16_t *remote_port, uint32_t *rx_len)
- {
- if (!ctrl->new_rx_flag && ((ctrl->need_close) || (ctrl->socket_id < 0) || (ctrl->state != NW_STATE_ONLINE)))
- {
- return -1;
- }
- NW_LOCK;
- int result = -1;
- ctrl->auto_mode = 1;
- uint32_t read_len = 0;
- uint8_t is_error = 0;
- if (data)
- {
- ctrl->new_rx_flag = 0;
- #ifdef LUAT_USE_TLS
- if (ctrl->tls_mode)
- {
- do
- {
- result = mbedtls_ssl_read(ctrl->ssl, data + read_len, len - read_len);
- if (result < 0 && (result != MBEDTLS_ERR_SSL_WANT_READ))
- {
- is_error = 1;
- break;
- }
- else
- {
- read_len += result;
- if (read_len >= len)
- {
- break;
- }
- }
- }while(network_socket_receive(ctrl, NULL, len, flags, remote_ip, remote_port) > 0);
- if ( !is_error )
- {
- result = read_len;
- }
- else
- {
- result = -1;
- }
- }
- else
- #endif
- {
- result = network_socket_receive(ctrl, data, len, flags, remote_ip, remote_port);
- }
- }
- else
- {
- #ifdef LUAT_USE_TLS
- if (ctrl->tls_mode)
- {
- read_len = 0;
- do
- {
- result = mbedtls_ssl_read(ctrl->ssl, NULL, 0);
- if ((result < 0) && (result != (MBEDTLS_ERR_SSL_WANT_READ)))
- {
- is_error = 1;
- read_len = 0;
- break;
- }
- else if (!result)
- {
- #if MBEDTLS_VERSION_NUMBER >= 0x03000000
- read_len = ctrl->ssl->MBEDTLS_PRIVATE(in_msglen);
- #else
- read_len = ctrl->ssl->in_msglen;
- #endif
- break;
- }
- else if ((MBEDTLS_ERR_SSL_WANT_READ) == result)
- {
- read_len = 0;
- ctrl->new_rx_flag = 0;
- DBG("socket %d ssl data need more", ctrl->socket_id);
- break;
- }
- }while(network_socket_receive(ctrl, NULL, len, flags, remote_ip, remote_port) > 0);
- if ( !is_error )
- {
- result = read_len;
- }
- else
- {
- result = -1;
- }
- }
- else
- #endif
- {
- result = network_socket_receive(ctrl, data, len, flags, remote_ip, remote_port);
- }
- }
- NW_UNLOCK;
- if (result >= 0)
- {
- *rx_len = result;
- return 0;
- }
- else
- {
- return -1;
- }
- }
- int network_wait_event(network_ctrl_t *ctrl, OS_EVENT *out_event, uint32_t timeout_ms, uint8_t *is_timeout)
- {
- if (ctrl->new_rx_flag)
- {
- ctrl->wait_target_state = NW_WAIT_EVENT;
- if (out_event)
- {
- out_event->ID = 0;
- }
- return 0;
- }
- if ((ctrl->need_close) || (ctrl->socket_id < 0) || (ctrl->state != NW_STATE_ONLINE))
- {
- return -1;
- }
- NW_LOCK;
- ctrl->auto_mode = 1;
- ctrl->wait_target_state = NW_WAIT_EVENT;
- NW_UNLOCK;
- if (!ctrl->task_handle || !timeout_ms)
- {
- return 1;
- }
- *is_timeout = 0;
- uint8_t finish = 0;
- OS_EVENT event;
- int result;
- //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
- platform_start_timer(ctrl->timer, timeout_ms, 0);
- while (!finish)
- {
- platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
- switch (event.ID)
- {
- case EV_NW_RESULT_EVENT:
- result = (int)event.Param1;
- if (result)
- {
- result = -1;
- }
- if (out_event)
- {
- out_event->ID = 0;
- }
- finish = 1;
- break;
- case EV_NW_TIMEOUT:
- *is_timeout = 1;
- result = 0;
- finish = 1;
- break;
- case EV_NW_BREAK_WAIT:
- if (out_event)
- {
- *out_event = event;
- }
- result = 0;
- finish = 1;
- break;
- default:
- if (out_event)
- {
- *out_event = event;
- result = 0;
- finish = 1;
- break;
- }
- else if (ctrl->user_callback)
- {
- ctrl->user_callback((void *)&event, ctrl->user_data);
- }
- break;
- }
- }
- platform_stop_timer(ctrl->timer);
- return result;
- }
- int network_wait_rx(network_ctrl_t *ctrl, uint32_t timeout_ms, uint8_t *is_break, uint8_t *is_timeout)
- {
- *is_timeout = 0;
- *is_break = 0;
- if (ctrl->new_rx_flag)
- {
- ctrl->wait_target_state = NW_WAIT_EVENT;
- return 0;
- }
- if ((ctrl->need_close) || (ctrl->socket_id < 0) || (ctrl->state != NW_STATE_ONLINE))
- {
- return -1;
- }
- NW_LOCK;
- ctrl->auto_mode = 1;
- ctrl->wait_target_state = NW_WAIT_EVENT;
- NW_UNLOCK;
- uint8_t finish = 0;
- OS_EVENT event;
- int result;
- //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
- platform_start_timer(ctrl->timer, timeout_ms, 0);
- while (!finish)
- {
- platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
- switch (event.ID)
- {
- case EV_NW_RESULT_EVENT:
- result = (int)event.Param1;
- if (result)
- {
- result = -1;
- finish = 1;
- }
- else if (ctrl->new_rx_flag)
- {
- result = 0;
- finish = 1;
- }
- break;
- case EV_NW_TIMEOUT:
- *is_timeout = 1;
- result = 0;
- finish = 1;
- break;
- case EV_NW_BREAK_WAIT:
- *is_break = 1;
- result = 0;
- finish = 1;
- break;
- default:
- if (ctrl->user_callback)
- {
- ctrl->user_callback((void *)&event, ctrl->user_data);
- }
- break;
- }
- ctrl->wait_target_state = NW_WAIT_EVENT;
- }
- platform_stop_timer(ctrl->timer);
- return result;
- }
- #else
- #include "dhcp_def.h"
- extern void DBG_Printf(const char* format, ...);
- extern void DBG_HexPrintf(void *Data, unsigned int len);
- //#define DBG(x,y...) DBG_Printf("%s %d:"x"\r\n", __FUNCTION__,__LINE__,##y)
- //#define DBG_ERR(x,y...) DBG_Printf("%s %d:"x"\r\n", __FUNCTION__,__LINE__,##y)
- #define __NW_DEBUG_ENABLE__
- #ifdef __NW_DEBUG_ENABLE__
- #define DBG(x,y...) do {if (ctrl->is_debug) {DBG_Printf("%s %d:"x"\r\n", __FUNCTION__,__LINE__,##y);}} while(0)
- #define DBG_ERR(x,y...) DBG_Printf("%s %d:"x"\r\n", __FUNCTION__,__LINE__,##y)
- #else
- #define DBG(x,y...)
- #define DBG_ERR(x,y...)
- #endif
- //#define NW_LOCK OS_LOCK
- //#define NW_UNLOCK OS_UNLOCK
- #define NW_LOCK platform_lock_mutex(ctrl->mutex)
- #define NW_UNLOCK platform_unlock_mutex(ctrl->mutex)
- #define SOL_SOCKET 0xfff /* options for socket level */
- #define SO_REUSEADDR 0x0004 /* Allow local address reuse */
- #define SO_KEEPALIVE 0x0008 /* keep connections alive */
- #define IPPROTO_ICMP 1
- #define IPPROTO_TCP 6
- #define TCP_NODELAY 0x01 /* don't delay send to coalesce packets */
- #define TCP_KEEPALIVE 0x02 /* send KEEPALIVE probes when idle for pcb->keep_idle milliseconds */
- #define TCP_KEEPIDLE 0x03 /* set pcb->keep_idle - Same as TCP_KEEPALIVE, but use seconds for get/setsockopt */
- #define TCP_KEEPINTVL 0x04 /* set pcb->keep_intvl - Use seconds for get/setsockopt */
- #define TCP_KEEPCNT 0x05 /* set pcb->keep_cnt - Use number of probes sent for get/setsockopt */
- static uint8_t network_check_ip_same(luat_ip_addr_t *ip1, luat_ip_addr_t *ip2)
- {
- if (ip1->is_ipv6 != ip2->is_ipv6)
- {
- return 0;
- }
- if (ip1->is_ipv6)
- {
- return !memcmp(ip1->ipv6_u8_addr, ip2->ipv6_u8_addr, 16);
- }
- else
- {
- return (ip1->ipv4 == ip2->ipv4);
- }
- }
- static int network_base_tx(network_ctrl_t *ctrl, const uint8_t *data, uint32_t len, int flags, luat_ip_addr_t *remote_ip, uint16_t remote_port)
- {
- int result = -1;
- if (ctrl->is_tcp)
- {
- result = network_socket_send(ctrl, data, len, flags, NULL, 0);
- }
- else
- {
- if (remote_ip)
- {
- result = network_socket_send(ctrl, data, len, flags, remote_ip, remote_port);
- }
- else
- {
- result = network_socket_send(ctrl, data, len, flags, ctrl->online_ip, ctrl->remote_port);
- }
- }
- if (result >= 0)
- {
- ctrl->tx_size += len;
- }
- else
- {
- ctrl->need_close = 1;
- }
- return result;
- }
- static int32_t tls_shorttimeout(void *data, void *param)
- {
- network_ctrl_t *ctrl = (network_ctrl_t *)param;
- if (!ctrl->tls_mode)
- {
- platform_stop_timer(ctrl->tls_long_timer);
- return 0;
- }
- if (0 == ctrl->tls_timer_state)
- {
- ctrl->tls_timer_state = 1;
- }
- return 0;
- }
- static int32_t tls_longtimeout(void *data, void *param)
- {
- network_ctrl_t *ctrl = (network_ctrl_t *)param;
- platform_stop_timer(ctrl->tls_short_timer);
- if (!ctrl->tls_mode)
- {
- return 0;
- }
- ctrl->tls_timer_state = 2;
- return 0;
- }
- static void tls_settimer( void *data, uint32_t int_ms, uint32_t fin_ms )
- {
- network_ctrl_t *ctrl = (network_ctrl_t *)data;
- if (!ctrl->tls_mode)
- {
- return;
- }
- if (!fin_ms)
- {
- platform_stop_timer(ctrl->tls_short_timer);
- platform_stop_timer(ctrl->tls_long_timer);
- ctrl->tls_timer_state = -1;
- return ;
- }
- platform_start_timer(ctrl->tls_short_timer, int_ms, 0);
- platform_start_timer(ctrl->tls_long_timer, fin_ms, 0);
- ctrl->tls_timer_state = 0;
- }
- static int tls_gettimer( void *data )
- {
- network_ctrl_t *ctrl = (network_ctrl_t *)data;
- if (!ctrl->tls_mode)
- {
- return -ERROR_PARAM_INVALID;
- }
- #if MBEDTLS_VERSION_NUMBER >= 0x03000000
- if(!mbedtls_ssl_is_handshake_over(ctrl->ssl))
- #else
- if (ctrl->ssl->state != MBEDTLS_SSL_HANDSHAKE_OVER)
- #endif
- {
- return ctrl->tls_timer_state;
- }
- else
- {
- return 0;
- }
- }
- static void tls_dbg(void *data, int level,
- const char *file, int line,
- const char *str)
- {
- DBG_Printf("%s %d:%s", file, line, str);
- }
- static int tls_send(void *ctx, const unsigned char *buf, size_t len )
- {
- network_ctrl_t *ctrl = (network_ctrl_t *)ctx;
- if (!ctrl->tls_mode)
- {
- return -ERROR_PERMISSION_DENIED;
- }
- if (network_base_tx(ctrl, buf, len, 0, NULL, 0) != len)
- {
- return -0x004E;
- }
- else
- {
- return len;
- }
- }
- static int tls_recv(void *ctx, unsigned char *buf, size_t len )
- {
- #ifdef LUAT_USE_TLS
- network_ctrl_t *ctrl = (network_ctrl_t *)ctx;
- luat_ip_addr_t remote_ip;
- uint16_t remote_port;
- int result = -1;
- if (!ctrl->tls_mode)
- {
- return -1;
- }
- TLS_RECV:
- result = network_socket_receive(ctrl, buf, len, 0, &remote_ip, &remote_port);
- if (result < 0)
- {
- return -0x004C;
- }
- if (result > 0)
- {
- if (!ctrl->is_tcp)
- {
- if ((remote_port == ctrl->remote_port) && network_check_ip_same(&remote_ip, ctrl->online_ip))
- {
- goto TLS_RECV;
- }
- }
- return result;
- }
- return MBEDTLS_ERR_SSL_WANT_READ;
- #else
- return -1;
- #endif
- }
- static int network_get_host_by_name(network_ctrl_t *ctrl)
- {
- ctrl->remote_ip.is_ipv6 = 0xff;
- if (network_string_is_ipv4(ctrl->domain_name, ctrl->domain_name_len))
- {
- ctrl->remote_ip.is_ipv6 = 0;
- ctrl->remote_ip.ipv4 = network_string_to_ipv4(ctrl->domain_name, ctrl->domain_name_len);
- }
- else
- {
- char *name = zalloc(ctrl->domain_name_len + 1);
- memcpy(name, ctrl->domain_name, ctrl->domain_name_len);
- network_string_to_ipv6(name, &ctrl->remote_ip);
- free(name);
- }
- if (ctrl->remote_ip.is_ipv6 != 0xff)
- {
- return 0;
- }
- else
- {
- return -1;
- }
- }
- static void network_update_dns_cache(network_ctrl_t *ctrl)
- {
- }
- static void network_get_dns_cache(network_ctrl_t *ctrl)
- {
- }
- static int network_base_connect(network_ctrl_t *ctrl, luat_ip_addr_t *remote_ip)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- if (ctrl->socket_id >= 0)
- {
- network_force_close_socket(ctrl);
- }
- if (remote_ip)
- {
- if (network_create_soceket(ctrl, remote_ip->is_ipv6) < 0)
- {
- network_clean_invaild_socket(ctrl->adapter_index);
- if (network_create_soceket(ctrl, remote_ip->is_ipv6) < 0)
- {
- return -1;
- }
- }
- if (adapter->opt->is_posix)
- {
- network_setsockopt(ctrl, SOL_SOCKET, SO_KEEPALIVE, (void *)&ctrl->tcp_keep_alive, sizeof(ctrl->tcp_keep_alive));
- if (ctrl->tcp_keep_alive)
- {
- network_setsockopt(ctrl, IPPROTO_TCP, TCP_KEEPIDLE, (void*)&ctrl->tcp_keep_idle, sizeof(ctrl->tcp_keep_idle));
- network_setsockopt(ctrl, IPPROTO_TCP, TCP_KEEPINTVL, (void *)&ctrl->tcp_keep_interval, sizeof(ctrl->tcp_keep_interval));
- network_setsockopt(ctrl, IPPROTO_TCP, TCP_KEEPCNT, (void *)&ctrl->tcp_keep_cnt, sizeof(ctrl->tcp_keep_cnt));
- }
- }
- else
- {
- network_user_cmd(ctrl, NW_CMD_AUTO_HEART_TIME, ctrl->tcp_keep_idle);
- }
- return network_socket_connect(ctrl, remote_ip);
- }
- else
- {
- luat_ip_addr_t local_ip, net_mask, gate_way;
- network_get_local_ip_info(ctrl, &local_ip, &net_mask, &gate_way);
- if (network_create_soceket(ctrl, local_ip.is_ipv6) < 0)
- {
- network_clean_invaild_socket(ctrl->adapter_index);
- if (network_create_soceket(ctrl, local_ip.is_ipv6) < 0)
- {
- return -1;
- }
- }
- return network_socket_listen(ctrl);
- }
- }
- static int network_prepare_connect(network_ctrl_t *ctrl)
- {
- if (ctrl->remote_ip.is_ipv6 != 0xff)
- {
- ;
- }
- else if (ctrl->domain_name)
- {
- if (network_get_host_by_name(ctrl))
- {
- if (network_dns(ctrl))
- {
- network_socket_force_close(ctrl);
- return -1;
- }
- ctrl->state = NW_STATE_WAIT_DNS;
- return 0;
- }
- }
- else
- {
- return -1;
- }
- if (network_base_connect(ctrl, &ctrl->remote_ip))
- {
- network_socket_force_close(ctrl);
- return -1;
- }
- ctrl->state = NW_STATE_CONNECTING;
- return 0;
- }
- static int network_state_link_off(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- if (EV_NW_STATE == event->ID)
- {
- if (event->Param2)
- {
- ctrl->state = NW_STATE_OFF_LINE;
- if (NW_WAIT_LINK_UP == ctrl->wait_target_state)
- {
- return 0;
- }
- else if (NW_WAIT_ON_LINE == ctrl->wait_target_state)
- {
- if (ctrl->is_server_mode)
- {
- if (network_base_connect(ctrl, NULL))
- {
- return -1;
- }
- ctrl->state = NW_STATE_CONNECTING;
- }
- else
- {
- if (network_prepare_connect(ctrl))
- {
- return -1;
- }
- }
- return 1;
- }
- }
- }
- return 1;
- }
- static int network_state_off_line(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- return 1;
- }
- static int network_state_wait_dns(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- if ((ctrl->need_close) || ctrl->wait_target_state != NW_WAIT_ON_LINE) return -1;
- switch(event->ID)
- {
- case EV_NW_RESET:
- case EV_NW_SOCKET_ERROR:
- return -1;
- case EV_NW_STATE:
- if (!event->Param2)
- {
- return -1;
- }
- break;
- case EV_NW_DNS_RESULT:
- if (event->Param1)
- {
- //更新dns cache
- ctrl->dns_ip = event->Param2;
- ctrl->dns_ip_nums = event->Param1;
- network_update_dns_cache(ctrl);
- }
- else
- {
- ctrl->dns_ip_nums = 0;
- network_get_dns_cache(ctrl);
- if (!ctrl->dns_ip_nums)
- {
- return -1;
- }
- }
- ctrl->dns_ip_cnt = 0;
- if (network_base_connect(ctrl, &ctrl->dns_ip[ctrl->dns_ip_cnt].ip))
- {
- network_socket_force_close(ctrl);
- return -1;
- }
- else
- {
- ctrl->state = NW_STATE_CONNECTING;
- return 1;
- }
- default:
- return 1;
- }
- return -1; // TODO 返回值不确定
- }
- static int network_state_connecting(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- if ((ctrl->need_close) || (ctrl->wait_target_state != NW_WAIT_ON_LINE)) return -1;
- switch(event->ID)
- {
- case EV_NW_RESET:
- return -1;
- case EV_NW_SOCKET_ERROR:
- case EV_NW_SOCKET_REMOTE_CLOSE:
- case EV_NW_SOCKET_CLOSE_OK:
- if (ctrl->remote_ip.is_ipv6 != 0xff)
- {
- return -1;
- }
- DBG("dns ip %d no connect!", ctrl->dns_ip_cnt);
- ctrl->dns_ip_cnt++;
- if (ctrl->dns_ip_cnt >= ctrl->dns_ip_nums)
- {
- return -1;
- }
- if (network_base_connect(ctrl, &ctrl->dns_ip[ctrl->dns_ip_cnt].ip))
- {
- network_socket_force_close(ctrl);
- return -1;
- }
- else
- {
- ctrl->state = NW_STATE_CONNECTING;
- return 1;
- }
- break;
- case EV_NW_STATE:
- if (!event->Param2)
- {
- return -1;
- }
- break;
- case EV_NW_SOCKET_LISTEN:
- if (ctrl->is_server_mode)
- {
- ctrl->state = NW_STATE_LISTEN;
- return 1;
- }
- break;
- case EV_NW_SOCKET_CONNECT_OK:
- #ifdef LUAT_USE_TLS
- if (ctrl->tls_mode)
- {
- mbedtls_ssl_free(ctrl->ssl);
- memset(ctrl->ssl, 0, sizeof(mbedtls_ssl_context));
- mbedtls_ssl_setup(ctrl->ssl, ctrl->config);
- ctrl->ssl->f_set_timer = tls_settimer;
- ctrl->ssl->f_get_timer = tls_gettimer;
- ctrl->ssl->p_timer = ctrl;
- ctrl->ssl->p_bio = ctrl;
- ctrl->ssl->f_send = tls_send;
- ctrl->ssl->f_recv = tls_recv;
- // add by wendal
- // cloudflare的https需要设置hostname才能访问
- if (ctrl->domain_name_len > 0 && ctrl->domain_name_len < 256) {
- char host[257] = {0};
- memcpy(host, ctrl->domain_name, ctrl->domain_name_len);
- mbedtls_ssl_set_hostname(ctrl->ssl, host);
- //LLOGD("CALL mbedtls_ssl_set_hostname(%s)", host);
- }
- else {
- //LLOGD("skip mbedtls_ssl_set_hostname");
- }
- ctrl->state = NW_STATE_SHAKEHAND;
- do
- {
- int result = mbedtls_ssl_handshake_step( ctrl->ssl );
- switch(result)
- {
- case MBEDTLS_ERR_SSL_WANT_READ:
- return 1;
- case 0:
- break;
- default:
- DBG_ERR("0x%x, %d", -result, ctrl->ssl->state);
- return -1;
- }
- }while(ctrl->ssl->state != MBEDTLS_SSL_HANDSHAKE_OVER);
- return 0;
- }
- else
- #endif
- {
- ctrl->state = NW_STATE_ONLINE;
- return 0;
- }
- default:
- return 1;
- }
- return 1; // TODO 返回值不确定
- }
- static int network_state_shakehand(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- if ((ctrl->need_close) || ((ctrl->wait_target_state != NW_WAIT_ON_LINE) && (ctrl->wait_target_state != NW_WAIT_TX_OK))) return -1;
- switch(event->ID)
- {
- case EV_NW_RESET:
- case EV_NW_SOCKET_ERROR:
- case EV_NW_SOCKET_REMOTE_CLOSE:
- case EV_NW_SOCKET_CLOSE_OK:
- ctrl->need_close = 1;
- return -1;
- case EV_NW_STATE:
- if (!event->Param2)
- {
- ctrl->need_close = 1;
- return -1;
- }
- break;
- case EV_NW_SOCKET_TX_OK:
- ctrl->ack_size += event->Param2;
- if (ctrl->is_debug)
- {
- DBG("%llu,%llu",ctrl->tx_size, ctrl->ack_size);
- }
- break;
- #ifdef LUAT_USE_TLS
- case EV_NW_SOCKET_RX_NEW:
- do
- {
- int result = mbedtls_ssl_handshake_step( ctrl->ssl );
- switch(result)
- {
- case MBEDTLS_ERR_SSL_WANT_READ:
- return 1;
- case 0:
- break;
- default:
- DBG_ERR("0x%x, %d", -result, ctrl->ssl->state);
- ctrl->need_close = 1;
- return -1;
- }
- }while(ctrl->ssl->state != MBEDTLS_SSL_HANDSHAKE_OVER);
- ctrl->state = NW_STATE_ONLINE;
- if (NW_WAIT_TX_OK == ctrl->wait_target_state)
- {
- if (!ctrl->cache_data)
- {
- ctrl->need_close = 1;
- return -1;
- }
- int result = mbedtls_ssl_write(ctrl->ssl, ctrl->cache_data, ctrl->cache_len);
- free(ctrl->cache_data);
- ctrl->cache_data = NULL;
- ctrl->cache_len = 0;
- if (result < 0)
- {
- DBG("%08x", -result);
- ctrl->need_close = 1;
- return -1;
- }
- return 1;
- }
- return 0;
- #endif
- case EV_NW_SOCKET_CONNECT_OK:
- DBG_ERR("!");
- return 1;
- default:
- return 1;
- }
- return 1;
- }
- static int network_state_on_line(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- if ((ctrl->need_close) || NW_WAIT_OFF_LINE == ctrl->wait_target_state)
- {
- return -1;
- }
- switch(event->ID)
- {
- case EV_NW_RESET:
- case EV_NW_SOCKET_ERROR:
- case EV_NW_SOCKET_REMOTE_CLOSE:
- case EV_NW_SOCKET_CLOSE_OK:
- ctrl->need_close = 1;
- return -1;
- case EV_NW_STATE:
- if (!event->Param2)
- {
- ctrl->need_close = 1;
- return -1;
- }
- break;
- case EV_NW_SOCKET_TX_OK:
- ctrl->ack_size += event->Param2;
- if (NW_WAIT_TX_OK == ctrl->wait_target_state)
- {
- if (ctrl->ack_size == ctrl->tx_size)
- {
- return 0;
- }
- }
- break;
- case EV_NW_SOCKET_RX_NEW:
- ctrl->new_rx_flag = 1;
- if (NW_WAIT_TX_OK != ctrl->wait_target_state)
- {
- return 0;
- }
- break;
- default:
- return 1;
- }
- return 1;
- }
- static int network_state_listen(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- if ((ctrl->need_close) || NW_WAIT_OFF_LINE == ctrl->wait_target_state)
- {
- return -1;
- }
- switch(event->ID)
- {
- case EV_NW_RESET:
- case EV_NW_SOCKET_ERROR:
- case EV_NW_SOCKET_REMOTE_CLOSE:
- case EV_NW_SOCKET_CLOSE_OK:
- ctrl->need_close = 1;
- return -1;
- case EV_NW_STATE:
- if (!event->Param2)
- {
- ctrl->need_close = 1;
- return -1;
- }
- break;
- case EV_NW_SOCKET_NEW_CONNECT:
- case EV_NW_SOCKET_CONNECT_OK:
- ctrl->state = NW_STATE_ONLINE;
- return 0;
- default:
- return 1;
- }
- return 1;
- }
- static int network_state_disconnecting(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- if (ctrl->wait_target_state != NW_WAIT_OFF_LINE)
- {
- return -1;
- }
- switch(event->ID)
- {
- case EV_NW_RESET:
- case EV_NW_SOCKET_ERROR:
- case EV_NW_SOCKET_REMOTE_CLOSE:
- case EV_NW_SOCKET_CLOSE_OK:
- network_force_close_socket(ctrl);
- ctrl->state = NW_STATE_OFF_LINE;
- ctrl->socket_id = -1;
- return 0;
- case EV_NW_STATE:
- if (!event->Param2)
- {
- return -1;
- }
- else
- {
- network_force_close_socket(ctrl);
- ctrl->state = NW_STATE_OFF_LINE;
- ctrl->socket_id = -1;
- }
- break;
- default:
- return 1;
- }
- return -1;
- }
- typedef int (*network_state_fun)(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter);
- static network_state_fun network_state_fun_list[]=
- {
- network_state_link_off,
- network_state_off_line,
- network_state_wait_dns,
- network_state_connecting,
- network_state_shakehand,
- network_state_on_line,
- network_state_listen,
- network_state_disconnecting,
- };
- static void network_default_statemachine(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
- {
- int result = 0;
- if (ctrl->state > NW_STATE_DISCONNECTING)
- {
- ctrl->state = NW_STATE_LINK_OFF;
- event->Param1 = -1;
- network_force_close_socket(ctrl);
- event->ID = ctrl->wait_target_state + EV_NW_RESULT_BASE;
- }
- else
- {
- result = network_state_fun_list[ctrl->state](ctrl, event, adapter);
- if (result > 0)
- {
- return ;
- }
- event->ID = (ctrl->wait_target_state?ctrl->wait_target_state:NW_WAIT_EVENT) + EV_NW_RESULT_BASE;
- event->Param1 = result;
- }
- if ((ctrl->state != NW_STATE_LISTEN) || (result < 0))
- {
- ctrl->wait_target_state = NW_WAIT_NONE;
- }
- if (ctrl->task_handle)
- {
- platform_send_event(ctrl->task_handle, event->ID, event->Param1, event->Param2, event->Param3);
- }
- else if (ctrl->user_callback)
- {
- ctrl->user_callback(event, ctrl->user_data);
- }
- }
- static int32_t network_default_socket_callback(void *data, void *param)
- {
- OS_EVENT *event = (OS_EVENT *)data;
- OS_EVENT temp_event;
- luat_network_cb_param_t *cb_param = (luat_network_cb_param_t *)param;
- network_adapter_t *adapter =(network_adapter_t *)(cb_param->param);
- int i = 0;
- network_ctrl_t *ctrl = (network_ctrl_t *)event->Param3;
- if (event->ID > EV_NW_TIMEOUT)
- {
- if (ctrl && ((ctrl->tag == cb_param->tag) || (event->ID == EV_NW_DNS_RESULT)))
- {
- if (ctrl->auto_mode)
- {
- DBG("socket %d,%s,%s,%s", ctrl->socket_id, network_ctrl_event_id_string(event->ID),
- network_ctrl_state_string(ctrl->state),
- network_ctrl_wait_state_string(ctrl->wait_target_state));
- network_default_statemachine(ctrl, event, adapter);
- DBG("%s,%s",
- network_ctrl_state_string(ctrl->state),
- network_ctrl_wait_state_string(ctrl->wait_target_state));
- }
- else if (ctrl->task_handle)
- {
- platform_send_event(ctrl->task_handle, event->ID, event->Param1, event->Param2, event->Param3);
- }
- else if (ctrl->user_callback)
- {
- ctrl->user_callback(event, ctrl->user_data);
- }
- }
- else
- {
- DBG_ERR("cb ctrl invaild %x", ctrl);
- DBG_HexPrintf(&ctrl->tag, 8);
- DBG_HexPrintf(&cb_param->tag, 8);
- }
- }
- else
- {
- for (i = 0; i < adapter->opt->max_socket_num; i++)
- {
- temp_event = *event;
- if (adapter->ctrl_busy[i])
- {
- ctrl = &adapter->ctrl_table[i];
- if (ctrl->auto_mode)
- {
- DBG("socket %d,%s,%s,%s", ctrl->socket_id, network_ctrl_event_id_string(event->ID),
- network_ctrl_state_string(ctrl->state),
- network_ctrl_wait_state_string(ctrl->wait_target_state));
- network_default_statemachine(ctrl, &temp_event, adapter);
- DBG("%s,%s",
- network_ctrl_state_string(ctrl->state),
- network_ctrl_wait_state_string(ctrl->wait_target_state));
- }
- else if (ctrl->task_handle)
- {
- platform_send_event(ctrl->task_handle, event->ID, event->Param1, event->Param2, event->Param3);
- }
- else if (ctrl->user_callback)
- {
- ctrl->user_callback(&temp_event, ctrl->user_data);
- }
- }
- }
- }
- return 0;
- }
- static int32_t network_default_timer_callback(void *data, void *param)
- {
- platform_send_event(param, EV_NW_TIMEOUT, 0, 0, 0);
- return 0;
- }
- uint8_t network_string_is_ipv4(const char *string, uint32_t len)
- {
- int i;
- for(i = 0; i < len; i++)
- {
- if (!isdigit((int)string[i]) && (string[i] != '.'))
- {
- return 0;
- }
- }
- return 1;
- }
- uint32_t network_string_to_ipv4(const char *string, uint32_t len)
- {
- int i;
- int8_t Buf[4][4];
- CmdParam CP;
- PV_Union uIP;
- char temp[32];
- memset(Buf, 0, sizeof(Buf));
- CP.param_max_len = 4;
- CP.param_max_num = 4;
- CP.param_num = 0;
- CP.param_str = (int8_t *)Buf;
- memcpy(temp, string, len);
- temp[len] = 0;
- CmdParseParam((int8_t*)temp, &CP, '.');
- for(i = 0; i < 4; i++)
- {
- uIP.u8[i] = strtol((char *)Buf[i], NULL, 10);
- }
- // DBG("%d.%d.%d.%d", uIP.u8[0], uIP.u8[1], uIP.u8[2], uIP.u8[3]);
- return uIP.u32;
- }
- #define SWAP(x) ((((x) & (uint32_t)0x000000ffUL) << 24) | \
- (((x) & (uint32_t)0x0000ff00UL) << 8) | \
- (((x) & (uint32_t)0x00ff0000UL) >> 8) | \
- (((x) & (uint32_t)0xff000000UL) >> 24))
- int network_string_to_ipv6(const char *string, luat_ip_addr_t *ip_addr)
- {
- uint32_t addr_index, zero_blocks, current_block_index, current_block_value;
- const char *s;
- ip_addr->is_ipv6 = 0xff;
- /* Count the number of colons, to count the number of blocks in a "::" sequence
- zero_blocks may be 1 even if there are no :: sequences */
- zero_blocks = 8;
- for (s = string; *s != 0; s++) {
- if (*s == ':') {
- zero_blocks--;
- } else if (!isxdigit((int)*s)) {
- break;
- }
- }
- /* parse each block */
- addr_index = 0;
- current_block_index = 0;
- current_block_value = 0;
- for (s = string; *s != 0; s++) {
- if (*s == ':') {
- if (current_block_index & 0x1) {
- ip_addr->ipv6_u32_addr[addr_index++] |= current_block_value;
- }
- else {
- ip_addr->ipv6_u32_addr[addr_index] = current_block_value << 16;
- }
- current_block_index++;
- current_block_value = 0;
- if (current_block_index > 7) {
- /* address too long! */
- return 0;
- }
- if (s[1] == ':') {
- if (s[2] == ':') {
- /* invalid format: three successive colons */
- return 0;
- }
- s++;
- /* "::" found, set zeros */
- while (zero_blocks > 0) {
- zero_blocks--;
- if (current_block_index & 0x1) {
- addr_index++;
- } else {
- ip_addr->ipv6_u32_addr[addr_index] = 0;
- }
- current_block_index++;
- if (current_block_index > 7) {
- /* address too long! */
- return 0;
- }
- }
- }
- } else if (isxdigit((int)*s)) {
- /* add current digit */
- current_block_value = (current_block_value << 4) +
- (isxdigit((int)*s) ? (uint32_t)(*s - '0') :
- (uint32_t)(10 + (isxdigit((int)*s) ? *s - 'a' : *s - 'A')));
- } else {
- /* unexpected digit, space? CRLF? */
- break;
- }
- }
- if (current_block_index & 0x1) {
- ip_addr->ipv6_u32_addr[addr_index++] |= current_block_value;
- }
- else {
- ip_addr->ipv6_u32_addr[addr_index] = current_block_value << 16;
- }
- /* convert to network byte order. */
- for (addr_index = 0; addr_index < 4; addr_index++) {
- ip_addr->ipv6_u32_addr[addr_index] = SWAP(ip_addr->ipv6_u32_addr[addr_index]);
- }
- if (current_block_index != 7) {
- return 0;
- }
- ip_addr->is_ipv6 = 1;
- return 1;
- }
- int network_get_last_register_adapter(void)
- {
- return prv_network.last_adapter_index;
- }
- int network_register_adapter(uint8_t adapter_index, network_adapter_info *info, void *user_data)
- {
- prv_adapter_table[adapter_index].opt = info;
- prv_adapter_table[adapter_index].user_data = user_data;
- info->socket_set_callback(network_default_socket_callback, &prv_adapter_table[adapter_index], user_data);
- prv_adapter_table[adapter_index].ctrl_table = zalloc((info->max_socket_num) * sizeof(network_ctrl_t));
- prv_adapter_table[adapter_index].ctrl_busy = zalloc(info->max_socket_num);
- prv_adapter_table[adapter_index].port = 60000;
- if (!prv_network.is_init)
- {
- INIT_LLIST_HEAD(&prv_network.dns_cache_head);
- prv_network.is_init = 0;
- }
- prv_network.last_adapter_index = adapter_index;
- return 0;
- }
- void network_set_dns_server(uint8_t adapter_index, uint8_t server_index, luat_ip_addr_t *ip)
- {
- network_adapter_t *adapter = &prv_adapter_table[adapter_index];
- adapter->opt->set_dns_server(server_index, ip, adapter->user_data);
- }
- /*
- * 申请一个network_ctrl
- */
- network_ctrl_t *network_alloc_ctrl(uint8_t adapter_index)
- {
- int i;
- network_ctrl_t *ctrl = NULL;
- network_adapter_t *adapter = &prv_adapter_table[adapter_index];
- OS_LOCK;
- for (i = 0; i < adapter->opt->max_socket_num; i++)
- {
- if (!adapter->ctrl_busy[i])
- {
- adapter->ctrl_busy[i] = 1;
- ctrl = &adapter->ctrl_table[i];
- ctrl->adapter_index = adapter_index;
- break;
- }
- }
- if (i >= adapter->opt->max_socket_num) {DBG_ERR("adapter no more ctrl!");}
- OS_UNLOCK;
- return ctrl;
- }
- /*
- * 归还一个network_ctrl
- */
- void network_release_ctrl(network_ctrl_t *ctrl)
- {
- int i;
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- NW_UNLOCK;
- OS_LOCK;
- for (i = 0; i < adapter->opt->max_socket_num; i++)
- {
- if (&adapter->ctrl_table[i] == ctrl)
- {
- network_deinit_tls(ctrl);
- if (ctrl->timer)
- {
- platform_release_timer(ctrl->timer);
- }
- if (ctrl->cache_data)
- {
- free(ctrl->cache_data);
- ctrl->cache_data = NULL;
- }
- if (ctrl->dns_ip)
- {
- free(ctrl->dns_ip);
- ctrl->dns_ip = NULL;
- }
- adapter->ctrl_busy[i] = 0;
- platform_release_mutex(ctrl->mutex);
- ctrl->mutex = NULL;
- break;
- }
- }
- OS_UNLOCK;
- // NW_UNLOCK;
- if (i >= adapter->opt->max_socket_num) {DBG_ERR("adapter index maybe error!, %d, %x", ctrl->adapter_index, ctrl);}
- }
- void network_init_ctrl(network_ctrl_t *ctrl, HANDLE task_handle, CBFuncEx_t callback, void *param)
- {
- uint8_t adapter_index = ctrl->adapter_index;
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- if (ctrl->dns_ip)
- {
- free(ctrl->dns_ip);
- ctrl->dns_ip = NULL;
- }
- if (ctrl->cache_data)
- {
- free(ctrl->cache_data);
- ctrl->cache_data = NULL;
- }
- HANDLE sem = ctrl->mutex;
- memset(ctrl, 0, sizeof(network_ctrl_t));
- ctrl->adapter_index = adapter_index;
- ctrl->task_handle = task_handle;
- ctrl->user_callback = callback;
- ctrl->user_data = param;
- ctrl->socket_id = -1;
- ctrl->socket_param = ctrl;
- ctrl->remote_ip.is_ipv6 = 0xff;
- ctrl->mutex = sem;
- if (task_handle)
- {
- ctrl->timer = platform_create_timer(network_default_timer_callback, task_handle, NULL);
- }
- if (!ctrl->mutex)
- {
- ctrl->mutex = platform_create_mutex();
- }
- }
- void network_set_base_mode(network_ctrl_t *ctrl, uint8_t is_tcp, uint32_t tcp_timeout_ms, uint8_t keep_alive, uint32_t keep_idle, uint8_t keep_interval, uint8_t keep_cnt)
- {
- ctrl->is_tcp = is_tcp;
- ctrl->tcp_keep_alive = keep_alive;
- ctrl->tcp_keep_idle = keep_idle;
- ctrl->tcp_keep_interval = keep_interval;
- ctrl->tcp_keep_cnt = keep_cnt;
- ctrl->tcp_timeout_ms = tcp_timeout_ms;
- }
- void network_connect_ipv6_domain(network_ctrl_t *ctrl, uint8_t onoff)
- {
- ctrl->domain_ipv6 = onoff;
- }
- int network_set_local_port(network_ctrl_t *ctrl, uint16_t local_port)
- {
- int i;
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- if (local_port)
- {
- NW_LOCK;
- for (i = 0; i < adapter->opt->max_socket_num; i++)
- {
- if (&adapter->ctrl_table[i] != ctrl)
- {
- if (adapter->ctrl_table[i].local_port == local_port)
- {
- NW_UNLOCK;
- return -1;
- }
- }
- }
- ctrl->local_port = local_port;
- NW_UNLOCK;
- return 0;
- }
- else
- {
- NW_LOCK;
- adapter->port++;
- if (adapter->port < 60000)
- {
- adapter->port = 60000;
- }
- ctrl->local_port = adapter->port;
- NW_UNLOCK;
- return 0;
- }
- }
- uint8_t network_check_ready(network_ctrl_t *ctrl)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- return adapter->opt->check_ready(adapter->user_data);
- }
- int network_create_soceket(network_ctrl_t *ctrl, uint8_t is_ipv6)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- ctrl->socket_id = adapter->opt->create_soceket(ctrl->is_tcp, &ctrl->tag, ctrl->socket_param, is_ipv6, adapter->user_data);
- if (ctrl->socket_id < 0)
- {
- ctrl->tag = 0;
- return -1;
- }
- return 0;
- }
- int network_socket_connect(network_ctrl_t *ctrl, luat_ip_addr_t *remote_ip)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- ctrl->is_server_mode = 0;
- ctrl->online_ip = remote_ip;
- return adapter->opt->socket_connect(ctrl->socket_id, ctrl->tag, ctrl->local_port, remote_ip, ctrl->remote_port, adapter->user_data);
- }
- int network_socket_listen(network_ctrl_t *ctrl)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- ctrl->is_server_mode = 1;
- return adapter->opt->socket_listen(ctrl->socket_id, ctrl->tag, ctrl->local_port, adapter->user_data);
- }
- uint8_t network_accept_enable(network_ctrl_t *ctrl)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- return !(adapter->opt->no_accept);
- }
- //作为server接受一个client
- //成功返回0,失败 < 0
- int network_socket_accept(network_ctrl_t *ctrl, network_ctrl_t *accept_ctrl)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- if (adapter->opt->no_accept)
- {
- // DBG("%x,%d,%llu,%x,%x,%x",adapter->opt->socket_accept, ctrl->socket_id, ctrl->tag, &ctrl->remote_ip, &ctrl->remote_port, adapter->user_data);
- adapter->opt->socket_accept(ctrl->socket_id, ctrl->tag, &ctrl->remote_ip, &ctrl->remote_port, adapter->user_data);
- return 0;
- }
- accept_ctrl->socket_id = adapter->opt->socket_accept(ctrl->socket_id, ctrl->tag, &accept_ctrl->remote_ip, &accept_ctrl->remote_port, adapter->user_data);
- if (accept_ctrl->socket_id < 0)
- {
- return -1;
- }
- else
- {
- accept_ctrl->is_tcp = ctrl->is_tcp;
- accept_ctrl->tcp_keep_alive = ctrl->tcp_keep_alive;
- accept_ctrl->tcp_keep_idle = ctrl->tcp_keep_idle;
- accept_ctrl->tcp_keep_interval = ctrl->tcp_keep_interval;
- accept_ctrl->tcp_keep_cnt = ctrl->tcp_keep_cnt;
- accept_ctrl->tcp_timeout_ms = ctrl->tcp_timeout_ms;
- accept_ctrl->local_port = ctrl->local_port;
- accept_ctrl->state = NW_STATE_ONLINE;
- return 0;
- }
- }
- //主动断开一个tcp连接,需要走完整个tcp流程,用户需要接收到close ok回调才能确认彻底断开
- //成功返回0,失败 < 0
- int network_socket_disconnect(network_ctrl_t *ctrl)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- if (ctrl->socket_id >= 0)
- {
- return adapter->opt->socket_disconnect(ctrl->socket_id, ctrl->tag, adapter->user_data);
- }
- return 0;
- }
- //释放掉socket的控制权
- //成功返回0,失败 < 0
- int network_socket_close(network_ctrl_t *ctrl)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- if (ctrl->socket_id >= 0)
- {
- return adapter->opt->socket_close(ctrl->socket_id, ctrl->tag, adapter->user_data);
- }
- return 0;
- }
- //强行释放掉socket的控制权
- //成功返回0,失败 < 0
- int network_socket_force_close(network_ctrl_t *ctrl)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- if (ctrl->socket_id >= 0)
- {
- adapter->opt->socket_force_close(ctrl->socket_id, adapter->user_data);
- }
- ctrl->socket_id = -1;
- return 0;
- }
- //tcp时,不需要remote_ip和remote_port,如果buf为NULL,则返回当前缓存区的数据量,当返回值小于len时说明已经读完了
- //udp时,只返回1个block数据,需要多次读直到没有数据为止
- //成功返回实际读取的值,失败 < 0
- int network_socket_receive(network_ctrl_t *ctrl,uint8_t *buf, uint32_t len, int flags, luat_ip_addr_t *remote_ip, uint16_t *remote_port)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- return adapter->opt->socket_receive(ctrl->socket_id, ctrl->tag, buf, len, flags, remote_ip, remote_port, adapter->user_data);
- }
- //tcp时,不需要remote_ip和remote_port
- //成功返回0,失败 < 0
- int network_socket_send(network_ctrl_t *ctrl,const uint8_t *buf, uint32_t len, int flags, luat_ip_addr_t *remote_ip, uint16_t remote_port)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- return adapter->opt->socket_send(ctrl->socket_id, ctrl->tag, buf, len, flags, remote_ip, remote_port, adapter->user_data);
- }
- int network_getsockopt(network_ctrl_t *ctrl, int level, int optname, void *optval, uint32_t *optlen)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- return adapter->opt->getsockopt(ctrl->socket_id, ctrl->tag, level, optname, optval, optlen, adapter->user_data);
- }
- int network_setsockopt(network_ctrl_t *ctrl, int level, int optname, const void *optval, uint32_t optlen)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- return adapter->opt->setsockopt(ctrl->socket_id, ctrl->tag, level, optname, optval, optlen, adapter->user_data);
- }
- //非posix的socket,用这个根据实际硬件设置参数
- int network_user_cmd(network_ctrl_t *ctrl, uint32_t cmd, uint32_t value)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- return adapter->opt->user_cmd(ctrl->socket_id, ctrl->tag, cmd, value, adapter->user_data);
- }
- int network_dns(network_ctrl_t *ctrl)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- return adapter->opt->dns(ctrl->domain_name, ctrl->domain_name_len, ctrl, adapter->user_data);
- }
- int network_get_local_ip_info(network_ctrl_t *ctrl, luat_ip_addr_t *ip, luat_ip_addr_t *submask, luat_ip_addr_t *gateway)
- {
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- return adapter->opt->get_local_ip_info(ip, submask, gateway, adapter->user_data);
- }
- void network_force_close_socket(network_ctrl_t *ctrl)
- {
- #ifdef LUAT_USE_TLS
- if (ctrl->tls_mode)
- {
- mbedtls_ssl_free(ctrl->ssl);
- }
- #endif
- if (network_socket_close(ctrl))
- {
- network_clean_invaild_socket(ctrl->adapter_index);
- network_socket_force_close(ctrl);
- }
- ctrl->need_close = 0;
- ctrl->socket_id = -1;
- if (ctrl->dns_ip)
- {
- free(ctrl->dns_ip);
- ctrl->dns_ip = NULL;
- }
- ctrl->dns_ip_cnt = 0;
- ctrl->dns_ip_nums = 0;
- }
- void network_clean_invaild_socket(uint8_t adapter_index)
- {
- int i = 0;
- int *list = NULL;
- network_adapter_t *adapter = &prv_adapter_table[adapter_index];
- network_ctrl_t *ctrl = NULL;
- list = malloc(adapter->opt->max_socket_num * sizeof(int));
- NW_LOCK;
- for (i = 0; i < adapter->opt->max_socket_num; i++)
- {
- ctrl = &adapter->ctrl_table[i];
- if (!adapter->opt->socket_check(ctrl->socket_id, ctrl->tag, adapter->user_data))
- {
- list[i] = ctrl->socket_id;
- }
- else
- {
- ctrl->socket_id = -1;
- list[i] = -1;
- }
- DBG("%d,%d", i, list[i]);
- }
- adapter->opt->socket_clean(list, adapter->opt->max_socket_num, adapter->user_data);
- NW_UNLOCK;
- free(list);
- }
- #ifdef LUAT_USE_TLS
- static int tls_verify(void *ctx, mbedtls_x509_crt *crt, int Index, uint32_t *result)
- {
- network_ctrl_t *ctrl = (network_ctrl_t *)ctx;
- DBG("%d, %08x", Index, *result);
- return 0;
- }
- #endif
- int network_set_psk_info(network_ctrl_t *ctrl,
- const unsigned char *psk, size_t psk_len,
- const unsigned char *psk_identity, size_t psk_identity_len)
- {
- #ifdef LUAT_USE_TLS
- if (!ctrl->tls_mode)
- {
- return -ERROR_PERMISSION_DENIED;
- }
- // DBG("%.*s, %.*s", psk_len, psk, psk_identity_len, psk_identity);
- int ret = mbedtls_ssl_conf_psk( ctrl->config,
- psk, psk_len, psk_identity, psk_identity_len );
- if (ret != 0)
- {
- DBG("0x%x", -ret);
- return -ERROR_OPERATION_FAILED;
- }
- return ERROR_NONE;
- #else
- return -1;
- #endif
- }
- int network_set_server_cert(network_ctrl_t *ctrl, const unsigned char *cert, size_t cert_len)
- {
- #ifdef LUAT_USE_TLS
- int ret;
- if (!ctrl->tls_mode)
- {
- return -ERROR_PERMISSION_DENIED;
- }
- ret = mbedtls_x509_crt_parse( ctrl->ca_cert, cert, cert_len);
- if (ret != 0)
- {
- DBG("%08x", -ret);
- return -ERROR_OPERATION_FAILED;
- }
- return ERROR_NONE;
- #else
- return -1;
- #endif
- }
- int network_set_client_cert(network_ctrl_t *ctrl,
- const unsigned char *cert, size_t certLen,
- const unsigned char *key, size_t keylen,
- const unsigned char *pwd, size_t pwdlen)
- {
- #ifdef LUAT_USE_TLS
- int ret;
- mbedtls_x509_crt *client_cert = NULL;
- mbedtls_pk_context *pkey = NULL;
- if (!ctrl->tls_mode)
- {
- return -ERROR_PERMISSION_DENIED;
- }
- client_cert = zalloc(sizeof(mbedtls_x509_crt));
- pkey = zalloc(sizeof(mbedtls_pk_context));
- if (!client_cert || !pkey)
- {
- goto ERROR_OUT;
- }
- ret = mbedtls_x509_crt_parse( client_cert, cert, certLen );
- if (ret != 0)
- {
- DBG("%08x", -ret);
- goto ERROR_OUT;
- }
- ret = mbedtls_pk_parse_key( pkey, key, keylen, pwd, pwdlen );
- if (ret != 0)
- {
- DBG("%08x", -ret);
- goto ERROR_OUT;
- }
- ret = mbedtls_ssl_conf_own_cert( ctrl->config, client_cert, pkey );
- if (ret != 0)
- {
- DBG("%08x", -ret);
- goto ERROR_OUT;
- }
- return ERROR_NONE;
- ERROR_OUT:
- if (client_cert) free(client_cert);
- if (pkey) free(pkey);
- return -1;
- #else
- return -1;
- #endif
- }
- int network_cert_verify_result(network_ctrl_t *ctrl)
- {
- #ifdef LUAT_USE_TLS
- if (!ctrl->tls_mode)
- {
- return -1;
- }
- return mbedtls_ssl_get_verify_result(ctrl->ssl);
- #else
- return -1;
- #endif
- }
- static int tls_random( void *p_rng,
- unsigned char *output, size_t output_len )
- {
- platform_random((char*)output, output_len);
- return 0;
- }
- void network_init_tls(network_ctrl_t *ctrl, int verify_mode)
- {
- #ifdef LUAT_USE_TLS
- ctrl->tls_mode = 1;
- if (!ctrl->ssl)
- {
- ctrl->ssl = zalloc(sizeof(mbedtls_ssl_context));
- ctrl->ca_cert = zalloc(sizeof(mbedtls_x509_crt));
- ctrl->config = zalloc(sizeof(mbedtls_ssl_config));
- mbedtls_ssl_config_defaults( ctrl->config, MBEDTLS_SSL_IS_CLIENT, ctrl->is_tcp?MBEDTLS_SSL_TRANSPORT_STREAM:MBEDTLS_SSL_TRANSPORT_DATAGRAM, MBEDTLS_SSL_PRESET_DEFAULT);
- ctrl->config->authmode = verify_mode;
- ctrl->config->hs_timeout_min = 20000;
- ctrl->config->f_rng = tls_random;
- ctrl->config->p_rng = NULL;
- ctrl->config->f_dbg = tls_dbg;
- ctrl->config->p_dbg = NULL;
- ctrl->config->f_vrfy = tls_verify;
- ctrl->config->p_vrfy = ctrl;
- ctrl->config->ca_chain = ctrl->ca_cert;
- ctrl->config->read_timeout = 20000;
- ctrl->tls_long_timer = platform_create_timer(tls_longtimeout, ctrl, NULL);
- ctrl->tls_short_timer = platform_create_timer(tls_shorttimeout, ctrl, NULL);
- }
- ctrl->tls_timer_state = -1;
- #endif
- }
- void network_deinit_tls(network_ctrl_t *ctrl)
- {
- #ifdef LUAT_USE_TLS
- if (ctrl->ssl)
- {
- mbedtls_ssl_free(ctrl->ssl);
- free(ctrl->ssl);
- ctrl->ssl = NULL;
- }
- if (ctrl->config)
- {
- mbedtls_ssl_config_free(ctrl->config);
- free(ctrl->config);
- ctrl->config = NULL;
- }
- if (ctrl->ca_cert)
- {
- mbedtls_x509_crt_free(ctrl->ca_cert);
- free(ctrl->ca_cert);
- ctrl->ca_cert = NULL;
- }
- ctrl->tls_mode = 0;
- ctrl->tls_timer_state = -1;
- if (ctrl->tls_short_timer)
- {
- platform_release_timer(ctrl->tls_short_timer);
- ctrl->tls_short_timer = NULL;
- }
- if (ctrl->tls_long_timer)
- {
- platform_release_timer(ctrl->tls_long_timer);
- ctrl->tls_long_timer = NULL;
- }
- #endif
- }
- int network_wait_link_up(network_ctrl_t *ctrl, uint32_t timeout_ms)
- {
- NW_LOCK;
- ctrl->auto_mode = 1;
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- if (network_check_ready(ctrl))
- {
- ctrl->state = NW_STATE_OFF_LINE;
- ctrl->wait_target_state = NW_WAIT_NONE;
- NW_UNLOCK;
- return 0;
- }
- ctrl->state = NW_STATE_LINK_OFF;
- ctrl->wait_target_state = NW_WAIT_LINK_UP;
- NW_UNLOCK;
- if (!ctrl->task_handle || !timeout_ms)
- {
- return 1;
- }
- uint8_t finish = 0;
- OS_EVENT event;
- int result;
- //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
- platform_start_timer(ctrl->timer, timeout_ms, 0);
- while (!finish)
- {
- platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
- switch (event.ID)
- {
- case EV_NW_RESULT_LINK:
- result = (int)event.Param1;
- finish = 1;
- break;
- case EV_NW_TIMEOUT:
- result = -1;
- finish = 1;
- break;
- default:
- if (ctrl->user_callback)
- {
- ctrl->user_callback((void *)&event, ctrl->user_data);
- }
- break;
- }
- }
- platform_stop_timer(ctrl->timer);
- return result;
- }
- /*
- * 1.进行ready检测和等待ready
- * 2.有remote_ip则开始连接服务器并等待连接结果
- * 3.没有remote_ip则开始对url进行dns解析,解析完成后对所有ip进行尝试连接直到有个成功或者全部失败
- * 4.如果是加密模式,还要走握手环节,等到握手环节完成后才能返回结果
- * local_port如果为0则api内部自动生成一个
- */
- int network_connect(network_ctrl_t *ctrl, const char *domain_name, uint32_t domain_name_len, luat_ip_addr_t *remote_ip, uint16_t remote_port, uint32_t timeout_ms)
- {
- if (ctrl->socket_id >= 0)
- {
- return -1;
- }
- NW_LOCK;
- ctrl->is_server_mode = 0;
- if (ctrl->dns_ip)
- {
- free(ctrl->dns_ip);
- ctrl->dns_ip = NULL;
- }
- if (ctrl->cache_data)
- {
- free(ctrl->cache_data);
- ctrl->cache_data = NULL;
- }
- ctrl->need_close = 0;
- ctrl->domain_name = domain_name;
- ctrl->domain_name_len = domain_name_len;
- if (remote_ip)
- {
- ctrl->remote_ip = *remote_ip;
- }
- else
- {
- ctrl->remote_ip.is_ipv6 = 0xff;
- }
- ctrl->auto_mode = 1;
- ctrl->remote_port = remote_port;
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- ctrl->wait_target_state = NW_WAIT_ON_LINE;
- if (!network_check_ready(ctrl))
- {
- ctrl->state = NW_STATE_LINK_OFF;
- goto NETWORK_CONNECT_WAIT;
- }
- if (network_prepare_connect(ctrl))
- {
- ctrl->state = NW_STATE_OFF_LINE;
- ctrl->wait_target_state = NW_WAIT_NONE;
- NW_UNLOCK;
- return -1;
- }
- NETWORK_CONNECT_WAIT:
- NW_UNLOCK;
- if (!ctrl->task_handle || !timeout_ms)
- {
- return 1;
- }
- uint8_t finish = 0;
- OS_EVENT event;
- int result;
- //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
- platform_start_timer(ctrl->timer, timeout_ms, 0);
- while (!finish)
- {
- platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
- switch (event.ID)
- {
- case EV_NW_RESULT_CONNECT:
- result = (int)event.Param1;
- finish = 1;
- break;
- case EV_NW_TIMEOUT:
- result = -1;
- finish = 1;
- break;
- default:
- if (ctrl->user_callback)
- {
- ctrl->user_callback((void *)&event, ctrl->user_data);
- }
- break;
- }
- }
- platform_stop_timer(ctrl->timer);
- return result;
- }
- int network_listen(network_ctrl_t *ctrl, uint32_t timeout_ms)
- {
- if (NW_STATE_LISTEN == ctrl->state)
- {
- DBG("socket %d is listen", ctrl->socket_id);
- return 0;
- }
- if (ctrl->socket_id >= 0)
- {
- return -1;
- }
- NW_LOCK;
- ctrl->is_server_mode = 1;
- ctrl->auto_mode = 1;
- ctrl->need_close = 0;
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- ctrl->wait_target_state = NW_WAIT_ON_LINE;
- if (!network_check_ready(ctrl))
- {
- ctrl->state = NW_STATE_LINK_OFF;
- goto NETWORK_LISTEN_WAIT;
- }
- if (network_base_connect(ctrl, NULL))
- {
- ctrl->state = NW_STATE_OFF_LINE;
- ctrl->wait_target_state = NW_WAIT_NONE;
- NW_UNLOCK;
- return -1;
- }
- ctrl->state = NW_STATE_CONNECTING;
- NETWORK_LISTEN_WAIT:
- NW_UNLOCK;
- if (!ctrl->task_handle || !timeout_ms)
- {
- return 1;
- }
- uint8_t finish = 0;
- OS_EVENT event;
- int result;
- //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
- platform_start_timer(ctrl->timer, timeout_ms, 0);
- while (!finish)
- {
- platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
- switch (event.ID)
- {
- case EV_NW_RESULT_CONNECT:
- result = (int)event.Param1;
- finish = 1;
- break;
- case EV_NW_TIMEOUT:
- result = -1;
- finish = 1;
- break;
- default:
- if (ctrl->user_callback)
- {
- ctrl->user_callback((void *)&event, ctrl->user_data);
- }
- break;
- }
- }
- platform_stop_timer(ctrl->timer);
- return result;
- }
- int network_close(network_ctrl_t *ctrl, uint32_t timeout_ms)
- {
- NW_LOCK;
- if (ctrl->cache_data)
- {
- free(ctrl->cache_data);
- ctrl->cache_data = NULL;
- }
- uint8_t old_state = ctrl->state;
- ctrl->auto_mode = 1;
- ctrl->need_close = 0;
- network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
- #ifdef LUAT_USE_TLS
- if (ctrl->tls_mode)
- {
- mbedtls_ssl_free(ctrl->ssl);
- }
- #endif
- if (ctrl->socket_id < 0)
- {
- ctrl->state = NW_STATE_OFF_LINE;
- ctrl->wait_target_state = NW_WAIT_NONE;
- NW_UNLOCK;
- return 0;
- }
- ctrl->state = NW_STATE_DISCONNECTING;
- ctrl->wait_target_state = NW_WAIT_OFF_LINE;
- if ((NW_STATE_ONLINE == old_state) && ctrl->is_tcp)
- {
- if (network_socket_disconnect(ctrl))
- {
- network_force_close_socket(ctrl);
- ctrl->state = NW_STATE_OFF_LINE;
- ctrl->wait_target_state = NW_WAIT_NONE;
- NW_UNLOCK;
- return 0;
- }
- }
- else
- {
- network_force_close_socket(ctrl);
- ctrl->state = NW_STATE_OFF_LINE;
- ctrl->wait_target_state = NW_WAIT_NONE;
- NW_UNLOCK;
- return 0;
- }
- NW_UNLOCK;
- if (!ctrl->task_handle || !timeout_ms)
- {
- return 1;
- }
- uint8_t finish = 0;
- OS_EVENT event;
- int result;
- //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
- platform_start_timer(ctrl->timer, timeout_ms, 0);
- while (!finish)
- {
- platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
- switch (event.ID)
- {
- case EV_NW_RESULT_CLOSE:
- result = 0;
- finish = 1;
- break;
- case EV_NW_TIMEOUT:
- result = 0;
- finish = 1;
- break;
- default:
- if (ctrl->user_callback)
- {
- ctrl->user_callback((void *)&event, ctrl->user_data);
- }
- break;
- }
- }
- platform_stop_timer(ctrl->timer);
- network_socket_force_close(ctrl);
- return result;
- }
- /*
- * timeout_ms=0时,为非阻塞接口
- */
- int network_tx(network_ctrl_t *ctrl, const uint8_t *data, uint32_t len, int flags, luat_ip_addr_t *remote_ip, uint16_t remote_port, uint32_t *tx_len, uint32_t timeout_ms)
- {
- if ((ctrl->need_close) || (ctrl->socket_id < 0) || (ctrl->state != NW_STATE_ONLINE))
- {
- return -1;
- }
- NW_LOCK;
- int result;
- ctrl->auto_mode = 1;
- #ifdef LUAT_USE_TLS
- if (ctrl->tls_mode)
- {
- if (ctrl->tls_need_reshakehand)
- {
- ctrl->tls_need_reshakehand = 0;
- if (ctrl->cache_data)
- {
- free(ctrl->cache_data);
- ctrl->cache_data = NULL;
- }
- ctrl->cache_data = malloc(len);
- memcpy(ctrl->cache_data, data, len);
- ctrl->cache_len = len;
- mbedtls_ssl_session_reset(ctrl->ssl);
- do
- {
- result = mbedtls_ssl_handshake_step( ctrl->ssl );
- switch(result)
- {
- case MBEDTLS_ERR_SSL_WANT_READ:
- ctrl->state = NW_STATE_SHAKEHAND;
- goto NETWORK_TX_WAIT;
- case 0:
- break;
- default:
- DBG_ERR("0x%x, %d", -result, ctrl->ssl->state);
- ctrl->need_close = 1;
- NW_UNLOCK;
- return -1;
- }
- }while(ctrl->ssl->state != MBEDTLS_SSL_HANDSHAKE_OVER);
- }
- result = mbedtls_ssl_write(ctrl->ssl, data, len);
- if (result < 0)
- {
- DBG("%08x", -result);
- ctrl->need_close = 1;
- NW_UNLOCK;
- return -1;
- }
- *tx_len = result;
- }
- else
- #endif
- {
- result = network_base_tx(ctrl, data, len, flags, remote_ip, remote_port);
- if (result < 0)
- {
- ctrl->need_close = 1;
- NW_UNLOCK;
- return -1;
- }
- *tx_len = result;
- if (!result && len)
- {
- NW_UNLOCK;
- return 0;
- }
- }
- #ifdef LUAT_USE_TLS
- NETWORK_TX_WAIT:
- #endif
- ctrl->wait_target_state = NW_WAIT_TX_OK;
- NW_UNLOCK;
- if (!ctrl->task_handle || !timeout_ms)
- {
- return 1;
- }
- uint8_t finish = 0;
- OS_EVENT event;
- //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
- platform_start_timer(ctrl->timer, timeout_ms, 0);
- while (!finish)
- {
- platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
- switch (event.ID)
- {
- case EV_NW_RESULT_TX:
- result = (int)event.Param1;
- finish = 1;
- break;
- case EV_NW_TIMEOUT:
- result = -1;
- finish = 1;
- break;
- default:
- if (ctrl->user_callback)
- {
- ctrl->user_callback((void *)&event, ctrl->user_data);
- }
- break;
- }
- }
- platform_stop_timer(ctrl->timer);
- return result;
- }
- /*
- * 实际读到的数据量在rx_len里,如果是UDP模式且为server时,需要看remote_ip和remote_port
- */
- int network_rx(network_ctrl_t *ctrl, uint8_t *data, uint32_t len, int flags, luat_ip_addr_t *remote_ip, uint16_t *remote_port, uint32_t *rx_len)
- {
- if (!ctrl->new_rx_flag && ((ctrl->need_close) || (ctrl->socket_id < 0) || (ctrl->state != NW_STATE_ONLINE)))
- {
- return -1;
- }
- NW_LOCK;
- int result = -1;
- ctrl->auto_mode = 1;
- uint32_t read_len = 0;
- uint8_t is_error = 0;
- if (data)
- {
- ctrl->new_rx_flag = 0;
- #ifdef LUAT_USE_TLS
- if (ctrl->tls_mode)
- {
- do
- {
- result = mbedtls_ssl_read(ctrl->ssl, data + read_len, len - read_len);
- if (result < 0 && (result != MBEDTLS_ERR_SSL_WANT_READ))
- {
- is_error = 1;
- break;
- }
- else
- {
- read_len += result;
- if (read_len >= len)
- {
- break;
- }
- }
- }while(network_socket_receive(ctrl, NULL, len, flags, remote_ip, remote_port) > 0);
- if ( !is_error )
- {
- result = read_len;
- }
- else
- {
- result = -1;
- }
- }
- else
- #endif
- {
- result = network_socket_receive(ctrl, data, len, flags, remote_ip, remote_port);
- }
- }
- else
- {
- #ifdef LUAT_USE_TLS
- if (ctrl->tls_mode)
- {
- read_len = 0;
- do
- {
- result = mbedtls_ssl_read(ctrl->ssl, NULL, 0);
- if ((result < 0) && (result != (MBEDTLS_ERR_SSL_WANT_READ)))
- {
- is_error = 1;
- read_len = 0;
- break;
- }
- else if (!result)
- {
- read_len = ctrl->ssl->in_msglen;
- break;
- }
- else if ((MBEDTLS_ERR_SSL_WANT_READ) == result)
- {
- read_len = 0;
- ctrl->new_rx_flag = 0;
- DBG("socket %d ssl data need more", ctrl->socket_id);
- break;
- }
- }while(network_socket_receive(ctrl, NULL, len, flags, remote_ip, remote_port) > 0);
- if ( !is_error )
- {
- result = read_len;
- }
- else
- {
- result = -1;
- }
- }
- else
- #endif
- {
- result = network_socket_receive(ctrl, data, len, flags, remote_ip, remote_port);
- }
- }
- NW_UNLOCK;
- if (result >= 0)
- {
- *rx_len = result;
- return 0;
- }
- else
- {
- return -1;
- }
- }
- int network_wait_event(network_ctrl_t *ctrl, OS_EVENT *out_event, uint32_t timeout_ms, uint8_t *is_timeout)
- {
- if (ctrl->new_rx_flag)
- {
- ctrl->wait_target_state = NW_WAIT_EVENT;
- if (out_event)
- {
- out_event->ID = 0;
- }
- return 0;
- }
- if ((ctrl->need_close) || (ctrl->socket_id < 0) || (ctrl->state != NW_STATE_ONLINE))
- {
- return -1;
- }
- NW_LOCK;
- ctrl->auto_mode = 1;
- ctrl->wait_target_state = NW_WAIT_EVENT;
- NW_UNLOCK;
- if (!ctrl->task_handle || !timeout_ms)
- {
- return 1;
- }
- *is_timeout = 0;
- uint8_t finish = 0;
- OS_EVENT event;
- int result;
- //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
- platform_start_timer(ctrl->timer, timeout_ms, 0);
- while (!finish)
- {
- platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
- switch (event.ID)
- {
- case EV_NW_RESULT_EVENT:
- result = (int)event.Param1;
- if (result)
- {
- result = -1;
- }
- if (out_event)
- {
- out_event->ID = 0;
- }
- finish = 1;
- break;
- case EV_NW_TIMEOUT:
- *is_timeout = 1;
- result = 0;
- finish = 1;
- break;
- case EV_NW_BREAK_WAIT:
- if (out_event)
- {
- *out_event = event;
- }
- result = 0;
- finish = 1;
- break;
- default:
- if (out_event)
- {
- *out_event = event;
- result = 0;
- finish = 1;
- break;
- }
- else if (ctrl->user_callback)
- {
- ctrl->user_callback((void *)&event, ctrl->user_data);
- }
- break;
- }
- }
- platform_stop_timer(ctrl->timer);
- return result;
- }
- int network_wait_rx(network_ctrl_t *ctrl, uint32_t timeout_ms, uint8_t *is_break, uint8_t *is_timeout)
- {
- *is_timeout = 0;
- *is_break = 0;
- if (ctrl->new_rx_flag)
- {
- ctrl->wait_target_state = NW_WAIT_EVENT;
- return 0;
- }
- if ((ctrl->need_close) || (ctrl->socket_id < 0) || (ctrl->state != NW_STATE_ONLINE))
- {
- return -1;
- }
- NW_LOCK;
- ctrl->auto_mode = 1;
- ctrl->wait_target_state = NW_WAIT_EVENT;
- NW_UNLOCK;
- uint8_t finish = 0;
- OS_EVENT event;
- int result;
- //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
- if (timeout_ms)
- {
- platform_start_timer(ctrl->timer, timeout_ms, 0);
- }
- while (!finish)
- {
- platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
- switch (event.ID)
- {
- case EV_NW_RESULT_EVENT:
- result = (int)event.Param1;
- if (result)
- {
- result = -1;
- finish = 1;
- }
- else if (ctrl->new_rx_flag)
- {
- result = 0;
- finish = 1;
- }
- break;
- case EV_NW_TIMEOUT:
- *is_timeout = 1;
- result = 0;
- finish = 1;
- break;
- case EV_NW_BREAK_WAIT:
- *is_break = 1;
- result = 0;
- finish = 1;
- break;
- default:
- if (ctrl->user_callback)
- {
- ctrl->user_callback((void *)&event, ctrl->user_data);
- }
- break;
- }
- ctrl->wait_target_state = NW_WAIT_EVENT;
- }
- platform_stop_timer(ctrl->timer);
- return result;
- }
- #endif
- #endif
|