luat_network_adapter.c 103 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016401740184019402040214022402340244025402640274028402940304031403240334034403540364037403840394040404140424043404440454046404740484049405040514052405340544055405640574058405940604061406240634064406540664067406840694070407140724073407440754076407740784079408040814082408340844085408640874088408940904091409240934094409540964097409840994100410141024103410441054106410741084109411041114112411341144115411641174118411941204121412241234124412541264127412841294130413141324133413441354136413741384139414041414142414341444145414641474148414941504151415241534154415541564157415841594160416141624163416441654166416741684169417041714172417341744175417641774178417941804181418241834184418541864187418841894190419141924193419441954196419741984199420042014202420342044205420642074208420942104211421242134214421542164217421842194220422142224223422442254226422742284229423042314232423342344235423642374238423942404241424242434244424542464247424842494250425142524253425442554256425742584259426042614262426342644265426642674268426942704271427242734274427542764277427842794280428142824283428442854286428742884289429042914292429342944295429642974298429943004301430243034304430543064307430843094310431143124313431443154316431743184319432043214322432343244325432643274328432943304331433243334334433543364337433843394340434143424343434443454346434743484349435043514352435343544355435643574358435943604361436243634364436543664367436843694370437143724373437443754376437743784379438043814382438343844385438643874388438943904391439243934394439543964397439843994400440144024403440444054406440744084409441044114412441344144415441644174418441944204421442244234424442544264427442844294430443144324433443444354436443744384439444044414442444344444445444644474448444944504451445244534454445544564457445844594460446144624463446444654466446744684469447044714472447344744475447644774478447944804481448244834484448544864487448844894490449144924493449444954496
  1. #include "luat_base.h"
  2. #include "luat_malloc.h"
  3. #include "luat_crypto.h"
  4. #include "luat_rtos.h"
  5. #ifdef LUAT_USE_NETWORK
  6. #include "luat_rtos.h"
  7. #include "platform_def.h"
  8. #include "ctype.h"
  9. #include "luat_network_adapter.h"
  10. #define LUAT_LOG_TAG "adapter"
  11. #include "luat_log.h"
  12. #ifndef LWIP_NUM_SOCKETS
  13. #define LWIP_NUM_SOCKETS 8
  14. #endif
  15. typedef struct
  16. {
  17. #ifdef LUAT_USE_LWIP
  18. network_ctrl_t lwip_ctrl_table[LWIP_NUM_SOCKETS];
  19. // HANDLE network_mutex;
  20. #endif
  21. int last_adapter_index;
  22. int default_adapter_index;
  23. llist_head dns_cache_head;
  24. #ifdef LUAT_USE_LWIP
  25. uint8_t lwip_ctrl_busy[LWIP_NUM_SOCKETS];
  26. #endif
  27. uint8_t is_init;
  28. }network_info_t;
  29. typedef struct
  30. {
  31. network_adapter_info *opt;
  32. void *user_data;
  33. uint8_t *ctrl_busy;
  34. network_ctrl_t *ctrl_table;
  35. uint16_t port;
  36. }network_adapter_t;
  37. static network_adapter_t prv_adapter_table[NW_ADAPTER_QTY];
  38. static network_info_t prv_network = {
  39. .last_adapter_index = -1,
  40. .default_adapter_index = -1,
  41. .is_init = 0,
  42. };
  43. static const char *prv_network_event_id_string[] =
  44. {
  45. "适配器复位",
  46. "LINK状态变更",
  47. "超时",
  48. "DNS结果",
  49. "发送成功",
  50. "有新的数据",
  51. "接收缓存满了",
  52. "断开成功",
  53. "对端关闭",
  54. "连接成功",
  55. "连接异常",
  56. "开始监听",
  57. "新的客户端来了",
  58. "唤醒",
  59. "未知",
  60. };
  61. static const char *prv_network_ctrl_state_string[] =
  62. {
  63. "硬件离线",
  64. "离线",
  65. "等待DNS",
  66. "正在连接",
  67. "正在TLS握手",
  68. "在线",
  69. "在监听",
  70. "正在离线",
  71. "未知"
  72. };
  73. static const char *prv_network_ctrl_wait_state_string[] =
  74. {
  75. "无等待",
  76. "等待硬件上线",
  77. "等待连接完成",
  78. "等待发送完成",
  79. "等待离线完成",
  80. "等待任意网络变化",
  81. "未知",
  82. };
  83. static const char *prv_network_ctrl_callback_event_string[] =
  84. {
  85. "硬件状态回调",
  86. "连接状态回调",
  87. "离线状态回调",
  88. "发送状态回调",
  89. "任意网络变化回调",
  90. };
  91. const char *network_ctrl_event_id_string(uint32_t event)
  92. {
  93. if (event > EV_NW_END || event < EV_NW_RESET)
  94. {
  95. return prv_network_event_id_string[EV_NW_END - EV_NW_RESET];
  96. }
  97. return prv_network_event_id_string[event - EV_NW_RESET];
  98. }
  99. const char *network_ctrl_state_string(uint8_t state)
  100. {
  101. if (state > NW_STATE_DISCONNECTING)
  102. {
  103. return prv_network_ctrl_state_string[NW_STATE_DISCONNECTING + 1];
  104. }
  105. return prv_network_ctrl_state_string[state];
  106. }
  107. const char *network_ctrl_wait_state_string(uint8_t state)
  108. {
  109. if (state > NW_WAIT_EVENT)
  110. {
  111. return prv_network_ctrl_wait_state_string[NW_WAIT_EVENT + 1];
  112. }
  113. return prv_network_ctrl_wait_state_string[state];
  114. }
  115. const char *network_ctrl_callback_event_string(uint32_t event)
  116. {
  117. if (event > EV_NW_RESULT_EVENT || event < EV_NW_RESULT_LINK)
  118. {
  119. return prv_network_ctrl_callback_event_string[event - EV_NW_RESULT_EVENT + 1];
  120. }
  121. return prv_network_ctrl_callback_event_string[event - EV_NW_RESULT_EVENT];
  122. }
  123. network_adapter_info* network_adapter_fetch(int id, void** userdata) {
  124. if (id >= 0 && id < NW_ADAPTER_QTY) {
  125. if (prv_adapter_table[id].opt) {
  126. *userdata = prv_adapter_table[id].user_data;
  127. return prv_adapter_table[id].opt;
  128. }
  129. }
  130. return NULL;
  131. }
  132. #ifdef LUAT_USE_LWIP
  133. #include "../lwip/port/net_lwip.h"
  134. extern void DBG_Printf(const char* format, ...);
  135. extern void DBG_HexPrintf(void *Data, unsigned int len);
  136. //#define DBG(x,y...) DBG_Printf("%s %d:"x"\r\n", __FUNCTION__,__LINE__,##y)
  137. //#define DBG_ERR(x,y...) DBG_Printf("%s %d:"x"\r\n", __FUNCTION__,__LINE__,##y)
  138. static int tls_random( void *p_rng,
  139. unsigned char *output, size_t output_len );
  140. #define __NW_DEBUG_ENABLE__
  141. #ifdef __NW_DEBUG_ENABLE__
  142. #ifdef LUAT_LOG_NO_NEWLINE
  143. #define DBG(x,y...) do {if (ctrl->is_debug) {DBG_Printf("%s %d:"x, __FUNCTION__,__LINE__,##y);}} while(0)
  144. #define DBG_ERR(x,y...) DBG_Printf("%s %d:"x, __FUNCTION__,__LINE__,##y)
  145. #else
  146. #define DBG(x,y...) do {if (ctrl->is_debug) {DBG_Printf("%s %d:"x"\r\n", __FUNCTION__,__LINE__,##y);}} while(0)
  147. #define DBG_ERR(x,y...) DBG_Printf("%s %d:"x"\r\n", __FUNCTION__,__LINE__,##y)
  148. #endif
  149. #else
  150. #define DBG(x,y...)
  151. #define DBG_ERR(x,y...)
  152. #endif
  153. #define NW_LOCK platform_lock_mutex(ctrl->mutex)
  154. #define NW_UNLOCK platform_unlock_mutex(ctrl->mutex)
  155. #define SOL_SOCKET 0xfff /* options for socket level */
  156. #define SO_REUSEADDR 0x0004 /* Allow local address reuse */
  157. #define SO_KEEPALIVE 0x0008 /* keep connections alive */
  158. #define IPPROTO_ICMP 1
  159. #define IPPROTO_TCP 6
  160. #define TCP_NODELAY 0x01 /* don't delay send to coalesce packets */
  161. #define TCP_KEEPALIVE 0x02 /* send KEEPALIVE probes when idle for pcb->keep_idle milliseconds */
  162. #define TCP_KEEPIDLE 0x03 /* set pcb->keep_idle - Same as TCP_KEEPALIVE, but use seconds for get/setsockopt */
  163. #define TCP_KEEPINTVL 0x04 /* set pcb->keep_intvl - Use seconds for get/setsockopt */
  164. #define TCP_KEEPCNT 0x05 /* set pcb->keep_cnt - Use number of probes sent for get/setsockopt */
  165. static uint8_t network_check_ip_same(luat_ip_addr_t *ip1, luat_ip_addr_t *ip2)
  166. {
  167. #if defined ENABLE_PSIF
  168. return ip_addr_cmp(ip1, ip2);
  169. #else
  170. return ip_addr_cmp_zoneless(ip1, ip2);
  171. #endif
  172. }
  173. 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)
  174. {
  175. int result = -1;
  176. if (ctrl->is_tcp)
  177. {
  178. result = network_socket_send(ctrl, data, len, flags, NULL, 0);
  179. }
  180. else
  181. {
  182. if (remote_ip)
  183. {
  184. result = network_socket_send(ctrl, data, len, flags, remote_ip, remote_port);
  185. }
  186. else
  187. {
  188. result = network_socket_send(ctrl, data, len, flags, ctrl->online_ip, ctrl->remote_port);
  189. }
  190. }
  191. if (result >= 0)
  192. {
  193. ctrl->tx_size += len;
  194. }
  195. else
  196. {
  197. ctrl->need_close = 1;
  198. }
  199. return result;
  200. }
  201. static LUAT_RT_RET_TYPE tls_shorttimeout(LUAT_RT_CB_PARAM)
  202. {
  203. network_ctrl_t *ctrl = (network_ctrl_t *)param;
  204. if (!ctrl->tls_mode)
  205. {
  206. platform_stop_timer(ctrl->tls_long_timer);
  207. return LUAT_RT_RET;
  208. }
  209. if (0 == ctrl->tls_timer_state)
  210. {
  211. ctrl->tls_timer_state = 1;
  212. }
  213. return LUAT_RT_RET;
  214. }
  215. static LUAT_RT_RET_TYPE tls_longtimeout(LUAT_RT_CB_PARAM)
  216. {
  217. network_ctrl_t *ctrl = (network_ctrl_t *)param;
  218. platform_stop_timer(ctrl->tls_short_timer);
  219. if (!ctrl->tls_mode)
  220. {
  221. return LUAT_RT_RET;
  222. }
  223. ctrl->tls_timer_state = 2;
  224. return LUAT_RT_RET;
  225. }
  226. static void tls_settimer( void *data, uint32_t int_ms, uint32_t fin_ms )
  227. {
  228. network_ctrl_t *ctrl = (network_ctrl_t *)data;
  229. if (!ctrl->tls_mode)
  230. {
  231. return;
  232. }
  233. if (!fin_ms)
  234. {
  235. platform_stop_timer(ctrl->tls_short_timer);
  236. platform_stop_timer(ctrl->tls_long_timer);
  237. ctrl->tls_timer_state = -1;
  238. return ;
  239. }
  240. platform_start_timer(ctrl->tls_short_timer, int_ms, 0);
  241. platform_start_timer(ctrl->tls_long_timer, fin_ms, 0);
  242. ctrl->tls_timer_state = 0;
  243. }
  244. static int tls_gettimer( void *data )
  245. {
  246. network_ctrl_t *ctrl = (network_ctrl_t *)data;
  247. if (!ctrl->tls_mode)
  248. {
  249. return -ERROR_PARAM_INVALID;
  250. }
  251. return ctrl->tls_timer_state;
  252. }
  253. static void tls_dbg(void *data, int level,
  254. const char *file, int line,
  255. const char *str)
  256. {
  257. DBG_Printf("%s %d:%s", file, line, str);
  258. }
  259. static int tls_send(void *ctx, const unsigned char *buf, size_t len )
  260. {
  261. network_ctrl_t *ctrl = (network_ctrl_t *)ctx;
  262. if (!ctrl->tls_mode)
  263. {
  264. return -ERROR_PERMISSION_DENIED;
  265. }
  266. if (network_base_tx(ctrl, buf, len, 0, NULL, 0) != len)
  267. {
  268. return -0x004E;
  269. }
  270. else
  271. {
  272. return len;
  273. }
  274. }
  275. static int tls_recv(void *ctx, unsigned char *buf, size_t len )
  276. {
  277. #ifdef LUAT_USE_TLS
  278. network_ctrl_t *ctrl = (network_ctrl_t *)ctx;
  279. luat_ip_addr_t remote_ip;
  280. uint16_t remote_port;
  281. int result = -1;
  282. if (!ctrl->tls_mode)
  283. {
  284. return -1;
  285. }
  286. TLS_RECV:
  287. result = network_socket_receive(ctrl, buf, len, 0, &remote_ip, &remote_port);
  288. if (result < 0)
  289. {
  290. return -0x004C;
  291. }
  292. if (result > 0)
  293. {
  294. if (!ctrl->is_tcp)
  295. {
  296. if ((remote_port == ctrl->remote_port) && network_check_ip_same(&remote_ip, ctrl->online_ip))
  297. {
  298. goto TLS_RECV;
  299. }
  300. }
  301. return result;
  302. }
  303. return MBEDTLS_ERR_SSL_WANT_READ;
  304. #else
  305. return -1;
  306. #endif
  307. }
  308. static int network_get_host_by_name(network_ctrl_t *ctrl)
  309. {
  310. ctrl->remote_ip.type = 0xff;
  311. if (ipaddr_aton(ctrl->domain_name, &ctrl->remote_ip))
  312. {
  313. return 0;
  314. }
  315. ctrl->remote_ip.type = 0xff;
  316. return -1;
  317. }
  318. static void network_update_dns_cache(network_ctrl_t *ctrl)
  319. {
  320. }
  321. static void network_get_dns_cache(network_ctrl_t *ctrl)
  322. {
  323. }
  324. static int network_base_connect(network_ctrl_t *ctrl, luat_ip_addr_t *remote_ip)
  325. {
  326. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  327. if (ctrl->socket_id >= 0)
  328. {
  329. network_force_close_socket(ctrl);
  330. }
  331. if (remote_ip)
  332. {
  333. if (network_create_soceket(ctrl, IPADDR_TYPE_V6 == remote_ip->type) < 0)
  334. {
  335. network_clean_invaild_socket(ctrl->adapter_index);
  336. if (network_create_soceket(ctrl, IPADDR_TYPE_V6 == remote_ip->type) < 0)
  337. {
  338. return -1;
  339. }
  340. }
  341. if (adapter->opt->is_posix)
  342. {
  343. volatile uint32_t val;
  344. val = ctrl->tcp_keep_alive;
  345. network_setsockopt(ctrl, SOL_SOCKET, SO_KEEPALIVE, (void *)&val, sizeof(val));
  346. if (ctrl->tcp_keep_alive)
  347. {
  348. val = ctrl->tcp_keep_idle;
  349. network_setsockopt(ctrl, IPPROTO_TCP, TCP_KEEPIDLE, (void*)&val, sizeof(val));
  350. val = ctrl->tcp_keep_interval;
  351. network_setsockopt(ctrl, IPPROTO_TCP, TCP_KEEPINTVL, (void *)&val, sizeof(val));
  352. val = ctrl->tcp_keep_cnt;
  353. network_setsockopt(ctrl, IPPROTO_TCP, TCP_KEEPCNT, (void *)&val, sizeof(val));
  354. }
  355. }
  356. else
  357. {
  358. network_user_cmd(ctrl, NW_CMD_AUTO_HEART_TIME, ctrl->tcp_keep_idle);
  359. }
  360. return network_socket_connect(ctrl, remote_ip);
  361. }
  362. else
  363. {
  364. if (network_create_soceket(ctrl, 0) < 0)
  365. {
  366. network_clean_invaild_socket(ctrl->adapter_index);
  367. if (network_create_soceket(ctrl, 0) < 0)
  368. {
  369. return -1;
  370. }
  371. }
  372. return network_socket_listen(ctrl);
  373. }
  374. }
  375. static int network_prepare_connect(network_ctrl_t *ctrl)
  376. {
  377. if (ctrl->remote_ip.type != 0xff)
  378. {
  379. ;
  380. }
  381. else if (ctrl->domain_name)
  382. {
  383. if (network_get_host_by_name(ctrl))
  384. {
  385. if (network_dns(ctrl))
  386. {
  387. network_socket_force_close(ctrl);
  388. return -1;
  389. }
  390. ctrl->state = NW_STATE_WAIT_DNS;
  391. return 0;
  392. }
  393. }
  394. else
  395. {
  396. return -1;
  397. }
  398. if (network_base_connect(ctrl, &ctrl->remote_ip))
  399. {
  400. network_socket_force_close(ctrl);
  401. return -1;
  402. }
  403. ctrl->state = NW_STATE_CONNECTING;
  404. return 0;
  405. }
  406. static int network_state_link_off(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  407. {
  408. if (EV_NW_STATE == event->ID)
  409. {
  410. if (event->Param2)
  411. {
  412. ctrl->state = NW_STATE_OFF_LINE;
  413. if (NW_WAIT_LINK_UP == ctrl->wait_target_state)
  414. {
  415. return 0;
  416. }
  417. else if (NW_WAIT_ON_LINE == ctrl->wait_target_state)
  418. {
  419. if (ctrl->is_server_mode)
  420. {
  421. if (network_base_connect(ctrl, NULL))
  422. {
  423. return -1;
  424. }
  425. ctrl->state = NW_STATE_CONNECTING;
  426. }
  427. else
  428. {
  429. if (network_prepare_connect(ctrl))
  430. {
  431. return -1;
  432. }
  433. }
  434. return 1;
  435. }
  436. }
  437. }
  438. return 1;
  439. }
  440. static int network_state_off_line(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  441. {
  442. return 1;
  443. }
  444. static int network_state_wait_dns(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  445. {
  446. if ((ctrl->need_close) || ctrl->wait_target_state != NW_WAIT_ON_LINE) return -1;
  447. int i;
  448. switch(event->ID)
  449. {
  450. case EV_NW_RESET:
  451. case EV_NW_SOCKET_ERROR:
  452. return -1;
  453. case EV_NW_STATE:
  454. if (!event->Param2)
  455. {
  456. return -1;
  457. }
  458. break;
  459. case EV_NW_DNS_RESULT:
  460. if (event->Param1)
  461. {
  462. //更新dns cache
  463. ctrl->dns_ip = event->Param2;
  464. ctrl->dns_ip_nums = event->Param1;
  465. for(i = 0; i < ctrl->dns_ip_nums; i++)
  466. {
  467. DBG("dns ip%d, ttl %u, %s", i, ctrl->dns_ip[i].ttl_end, ipaddr_ntoa(&ctrl->dns_ip[i].ip));
  468. }
  469. network_update_dns_cache(ctrl);
  470. }
  471. else
  472. {
  473. ctrl->dns_ip_nums = 0;
  474. network_get_dns_cache(ctrl);
  475. if (!ctrl->dns_ip_nums)
  476. {
  477. return -1;
  478. }
  479. }
  480. ctrl->dns_ip_cnt = 0;
  481. if (network_base_connect(ctrl, &ctrl->dns_ip[ctrl->dns_ip_cnt].ip))
  482. {
  483. network_socket_force_close(ctrl);
  484. return -1;
  485. }
  486. else
  487. {
  488. ctrl->state = NW_STATE_CONNECTING;
  489. return 1;
  490. }
  491. default:
  492. return 1;
  493. }
  494. return -1;
  495. }
  496. static int network_state_connecting(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  497. {
  498. if ((ctrl->need_close) || (ctrl->wait_target_state != NW_WAIT_ON_LINE)) return -1;
  499. switch(event->ID)
  500. {
  501. case EV_NW_RESET:
  502. return -1;
  503. case EV_NW_SOCKET_ERROR:
  504. case EV_NW_SOCKET_REMOTE_CLOSE:
  505. case EV_NW_SOCKET_CLOSE_OK:
  506. if (ctrl->remote_ip.type != 0xff)
  507. {
  508. return -1;
  509. }
  510. DBG("dns ip %d no connect!", ctrl->dns_ip_cnt);
  511. ctrl->dns_ip_cnt++;
  512. if (ctrl->dns_ip_cnt >= ctrl->dns_ip_nums)
  513. {
  514. DBG("all ip try connect failed");
  515. return -1;
  516. }
  517. if (network_base_connect(ctrl, &ctrl->dns_ip[ctrl->dns_ip_cnt].ip))
  518. {
  519. network_socket_force_close(ctrl);
  520. return -1;
  521. }
  522. else
  523. {
  524. ctrl->state = NW_STATE_CONNECTING;
  525. return 1;
  526. }
  527. break;
  528. case EV_NW_STATE:
  529. if (!event->Param2)
  530. {
  531. return -1;
  532. }
  533. break;
  534. case EV_NW_SOCKET_LISTEN:
  535. if (ctrl->is_server_mode)
  536. {
  537. ctrl->state = NW_STATE_LISTEN;
  538. return 1;
  539. }
  540. break;
  541. case EV_NW_SOCKET_CONNECT_OK:
  542. #ifdef LUAT_USE_TLS
  543. if (ctrl->tls_mode)
  544. {
  545. mbedtls_ssl_free(ctrl->ssl);
  546. memset(ctrl->ssl, 0, sizeof(mbedtls_ssl_context));
  547. mbedtls_ssl_setup(ctrl->ssl, ctrl->config);
  548. // ctrl->ssl->f_set_timer = tls_settimer;
  549. // ctrl->ssl->f_get_timer = tls_gettimer;
  550. // ctrl->ssl->p_timer = ctrl;
  551. mbedtls_ssl_set_timer_cb(ctrl->ssl, ctrl, tls_settimer, tls_gettimer);
  552. // ctrl->ssl->p_bio = ctrl;
  553. // ctrl->ssl->f_send = tls_send;
  554. // ctrl->ssl->f_recv = tls_recv;
  555. mbedtls_ssl_set_bio(ctrl->ssl, ctrl, tls_send, tls_recv, NULL);
  556. // add by wendal
  557. // cloudflare的https需要设置hostname才能访问
  558. if (ctrl->domain_name_len > 0 && ctrl->domain_name_len < 256) {
  559. char host[257] = {0};
  560. memcpy(host, ctrl->domain_name, ctrl->domain_name_len);
  561. mbedtls_ssl_set_hostname(ctrl->ssl, host);
  562. //LLOGD("CALL mbedtls_ssl_set_hostname(%s)", host);
  563. }
  564. else {
  565. //LLOGD("skip mbedtls_ssl_set_hostname");
  566. }
  567. ctrl->state = NW_STATE_SHAKEHAND;
  568. do
  569. {
  570. int result = mbedtls_ssl_handshake_step( ctrl->ssl );
  571. switch(result)
  572. {
  573. case MBEDTLS_ERR_SSL_WANT_READ:
  574. return 1;
  575. case 0:
  576. break;
  577. default:
  578. #if MBEDTLS_VERSION_NUMBER >= 0x03000000
  579. #else
  580. DBG_ERR("0x%x, %d", -result, ctrl->ssl->state);
  581. #endif
  582. return -1;
  583. }
  584. #if MBEDTLS_VERSION_NUMBER >= 0x03000000
  585. }while(!mbedtls_ssl_is_handshake_over(ctrl->ssl));
  586. #else
  587. }while(ctrl->ssl->state != MBEDTLS_SSL_HANDSHAKE_OVER);
  588. #endif
  589. return 0;
  590. }
  591. else
  592. #endif
  593. {
  594. ctrl->state = NW_STATE_ONLINE;
  595. return 0;
  596. }
  597. default:
  598. return 1;
  599. }
  600. return -1;
  601. }
  602. static int network_state_shakehand(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  603. {
  604. if ((ctrl->need_close) || ((ctrl->wait_target_state != NW_WAIT_ON_LINE) && (ctrl->wait_target_state != NW_WAIT_TX_OK))) return -1;
  605. switch(event->ID)
  606. {
  607. case EV_NW_RESET:
  608. case EV_NW_SOCKET_ERROR:
  609. case EV_NW_SOCKET_REMOTE_CLOSE:
  610. case EV_NW_SOCKET_CLOSE_OK:
  611. ctrl->need_close = 1;
  612. return -1;
  613. case EV_NW_STATE:
  614. if (!event->Param2)
  615. {
  616. ctrl->need_close = 1;
  617. return -1;
  618. }
  619. break;
  620. case EV_NW_SOCKET_TX_OK:
  621. ctrl->ack_size += event->Param2;
  622. if (ctrl->is_debug)
  623. {
  624. DBG("%llu,%llu",ctrl->tx_size, ctrl->ack_size);
  625. }
  626. break;
  627. #ifdef LUAT_USE_TLS
  628. case EV_NW_SOCKET_RX_NEW:
  629. do
  630. {
  631. int result = mbedtls_ssl_handshake_step( ctrl->ssl );
  632. switch(result)
  633. {
  634. case MBEDTLS_ERR_SSL_WANT_READ:
  635. return 1;
  636. case 0:
  637. break;
  638. default:
  639. #if MBEDTLS_VERSION_NUMBER >= 0x03000000
  640. #else
  641. DBG_ERR("0x%x, %d", -result, ctrl->ssl->state);
  642. #endif
  643. ctrl->need_close = 1;
  644. return -1;
  645. }
  646. #if MBEDTLS_VERSION_NUMBER >= 0x03000000
  647. }while(!mbedtls_ssl_is_handshake_over(ctrl->ssl));
  648. #else
  649. }while(ctrl->ssl->state != MBEDTLS_SSL_HANDSHAKE_OVER);
  650. #endif
  651. ctrl->state = NW_STATE_ONLINE;
  652. if (NW_WAIT_TX_OK == ctrl->wait_target_state)
  653. {
  654. if (!ctrl->cache_data)
  655. {
  656. ctrl->need_close = 1;
  657. return -1;
  658. }
  659. int result = mbedtls_ssl_write(ctrl->ssl, ctrl->cache_data, ctrl->cache_len);
  660. free(ctrl->cache_data);
  661. ctrl->cache_data = NULL;
  662. ctrl->cache_len = 0;
  663. if (result < 0)
  664. {
  665. DBG("%08x", -result);
  666. ctrl->need_close = 1;
  667. return -1;
  668. }
  669. return 1;
  670. }
  671. return 0;
  672. #endif
  673. case EV_NW_SOCKET_CONNECT_OK:
  674. DBG_ERR("!");
  675. return 1;
  676. default:
  677. return 1;
  678. }
  679. return 1;
  680. }
  681. static int network_state_on_line(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  682. {
  683. if ((ctrl->need_close) || NW_WAIT_OFF_LINE == ctrl->wait_target_state)
  684. {
  685. return -1;
  686. }
  687. switch(event->ID)
  688. {
  689. case EV_NW_RESET:
  690. case EV_NW_SOCKET_ERROR:
  691. case EV_NW_SOCKET_REMOTE_CLOSE:
  692. case EV_NW_SOCKET_CLOSE_OK:
  693. ctrl->need_close = 1;
  694. return -1;
  695. case EV_NW_STATE:
  696. if (!event->Param2)
  697. {
  698. ctrl->need_close = 1;
  699. return -1;
  700. }
  701. break;
  702. case EV_NW_SOCKET_TX_OK:
  703. ctrl->ack_size += event->Param2;
  704. if (NW_WAIT_TX_OK == ctrl->wait_target_state)
  705. {
  706. if (ctrl->ack_size == ctrl->tx_size)
  707. {
  708. #ifdef LUAT_USE_LWIP
  709. if (ctrl->is_tcp)
  710. {
  711. return net_lwip_check_all_ack(ctrl->socket_id);
  712. }
  713. return 0;
  714. #else
  715. return 0;
  716. #endif
  717. }
  718. }
  719. break;
  720. case EV_NW_SOCKET_RX_NEW:
  721. ctrl->new_rx_flag = 1;
  722. if (NW_WAIT_TX_OK != ctrl->wait_target_state)
  723. {
  724. return 0;
  725. }
  726. break;
  727. default:
  728. return 1;
  729. }
  730. return 1;
  731. }
  732. static int network_state_listen(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  733. {
  734. if ((ctrl->need_close) || NW_WAIT_OFF_LINE == ctrl->wait_target_state)
  735. {
  736. return -1;
  737. }
  738. switch(event->ID)
  739. {
  740. case EV_NW_RESET:
  741. case EV_NW_SOCKET_ERROR:
  742. case EV_NW_SOCKET_REMOTE_CLOSE:
  743. case EV_NW_SOCKET_CLOSE_OK:
  744. ctrl->need_close = 1;
  745. return -1;
  746. case EV_NW_STATE:
  747. if (!event->Param2)
  748. {
  749. ctrl->need_close = 1;
  750. return -1;
  751. }
  752. break;
  753. case EV_NW_SOCKET_NEW_CONNECT:
  754. case EV_NW_SOCKET_CONNECT_OK:
  755. ctrl->state = NW_STATE_ONLINE;
  756. return 0;
  757. default:
  758. return 1;
  759. }
  760. return 1;
  761. }
  762. static int network_state_disconnecting(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  763. {
  764. if (ctrl->wait_target_state != NW_WAIT_OFF_LINE)
  765. {
  766. return -1;
  767. }
  768. switch(event->ID)
  769. {
  770. case EV_NW_RESET:
  771. case EV_NW_SOCKET_ERROR:
  772. case EV_NW_SOCKET_REMOTE_CLOSE:
  773. case EV_NW_SOCKET_CLOSE_OK:
  774. network_force_close_socket(ctrl);
  775. ctrl->state = NW_STATE_OFF_LINE;
  776. ctrl->socket_id = -1;
  777. return 0;
  778. case EV_NW_STATE:
  779. if (!event->Param2)
  780. {
  781. return -1;
  782. }
  783. else
  784. {
  785. network_force_close_socket(ctrl);
  786. ctrl->state = NW_STATE_OFF_LINE;
  787. ctrl->socket_id = -1;
  788. }
  789. break;
  790. default:
  791. return 1;
  792. }
  793. return -1;
  794. }
  795. typedef int (*network_state_fun)(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter);
  796. static network_state_fun network_state_fun_list[]=
  797. {
  798. network_state_link_off,
  799. network_state_off_line,
  800. network_state_wait_dns,
  801. network_state_connecting,
  802. network_state_shakehand,
  803. network_state_on_line,
  804. network_state_listen,
  805. network_state_disconnecting,
  806. };
  807. static void network_default_statemachine(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  808. {
  809. int result = -1;
  810. NW_LOCK;
  811. if (ctrl->state > NW_STATE_DISCONNECTING)
  812. {
  813. ctrl->state = NW_STATE_LINK_OFF;
  814. event->Param1 = -1;
  815. network_force_close_socket(ctrl);
  816. event->ID = ctrl->wait_target_state + EV_NW_RESULT_BASE;
  817. }
  818. else
  819. {
  820. result = network_state_fun_list[ctrl->state](ctrl, event, adapter);
  821. if (result > 0)
  822. {
  823. NW_UNLOCK;
  824. return ;
  825. }
  826. event->ID = (ctrl->wait_target_state?ctrl->wait_target_state:NW_WAIT_EVENT) + EV_NW_RESULT_BASE;
  827. event->Param1 = result;
  828. }
  829. if ((ctrl->state != NW_STATE_LISTEN) || (result < 0))
  830. {
  831. ctrl->wait_target_state = NW_WAIT_NONE;
  832. }
  833. NW_UNLOCK;
  834. if (ctrl->task_handle)
  835. {
  836. platform_send_event(ctrl->task_handle, event->ID, event->Param1, event->Param2, event->Param3);
  837. }
  838. else if (ctrl->user_callback)
  839. {
  840. ctrl->user_callback(event, ctrl->user_data);
  841. }
  842. }
  843. static int32_t network_default_socket_callback(void *data, void *param)
  844. {
  845. OS_EVENT *event = (OS_EVENT *)data;
  846. OS_EVENT temp_event;
  847. luat_network_cb_param_t *cb_param = (luat_network_cb_param_t *)param;
  848. network_adapter_t *adapter =(network_adapter_t *)(cb_param->param);
  849. int i;
  850. network_ctrl_t *ctrl = (network_ctrl_t *)event->Param3;
  851. if (event->ID > EV_NW_TIMEOUT)
  852. {
  853. if (ctrl && ((ctrl->tag == cb_param->tag) || (event->ID == EV_NW_DNS_RESULT)))
  854. {
  855. if (ctrl->auto_mode)
  856. {
  857. DBG("socket %d,%s,%s,%s", ctrl->socket_id, network_ctrl_event_id_string(event->ID),
  858. network_ctrl_state_string(ctrl->state),
  859. network_ctrl_wait_state_string(ctrl->wait_target_state));
  860. network_default_statemachine(ctrl, event, adapter);
  861. DBG("%s,%s",network_ctrl_state_string(ctrl->state),network_ctrl_wait_state_string(ctrl->wait_target_state));
  862. }
  863. else if (ctrl->task_handle)
  864. {
  865. platform_send_event(ctrl->task_handle, event->ID, event->Param1, event->Param2, event->Param3);
  866. }
  867. else if (ctrl->user_callback)
  868. {
  869. ctrl->user_callback(event, ctrl->user_data);
  870. }
  871. }
  872. else
  873. {
  874. DBG_ERR("cb ctrl invaild %x", ctrl);
  875. DBG_HexPrintf(&ctrl->tag, 8);
  876. DBG_HexPrintf(&cb_param->tag, 8);
  877. }
  878. }
  879. else
  880. {
  881. for (i = 0; i < adapter->opt->max_socket_num; i++)
  882. {
  883. temp_event = *event;
  884. if (adapter->ctrl_busy[i])
  885. {
  886. ctrl = &adapter->ctrl_table[i];
  887. if (ctrl->adapter_index == (uint8_t)(event->Param3))
  888. {
  889. if (ctrl->auto_mode)
  890. {
  891. DBG("socket %d,%s,%s,%s", ctrl->socket_id, network_ctrl_event_id_string(event->ID),
  892. network_ctrl_state_string(ctrl->state),
  893. network_ctrl_wait_state_string(ctrl->wait_target_state));
  894. network_default_statemachine(ctrl, &temp_event, adapter);
  895. DBG("%s,%s",network_ctrl_state_string(ctrl->state), network_ctrl_wait_state_string(ctrl->wait_target_state));
  896. }
  897. else if (ctrl->task_handle)
  898. {
  899. platform_send_event(ctrl->task_handle, event->ID, event->Param1, event->Param2, event->Param3);
  900. }
  901. else if (ctrl->user_callback)
  902. {
  903. ctrl->user_callback(&temp_event, ctrl->user_data);
  904. }
  905. }
  906. }
  907. }
  908. }
  909. return 0;
  910. }
  911. static LUAT_RT_RET_TYPE network_default_timer_callback(LUAT_RT_CB_PARAM)
  912. {
  913. platform_send_event(param, EV_NW_TIMEOUT, 0, 0, 0);
  914. return LUAT_RT_RET;
  915. }
  916. int network_get_last_register_adapter(void)
  917. {
  918. if (prv_network.default_adapter_index != -1) return prv_network.default_adapter_index;
  919. return prv_network.last_adapter_index;
  920. }
  921. void network_register_set_default(uint8_t adapter_index)
  922. {
  923. prv_network.default_adapter_index = adapter_index;
  924. }
  925. int network_register_adapter(uint8_t adapter_index, network_adapter_info *info, void *user_data)
  926. {
  927. prv_adapter_table[adapter_index].opt = info;
  928. prv_adapter_table[adapter_index].user_data = user_data;
  929. info->socket_set_callback(network_default_socket_callback, &prv_adapter_table[adapter_index], user_data);
  930. if (adapter_index < NW_ADAPTER_INDEX_HW_PS_DEVICE)
  931. {
  932. prv_adapter_table[adapter_index].ctrl_table = prv_network.lwip_ctrl_table;
  933. prv_adapter_table[adapter_index].ctrl_busy = prv_network.lwip_ctrl_busy;
  934. }
  935. else
  936. {
  937. prv_adapter_table[adapter_index].ctrl_table = zalloc((info->max_socket_num) * sizeof(network_ctrl_t));
  938. prv_adapter_table[adapter_index].ctrl_busy = zalloc(info->max_socket_num);
  939. }
  940. prv_adapter_table[adapter_index].port = 60000;
  941. if (!prv_network.is_init)
  942. {
  943. //prv_network.network_mutex = platform_create_mutex();
  944. INIT_LLIST_HEAD(&prv_network.dns_cache_head);
  945. prv_network.is_init = 0;
  946. }
  947. prv_network.last_adapter_index = adapter_index;
  948. return 0;
  949. }
  950. void network_set_dns_server(uint8_t adapter_index, uint8_t server_index, luat_ip_addr_t *ip)
  951. {
  952. network_adapter_t *adapter = &prv_adapter_table[adapter_index];
  953. adapter->opt->set_dns_server(server_index, ip, adapter->user_data);
  954. }
  955. /*
  956. * 申请一个network_ctrl
  957. */
  958. network_ctrl_t *network_alloc_ctrl(uint8_t adapter_index)
  959. {
  960. int i;
  961. network_ctrl_t *ctrl = NULL;
  962. network_adapter_t *adapter = &prv_adapter_table[adapter_index];
  963. OS_LOCK;
  964. for (i = 0; i < adapter->opt->max_socket_num; i++)
  965. {
  966. if (!adapter->ctrl_busy[i])
  967. {
  968. adapter->ctrl_busy[i] = 1;
  969. ctrl = &adapter->ctrl_table[i];
  970. ctrl->adapter_index = adapter_index;
  971. ctrl->domain_ipv6 = 0;
  972. break;
  973. }
  974. }
  975. OS_UNLOCK;
  976. if (i >= adapter->opt->max_socket_num) {DBG_ERR("adapter no more ctrl!");}
  977. return ctrl;
  978. }
  979. /*
  980. * 归还一个network_ctrl
  981. */
  982. void network_release_ctrl(network_ctrl_t *ctrl)
  983. {
  984. int i;
  985. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  986. NW_UNLOCK;
  987. OS_LOCK;
  988. for (i = 0; i < adapter->opt->max_socket_num; i++)
  989. {
  990. if (&adapter->ctrl_table[i] == ctrl)
  991. {
  992. network_deinit_tls(ctrl);
  993. if (ctrl->timer)
  994. {
  995. platform_stop_timer(ctrl->timer);
  996. platform_release_timer(ctrl->timer);
  997. ctrl->timer = NULL;
  998. }
  999. if (ctrl->cache_data)
  1000. {
  1001. free(ctrl->cache_data);
  1002. ctrl->cache_data = NULL;
  1003. }
  1004. if (ctrl->dns_ip)
  1005. {
  1006. free(ctrl->dns_ip);
  1007. ctrl->dns_ip = NULL;
  1008. }
  1009. if (ctrl->domain_name)
  1010. {
  1011. free(ctrl->domain_name);
  1012. ctrl->domain_name = NULL;
  1013. }
  1014. adapter->ctrl_busy[i] = 0;
  1015. platform_release_mutex(ctrl->mutex);
  1016. ctrl->mutex = NULL;
  1017. break;
  1018. }
  1019. }
  1020. OS_UNLOCK;
  1021. if (i >= adapter->opt->max_socket_num) {DBG_ERR("adapter index maybe error!, %d, %x", ctrl->adapter_index, ctrl);}
  1022. }
  1023. void network_init_ctrl(network_ctrl_t *ctrl, HANDLE task_handle, CBFuncEx_t callback, void *param)
  1024. {
  1025. uint8_t adapter_index = ctrl->adapter_index;
  1026. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1027. if (ctrl->dns_ip)
  1028. {
  1029. free(ctrl->dns_ip);
  1030. }
  1031. if (ctrl->cache_data)
  1032. {
  1033. free(ctrl->cache_data);
  1034. }
  1035. if (ctrl->domain_name)
  1036. {
  1037. free(ctrl->domain_name);
  1038. }
  1039. HANDLE sem = ctrl->mutex;
  1040. memset(ctrl, 0, sizeof(network_ctrl_t));
  1041. ctrl->adapter_index = adapter_index;
  1042. ctrl->task_handle = task_handle;
  1043. ctrl->user_callback = callback;
  1044. ctrl->user_data = param;
  1045. ctrl->socket_id = -1;
  1046. ctrl->socket_param = ctrl;
  1047. ctrl->remote_ip.type = 0xff;
  1048. ctrl->mutex = sem;
  1049. if (task_handle)
  1050. {
  1051. ctrl->timer = platform_create_timer(network_default_timer_callback, task_handle, NULL);
  1052. }
  1053. if (!ctrl->mutex)
  1054. {
  1055. ctrl->mutex = platform_create_mutex();
  1056. }
  1057. }
  1058. 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)
  1059. {
  1060. ctrl->is_tcp = is_tcp;
  1061. ctrl->tcp_keep_alive = keep_alive;
  1062. ctrl->tcp_keep_idle = keep_idle;
  1063. ctrl->tcp_keep_interval = keep_interval;
  1064. ctrl->tcp_keep_cnt = keep_cnt;
  1065. ctrl->tcp_timeout_ms = tcp_timeout_ms;
  1066. }
  1067. void network_connect_ipv6_domain(network_ctrl_t *ctrl, uint8_t onoff)
  1068. {
  1069. ctrl->domain_ipv6 = onoff;
  1070. }
  1071. int network_set_local_port(network_ctrl_t *ctrl, uint16_t local_port)
  1072. {
  1073. int i;
  1074. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1075. if (local_port)
  1076. {
  1077. OS_LOCK;
  1078. for (i = 0; i < adapter->opt->max_socket_num; i++)
  1079. {
  1080. if (&adapter->ctrl_table[i] != ctrl)
  1081. {
  1082. if (adapter->ctrl_table[i].local_port == local_port)
  1083. {
  1084. OS_UNLOCK;
  1085. return -1;
  1086. }
  1087. }
  1088. }
  1089. ctrl->local_port = local_port;
  1090. OS_UNLOCK;
  1091. return 0;
  1092. }
  1093. else
  1094. {
  1095. // OS_LOCK;
  1096. // adapter->port++;
  1097. // if (adapter->port < 60000)
  1098. // {
  1099. // adapter->port = 60000;
  1100. // }
  1101. // ctrl->local_port = adapter->port;
  1102. // OS_UNLOCK;
  1103. ctrl->local_port = 0;
  1104. return 0;
  1105. }
  1106. }
  1107. uint8_t network_check_ready(network_ctrl_t *ctrl)
  1108. {
  1109. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1110. return adapter->opt->check_ready(adapter->user_data);
  1111. }
  1112. int network_create_soceket(network_ctrl_t *ctrl, uint8_t is_ipv6)
  1113. {
  1114. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1115. ctrl->socket_id = adapter->opt->create_soceket(ctrl->is_tcp, &ctrl->tag, ctrl->socket_param, is_ipv6, adapter->user_data);
  1116. if (ctrl->socket_id < 0)
  1117. {
  1118. ctrl->tag = 0;
  1119. return -1;
  1120. }
  1121. return 0;
  1122. }
  1123. int network_socket_connect(network_ctrl_t *ctrl, luat_ip_addr_t *remote_ip)
  1124. {
  1125. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1126. ctrl->is_server_mode = 0;
  1127. ctrl->online_ip = remote_ip;
  1128. return adapter->opt->socket_connect(ctrl->socket_id, ctrl->tag, ctrl->local_port, remote_ip, ctrl->remote_port, adapter->user_data);
  1129. }
  1130. int network_socket_listen(network_ctrl_t *ctrl)
  1131. {
  1132. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1133. ctrl->is_server_mode = 1;
  1134. return adapter->opt->socket_listen(ctrl->socket_id, ctrl->tag, ctrl->local_port, adapter->user_data);
  1135. }
  1136. uint8_t network_accept_enable(network_ctrl_t *ctrl)
  1137. {
  1138. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1139. return !(adapter->opt->no_accept);
  1140. }
  1141. //作为server接受一个client
  1142. //成功返回0,失败 < 0
  1143. int network_socket_accept(network_ctrl_t *ctrl, network_ctrl_t *accept_ctrl)
  1144. {
  1145. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1146. if (adapter->opt->no_accept)
  1147. {
  1148. // 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);
  1149. adapter->opt->socket_accept(ctrl->socket_id, ctrl->tag, &ctrl->remote_ip, &ctrl->remote_port, adapter->user_data);
  1150. return 0;
  1151. }
  1152. accept_ctrl->socket_id = adapter->opt->socket_accept(ctrl->socket_id, ctrl->tag, &accept_ctrl->remote_ip, &accept_ctrl->remote_port, adapter->user_data);
  1153. if (accept_ctrl->socket_id < 0)
  1154. {
  1155. return -1;
  1156. }
  1157. else
  1158. {
  1159. accept_ctrl->is_tcp = ctrl->is_tcp;
  1160. accept_ctrl->tcp_keep_alive = ctrl->tcp_keep_alive;
  1161. accept_ctrl->tcp_keep_idle = ctrl->tcp_keep_idle;
  1162. accept_ctrl->tcp_keep_interval = ctrl->tcp_keep_interval;
  1163. accept_ctrl->tcp_keep_cnt = ctrl->tcp_keep_cnt;
  1164. accept_ctrl->tcp_timeout_ms = ctrl->tcp_timeout_ms;
  1165. accept_ctrl->local_port = ctrl->local_port;
  1166. accept_ctrl->state = NW_STATE_ONLINE;
  1167. return 0;
  1168. }
  1169. }
  1170. //主动断开一个tcp连接,需要走完整个tcp流程,用户需要接收到close ok回调才能确认彻底断开
  1171. //成功返回0,失败 < 0
  1172. int network_socket_disconnect(network_ctrl_t *ctrl)
  1173. {
  1174. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1175. if (ctrl->socket_id >= 0)
  1176. {
  1177. return adapter->opt->socket_disconnect(ctrl->socket_id, ctrl->tag, adapter->user_data);
  1178. }
  1179. return 0;
  1180. }
  1181. //释放掉socket的控制权
  1182. //成功返回0,失败 < 0
  1183. int network_socket_close(network_ctrl_t *ctrl)
  1184. {
  1185. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1186. if (ctrl->socket_id >= 0)
  1187. {
  1188. return adapter->opt->socket_close(ctrl->socket_id, ctrl->tag, adapter->user_data);
  1189. }
  1190. return 0;
  1191. }
  1192. //强行释放掉socket的控制权
  1193. //成功返回0,失败 < 0
  1194. int network_socket_force_close(network_ctrl_t *ctrl)
  1195. {
  1196. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1197. if (ctrl->socket_id >= 0)
  1198. {
  1199. adapter->opt->socket_force_close(ctrl->socket_id, adapter->user_data);
  1200. }
  1201. ctrl->socket_id = -1;
  1202. return 0;
  1203. }
  1204. //tcp时,不需要remote_ip和remote_port,如果buf为NULL,则返回当前缓存区的数据量,当返回值小于len时说明已经读完了
  1205. //udp时,只返回1个block数据,需要多次读直到没有数据为止
  1206. //成功返回实际读取的值,失败 < 0
  1207. 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)
  1208. {
  1209. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1210. return adapter->opt->socket_receive(ctrl->socket_id, ctrl->tag, buf, len, flags, remote_ip, remote_port, adapter->user_data);
  1211. }
  1212. //tcp时,不需要remote_ip和remote_port
  1213. //成功返回0,失败 < 0
  1214. 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)
  1215. {
  1216. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1217. return adapter->opt->socket_send(ctrl->socket_id, ctrl->tag, buf, len, flags, remote_ip, remote_port, adapter->user_data);
  1218. }
  1219. int network_getsockopt(network_ctrl_t *ctrl, int level, int optname, void *optval, uint32_t *optlen)
  1220. {
  1221. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1222. return adapter->opt->getsockopt(ctrl->socket_id, ctrl->tag, level, optname, optval, optlen, adapter->user_data);
  1223. }
  1224. int network_setsockopt(network_ctrl_t *ctrl, int level, int optname, const void *optval, uint32_t optlen)
  1225. {
  1226. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1227. return adapter->opt->setsockopt(ctrl->socket_id, ctrl->tag, level, optname, optval, optlen, adapter->user_data);
  1228. }
  1229. //非posix的socket,用这个根据实际硬件设置参数
  1230. int network_user_cmd(network_ctrl_t *ctrl, uint32_t cmd, uint32_t value)
  1231. {
  1232. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1233. return adapter->opt->user_cmd(ctrl->socket_id, ctrl->tag, cmd, value, adapter->user_data);
  1234. }
  1235. int network_dns(network_ctrl_t *ctrl)
  1236. {
  1237. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1238. if (ctrl->domain_ipv6)
  1239. {
  1240. return adapter->opt->dns_ipv6(ctrl->domain_name, ctrl->domain_name_len, ctrl, adapter->user_data);
  1241. }
  1242. else
  1243. {
  1244. return adapter->opt->dns(ctrl->domain_name, ctrl->domain_name_len, ctrl, adapter->user_data);
  1245. }
  1246. }
  1247. int network_set_mac(uint8_t adapter_index, uint8_t *mac)
  1248. {
  1249. network_adapter_t *adapter = &prv_adapter_table[adapter_index];
  1250. return adapter->opt->set_mac(mac, adapter->user_data);
  1251. }
  1252. 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)
  1253. {
  1254. network_adapter_t *adapter = &prv_adapter_table[adapter_index];
  1255. return adapter->opt->set_static_ip(ip, submask, gateway, ipv6, adapter->user_data);
  1256. }
  1257. 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)
  1258. {
  1259. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1260. return adapter->opt->get_local_ip_info(ip, submask, gateway, adapter->user_data);
  1261. }
  1262. 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)
  1263. {
  1264. network_adapter_t *adapter;
  1265. if (ctrl)
  1266. {
  1267. adapter = &prv_adapter_table[ctrl->adapter_index];
  1268. }
  1269. else
  1270. {
  1271. adapter = &prv_adapter_table[index];
  1272. }
  1273. return adapter->opt->get_full_ip_info(ip, submask, gateway, ipv6, adapter->user_data);
  1274. }
  1275. void network_force_close_socket(network_ctrl_t *ctrl)
  1276. {
  1277. #ifdef LUAT_USE_TLS
  1278. if (ctrl->tls_mode)
  1279. {
  1280. mbedtls_ssl_free(ctrl->ssl);
  1281. }
  1282. #endif
  1283. if (network_socket_close(ctrl))
  1284. {
  1285. network_clean_invaild_socket(ctrl->adapter_index);
  1286. network_socket_force_close(ctrl);
  1287. }
  1288. ctrl->need_close = 0;
  1289. ctrl->socket_id = -1;
  1290. if (ctrl->dns_ip)
  1291. {
  1292. free(ctrl->dns_ip);
  1293. ctrl->dns_ip = NULL;
  1294. }
  1295. if (ctrl->domain_name)
  1296. {
  1297. free(ctrl->domain_name);
  1298. ctrl->domain_name = NULL;
  1299. }
  1300. ctrl->dns_ip_cnt = 0;
  1301. ctrl->dns_ip_nums = 0;
  1302. }
  1303. void network_clean_invaild_socket(uint8_t adapter_index)
  1304. {
  1305. int i;
  1306. int *list;
  1307. network_adapter_t *adapter = &prv_adapter_table[adapter_index];
  1308. network_ctrl_t *ctrl;
  1309. list = malloc(adapter->opt->max_socket_num * sizeof(int));
  1310. OS_LOCK;
  1311. for (i = 0; i < adapter->opt->max_socket_num; i++)
  1312. {
  1313. ctrl = &adapter->ctrl_table[i];
  1314. if (!adapter->opt->socket_check(ctrl->socket_id, ctrl->tag, adapter->user_data))
  1315. {
  1316. list[i] = ctrl->socket_id;
  1317. }
  1318. else
  1319. {
  1320. ctrl->socket_id = -1;
  1321. list[i] = -1;
  1322. }
  1323. DBG("%d,%d", i, list[i]);
  1324. }
  1325. OS_UNLOCK;
  1326. adapter->opt->socket_clean(list, adapter->opt->max_socket_num, adapter->user_data);
  1327. free(list);
  1328. }
  1329. #ifdef LUAT_USE_TLS
  1330. static int tls_verify(void *ctx, mbedtls_x509_crt *crt, int Index, uint32_t *result)
  1331. {
  1332. network_ctrl_t *ctrl = (network_ctrl_t *)ctx;
  1333. DBG("%d, %08x", Index, *result);
  1334. return 0;
  1335. }
  1336. #endif
  1337. int network_set_psk_info(network_ctrl_t *ctrl,
  1338. const unsigned char *psk, size_t psk_len,
  1339. const unsigned char *psk_identity, size_t psk_identity_len)
  1340. {
  1341. #ifdef LUAT_USE_TLS
  1342. if (!ctrl->tls_mode)
  1343. {
  1344. return -ERROR_PERMISSION_DENIED;
  1345. }
  1346. // DBG("%.*s, %.*s", psk_len, psk, psk_identity_len, psk_identity);
  1347. int ret = mbedtls_ssl_conf_psk( ctrl->config,
  1348. psk, psk_len, psk_identity, psk_identity_len );
  1349. if (ret != 0)
  1350. {
  1351. DBG("0x%x", -ret);
  1352. return -ERROR_OPERATION_FAILED;
  1353. }
  1354. return ERROR_NONE;
  1355. #else
  1356. return -1;
  1357. #endif
  1358. }
  1359. int network_set_server_cert(network_ctrl_t *ctrl, const unsigned char *cert, size_t cert_len)
  1360. {
  1361. #ifdef LUAT_USE_TLS
  1362. int ret;
  1363. if (!ctrl->tls_mode)
  1364. {
  1365. return -ERROR_PERMISSION_DENIED;
  1366. }
  1367. ret = mbedtls_x509_crt_parse( ctrl->ca_cert, cert, cert_len);
  1368. if (ret != 0)
  1369. {
  1370. DBG("%08x", -ret);
  1371. return -ERROR_OPERATION_FAILED;
  1372. }
  1373. return ERROR_NONE;
  1374. #else
  1375. return -1;
  1376. #endif
  1377. }
  1378. int network_set_client_cert(network_ctrl_t *ctrl,
  1379. const unsigned char *cert, size_t certLen,
  1380. const unsigned char *key, size_t keylen,
  1381. const unsigned char *pwd, size_t pwdlen)
  1382. {
  1383. #ifdef LUAT_USE_TLS
  1384. int ret;
  1385. mbedtls_x509_crt *client_cert = NULL;
  1386. mbedtls_pk_context *pkey = NULL;
  1387. if (!ctrl->tls_mode)
  1388. {
  1389. return -ERROR_PERMISSION_DENIED;
  1390. }
  1391. client_cert = zalloc(sizeof(mbedtls_x509_crt));
  1392. pkey = zalloc(sizeof(mbedtls_pk_context));
  1393. if (!client_cert || !pkey)
  1394. {
  1395. goto ERROR_OUT;
  1396. }
  1397. ret = mbedtls_x509_crt_parse( client_cert, cert, certLen );
  1398. if (ret != 0)
  1399. {
  1400. DBG("%08x", -ret);
  1401. goto ERROR_OUT;
  1402. }
  1403. #if MBEDTLS_VERSION_NUMBER >= 0x03000000
  1404. ret = mbedtls_pk_parse_key( pkey, key, keylen, pwd, pwdlen , tls_random, NULL);
  1405. #else
  1406. ret = mbedtls_pk_parse_key( pkey, key, keylen, pwd, pwdlen );
  1407. #endif
  1408. if (ret != 0)
  1409. {
  1410. DBG("%08x", -ret);
  1411. goto ERROR_OUT;
  1412. }
  1413. ret = mbedtls_ssl_conf_own_cert( ctrl->config, client_cert, pkey );
  1414. if (ret != 0)
  1415. {
  1416. DBG("%08x", -ret);
  1417. goto ERROR_OUT;
  1418. }
  1419. return ERROR_NONE;
  1420. ERROR_OUT:
  1421. if (client_cert) free(client_cert);
  1422. if (pkey) free(pkey);
  1423. return -1;
  1424. #else
  1425. return -1;
  1426. #endif
  1427. }
  1428. int network_cert_verify_result(network_ctrl_t *ctrl)
  1429. {
  1430. #ifdef LUAT_USE_TLS
  1431. if (!ctrl->tls_mode)
  1432. {
  1433. return -1;
  1434. }
  1435. return mbedtls_ssl_get_verify_result(ctrl->ssl);
  1436. #else
  1437. return -1;
  1438. #endif
  1439. }
  1440. static int tls_random( void *p_rng,
  1441. unsigned char *output, size_t output_len )
  1442. {
  1443. platform_random((char*)output, output_len);
  1444. return 0;
  1445. }
  1446. void network_init_tls(network_ctrl_t *ctrl, int verify_mode)
  1447. {
  1448. #ifdef LUAT_USE_TLS
  1449. ctrl->tls_mode = 1;
  1450. if (!ctrl->ssl)
  1451. {
  1452. ctrl->ssl = zalloc(sizeof(mbedtls_ssl_context));
  1453. ctrl->ca_cert = zalloc(sizeof(mbedtls_x509_crt));
  1454. ctrl->config = zalloc(sizeof(mbedtls_ssl_config));
  1455. 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);
  1456. // ctrl->config->authmode = verify_mode;
  1457. mbedtls_ssl_conf_authmode(ctrl->config, verify_mode);
  1458. // ctrl->config->hs_timeout_min = 20000;
  1459. #if defined(MBEDTLS_SSL_PROTO_DTLS)
  1460. mbedtls_ssl_conf_handshake_timeout(ctrl->config, 2000, MBEDTLS_SSL_DTLS_TIMEOUT_DFL_MAX);
  1461. #endif
  1462. // ctrl->config->f_rng = tls_random;
  1463. // ctrl->config->p_rng = NULL;
  1464. mbedtls_ssl_conf_rng(ctrl->config, tls_random, NULL);
  1465. // ctrl->config->f_dbg = tls_dbg;
  1466. // ctrl->config->p_dbg = NULL;
  1467. mbedtls_ssl_conf_dbg(ctrl->config, tls_dbg, NULL);
  1468. // ctrl->config->f_vrfy = tls_verify;
  1469. // ctrl->config->p_vrfy = ctrl;
  1470. mbedtls_ssl_conf_verify(ctrl->config, tls_verify, ctrl);
  1471. // ctrl->config->ca_chain = ctrl->ca_cert;
  1472. mbedtls_ssl_conf_ca_chain(ctrl->config, ctrl->ca_cert, NULL);
  1473. // ctrl->config->read_timeout = 20000;
  1474. mbedtls_ssl_conf_read_timeout(ctrl->config, 20000);
  1475. ctrl->tls_long_timer = platform_create_timer(tls_longtimeout, ctrl, NULL);
  1476. ctrl->tls_short_timer = platform_create_timer(tls_shorttimeout, ctrl, NULL);
  1477. }
  1478. ctrl->tls_timer_state = -1;
  1479. #endif
  1480. }
  1481. void network_deinit_tls(network_ctrl_t *ctrl)
  1482. {
  1483. #ifdef LUAT_USE_TLS
  1484. if (ctrl->ssl)
  1485. {
  1486. mbedtls_ssl_free(ctrl->ssl);
  1487. free(ctrl->ssl);
  1488. ctrl->ssl = NULL;
  1489. }
  1490. if (ctrl->config)
  1491. {
  1492. mbedtls_ssl_config_free(ctrl->config);
  1493. free(ctrl->config);
  1494. ctrl->config = NULL;
  1495. }
  1496. if (ctrl->ca_cert)
  1497. {
  1498. mbedtls_x509_crt_free(ctrl->ca_cert);
  1499. free(ctrl->ca_cert);
  1500. ctrl->ca_cert = NULL;
  1501. }
  1502. ctrl->tls_mode = 0;
  1503. ctrl->tls_timer_state = -1;
  1504. if (ctrl->tls_short_timer)
  1505. {
  1506. platform_release_timer(ctrl->tls_short_timer);
  1507. ctrl->tls_short_timer = NULL;
  1508. }
  1509. if (ctrl->tls_long_timer)
  1510. {
  1511. platform_release_timer(ctrl->tls_long_timer);
  1512. ctrl->tls_long_timer = NULL;
  1513. }
  1514. #endif
  1515. }
  1516. int network_wait_link_up(network_ctrl_t *ctrl, uint32_t timeout_ms)
  1517. {
  1518. NW_LOCK;
  1519. ctrl->auto_mode = 1;
  1520. // network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1521. if (network_check_ready(ctrl))
  1522. {
  1523. ctrl->state = NW_STATE_OFF_LINE;
  1524. ctrl->wait_target_state = NW_WAIT_NONE;
  1525. NW_UNLOCK;
  1526. return 0;
  1527. }
  1528. ctrl->state = NW_STATE_LINK_OFF;
  1529. ctrl->wait_target_state = NW_WAIT_LINK_UP;
  1530. NW_UNLOCK;
  1531. if (!ctrl->task_handle || !timeout_ms)
  1532. {
  1533. return 1;
  1534. }
  1535. uint8_t finish = 0;
  1536. OS_EVENT event;
  1537. int result;
  1538. //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
  1539. platform_start_timer(ctrl->timer, timeout_ms, 0);
  1540. while (!finish)
  1541. {
  1542. platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
  1543. switch (event.ID)
  1544. {
  1545. case EV_NW_RESULT_LINK:
  1546. result = (int)event.Param1;
  1547. finish = 1;
  1548. break;
  1549. case EV_NW_TIMEOUT:
  1550. result = -1;
  1551. finish = 1;
  1552. break;
  1553. default:
  1554. if (ctrl->user_callback)
  1555. {
  1556. ctrl->user_callback((void *)&event, ctrl->user_data);
  1557. }
  1558. break;
  1559. }
  1560. }
  1561. platform_stop_timer(ctrl->timer);
  1562. return result;
  1563. }
  1564. /*
  1565. * 1.进行ready检测和等待ready
  1566. * 2.有remote_ip则开始连接服务器并等待连接结果
  1567. * 3.没有remote_ip则开始对url进行dns解析,解析完成后对所有ip进行尝试连接直到有个成功或者全部失败
  1568. * 4.如果是加密模式,还要走握手环节,等到握手环节完成后才能返回结果
  1569. * local_port如果为0则api内部自动生成一个
  1570. */
  1571. 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)
  1572. {
  1573. if (ctrl->socket_id >= 0)
  1574. {
  1575. return -1;
  1576. }
  1577. NW_LOCK;
  1578. ctrl->is_server_mode = 0;
  1579. ctrl->tx_size = 0;
  1580. ctrl->ack_size = 0;
  1581. if (ctrl->dns_ip)
  1582. {
  1583. free(ctrl->dns_ip);
  1584. ctrl->dns_ip = NULL;
  1585. }
  1586. if (ctrl->cache_data)
  1587. {
  1588. free(ctrl->cache_data);
  1589. ctrl->cache_data = NULL;
  1590. }
  1591. ctrl->need_close = 0;
  1592. if (ctrl->domain_name)
  1593. {
  1594. free(ctrl->domain_name);
  1595. }
  1596. ctrl->domain_name = zalloc(domain_name_len + 1);
  1597. memcpy(ctrl->domain_name, domain_name, domain_name_len);
  1598. ctrl->domain_name_len = domain_name_len;
  1599. if (remote_ip)
  1600. {
  1601. ctrl->remote_ip = *remote_ip;
  1602. }
  1603. else
  1604. {
  1605. ctrl->remote_ip.type = 0xff;
  1606. }
  1607. ctrl->auto_mode = 1;
  1608. ctrl->remote_port = remote_port;
  1609. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1610. ctrl->wait_target_state = NW_WAIT_ON_LINE;
  1611. if (!network_check_ready(ctrl))
  1612. {
  1613. ctrl->state = NW_STATE_LINK_OFF;
  1614. goto NETWORK_CONNECT_WAIT;
  1615. }
  1616. if (network_prepare_connect(ctrl))
  1617. {
  1618. ctrl->state = NW_STATE_OFF_LINE;
  1619. ctrl->wait_target_state = NW_WAIT_NONE;
  1620. NW_UNLOCK;
  1621. return -1;
  1622. }
  1623. NETWORK_CONNECT_WAIT:
  1624. NW_UNLOCK;
  1625. if (!ctrl->task_handle || !timeout_ms)
  1626. {
  1627. return 1;
  1628. }
  1629. uint8_t finish = 0;
  1630. OS_EVENT event;
  1631. int result;
  1632. //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
  1633. platform_start_timer(ctrl->timer, timeout_ms, 0);
  1634. while (!finish)
  1635. {
  1636. platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
  1637. switch (event.ID)
  1638. {
  1639. case EV_NW_RESULT_CONNECT:
  1640. result = (int)event.Param1;
  1641. finish = 1;
  1642. break;
  1643. case EV_NW_TIMEOUT:
  1644. result = -1;
  1645. finish = 1;
  1646. break;
  1647. default:
  1648. if (ctrl->user_callback)
  1649. {
  1650. ctrl->user_callback((void *)&event, ctrl->user_data);
  1651. }
  1652. break;
  1653. }
  1654. }
  1655. platform_stop_timer(ctrl->timer);
  1656. return result;
  1657. }
  1658. int network_listen(network_ctrl_t *ctrl, uint32_t timeout_ms)
  1659. {
  1660. if (NW_STATE_LISTEN == ctrl->state)
  1661. {
  1662. DBG("socket %d is listen", ctrl->socket_id);
  1663. return 0;
  1664. }
  1665. if (ctrl->socket_id >= 0)
  1666. {
  1667. return -1;
  1668. }
  1669. NW_LOCK;
  1670. ctrl->is_server_mode = 1;
  1671. ctrl->auto_mode = 1;
  1672. ctrl->need_close = 0;
  1673. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1674. ctrl->wait_target_state = NW_WAIT_ON_LINE;
  1675. if (!network_check_ready(ctrl))
  1676. {
  1677. ctrl->state = NW_STATE_LINK_OFF;
  1678. goto NETWORK_LISTEN_WAIT;
  1679. }
  1680. if (network_base_connect(ctrl, NULL))
  1681. {
  1682. ctrl->state = NW_STATE_OFF_LINE;
  1683. ctrl->wait_target_state = NW_WAIT_NONE;
  1684. NW_UNLOCK;
  1685. return -1;
  1686. }
  1687. ctrl->state = NW_STATE_CONNECTING;
  1688. NETWORK_LISTEN_WAIT:
  1689. NW_UNLOCK;
  1690. if (!ctrl->task_handle || !timeout_ms)
  1691. {
  1692. return 1;
  1693. }
  1694. uint8_t finish = 0;
  1695. OS_EVENT event;
  1696. int result;
  1697. //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
  1698. if (timeout_ms != 0xffffffff)
  1699. {
  1700. platform_start_timer(ctrl->timer, timeout_ms, 0);
  1701. }
  1702. while (!finish)
  1703. {
  1704. platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
  1705. switch (event.ID)
  1706. {
  1707. case EV_NW_RESULT_CONNECT:
  1708. result = (int)event.Param1;
  1709. finish = 1;
  1710. break;
  1711. case EV_NW_TIMEOUT:
  1712. result = -1;
  1713. finish = 1;
  1714. break;
  1715. default:
  1716. if (ctrl->user_callback)
  1717. {
  1718. ctrl->user_callback((void *)&event, ctrl->user_data);
  1719. }
  1720. break;
  1721. }
  1722. }
  1723. if (timeout_ms != 0xffffffff)
  1724. {
  1725. platform_stop_timer(ctrl->timer);
  1726. }
  1727. return result;
  1728. }
  1729. int network_close(network_ctrl_t *ctrl, uint32_t timeout_ms)
  1730. {
  1731. NW_LOCK;
  1732. if (ctrl->cache_data)
  1733. {
  1734. free(ctrl->cache_data);
  1735. ctrl->cache_data = NULL;
  1736. }
  1737. uint8_t old_state = ctrl->state;
  1738. ctrl->auto_mode = 1;
  1739. ctrl->need_close = 0;
  1740. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  1741. #ifdef LUAT_USE_TLS
  1742. if (ctrl->tls_mode)
  1743. {
  1744. mbedtls_ssl_free(ctrl->ssl);
  1745. }
  1746. #endif
  1747. if (ctrl->socket_id < 0)
  1748. {
  1749. ctrl->state = NW_STATE_OFF_LINE;
  1750. ctrl->wait_target_state = NW_WAIT_NONE;
  1751. NW_UNLOCK;
  1752. return 0;
  1753. }
  1754. ctrl->state = NW_STATE_DISCONNECTING;
  1755. ctrl->wait_target_state = NW_WAIT_OFF_LINE;
  1756. if ((NW_STATE_ONLINE == old_state) && ctrl->is_tcp)
  1757. {
  1758. if (network_socket_disconnect(ctrl))
  1759. {
  1760. network_force_close_socket(ctrl);
  1761. ctrl->state = NW_STATE_OFF_LINE;
  1762. ctrl->wait_target_state = NW_WAIT_NONE;
  1763. NW_UNLOCK;
  1764. return 0;
  1765. }
  1766. }
  1767. else
  1768. {
  1769. network_force_close_socket(ctrl);
  1770. ctrl->state = NW_STATE_OFF_LINE;
  1771. ctrl->wait_target_state = NW_WAIT_NONE;
  1772. NW_UNLOCK;
  1773. return 0;
  1774. }
  1775. NW_UNLOCK;
  1776. if (!ctrl->task_handle || !timeout_ms)
  1777. {
  1778. return 1;
  1779. }
  1780. uint8_t finish = 0;
  1781. OS_EVENT event;
  1782. int result;
  1783. //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
  1784. platform_start_timer(ctrl->timer, timeout_ms, 0);
  1785. while (!finish)
  1786. {
  1787. platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
  1788. switch (event.ID)
  1789. {
  1790. case EV_NW_RESULT_CLOSE:
  1791. result = 0;
  1792. finish = 1;
  1793. break;
  1794. case EV_NW_TIMEOUT:
  1795. result = 0;
  1796. finish = 1;
  1797. break;
  1798. default:
  1799. if (ctrl->user_callback)
  1800. {
  1801. ctrl->user_callback((void *)&event, ctrl->user_data);
  1802. }
  1803. break;
  1804. }
  1805. }
  1806. platform_stop_timer(ctrl->timer);
  1807. network_socket_force_close(ctrl);
  1808. return result;
  1809. }
  1810. /*
  1811. * timeout_ms=0时,为非阻塞接口
  1812. */
  1813. 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)
  1814. {
  1815. if ((ctrl->need_close) || (ctrl->socket_id < 0) || (ctrl->state != NW_STATE_ONLINE))
  1816. {
  1817. return -1;
  1818. }
  1819. NW_LOCK;
  1820. int result;
  1821. ctrl->auto_mode = 1;
  1822. #ifdef LUAT_USE_TLS
  1823. if (ctrl->tls_mode)
  1824. {
  1825. if (ctrl->tls_need_reshakehand)
  1826. {
  1827. ctrl->tls_need_reshakehand = 0;
  1828. if (ctrl->cache_data)
  1829. {
  1830. free(ctrl->cache_data);
  1831. ctrl->cache_data = NULL;
  1832. }
  1833. ctrl->cache_data = malloc(len);
  1834. memcpy(ctrl->cache_data, data, len);
  1835. ctrl->cache_len = len;
  1836. mbedtls_ssl_session_reset(ctrl->ssl);
  1837. do
  1838. {
  1839. result = mbedtls_ssl_handshake_step( ctrl->ssl );
  1840. switch(result)
  1841. {
  1842. case MBEDTLS_ERR_SSL_WANT_READ:
  1843. ctrl->state = NW_STATE_SHAKEHAND;
  1844. goto NETWORK_TX_WAIT;
  1845. case 0:
  1846. break;
  1847. default:
  1848. #if MBEDTLS_VERSION_NUMBER >= 0x03000000
  1849. #else
  1850. DBG_ERR("0x%x, %d", -result, ctrl->ssl->state);
  1851. #endif
  1852. ctrl->need_close = 1;
  1853. NW_UNLOCK;
  1854. return -1;
  1855. }
  1856. #if MBEDTLS_VERSION_NUMBER >= 0x03000000
  1857. }while(!mbedtls_ssl_is_handshake_over(ctrl->ssl));
  1858. #else
  1859. }while(ctrl->ssl->state != MBEDTLS_SSL_HANDSHAKE_OVER);
  1860. #endif
  1861. }
  1862. result = mbedtls_ssl_write(ctrl->ssl, data, len);
  1863. if (result < 0)
  1864. {
  1865. DBG("%08x", -result);
  1866. ctrl->need_close = 1;
  1867. NW_UNLOCK;
  1868. return -1;
  1869. }
  1870. *tx_len = result;
  1871. }
  1872. else
  1873. #endif
  1874. {
  1875. result = network_base_tx(ctrl, data, len, flags, remote_ip, remote_port);
  1876. if (result < 0)
  1877. {
  1878. ctrl->need_close = 1;
  1879. NW_UNLOCK;
  1880. return -1;
  1881. }
  1882. *tx_len = result;
  1883. if (!result && len)
  1884. {
  1885. NW_UNLOCK;
  1886. return 0;
  1887. }
  1888. }
  1889. #ifdef LUAT_USE_TLS
  1890. NETWORK_TX_WAIT:
  1891. #endif
  1892. ctrl->wait_target_state = NW_WAIT_TX_OK;
  1893. NW_UNLOCK;
  1894. if (!ctrl->task_handle || !timeout_ms)
  1895. {
  1896. return 1;
  1897. }
  1898. uint8_t finish = 0;
  1899. OS_EVENT event;
  1900. //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
  1901. platform_start_timer(ctrl->timer, timeout_ms, 0);
  1902. while (!finish)
  1903. {
  1904. platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
  1905. switch (event.ID)
  1906. {
  1907. case EV_NW_RESULT_TX:
  1908. result = (int)event.Param1;
  1909. finish = 1;
  1910. break;
  1911. case EV_NW_TIMEOUT:
  1912. result = -1;
  1913. finish = 1;
  1914. break;
  1915. default:
  1916. if (ctrl->user_callback)
  1917. {
  1918. ctrl->user_callback((void *)&event, ctrl->user_data);
  1919. }
  1920. break;
  1921. }
  1922. }
  1923. platform_stop_timer(ctrl->timer);
  1924. return result;
  1925. }
  1926. /*
  1927. * 实际读到的数据量在rx_len里,如果是UDP模式且为server时,需要看remote_ip和remote_port
  1928. */
  1929. 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)
  1930. {
  1931. if (!ctrl->new_rx_flag && ((ctrl->need_close) || (ctrl->socket_id < 0) || (ctrl->state != NW_STATE_ONLINE)))
  1932. {
  1933. return -1;
  1934. }
  1935. NW_LOCK;
  1936. int result = -1;
  1937. ctrl->auto_mode = 1;
  1938. uint32_t read_len = 0;
  1939. uint8_t is_error = 0;
  1940. if (data)
  1941. {
  1942. ctrl->new_rx_flag = 0;
  1943. #ifdef LUAT_USE_TLS
  1944. if (ctrl->tls_mode)
  1945. {
  1946. do
  1947. {
  1948. result = mbedtls_ssl_read(ctrl->ssl, data + read_len, len - read_len);
  1949. if (result < 0 && (result != MBEDTLS_ERR_SSL_WANT_READ))
  1950. {
  1951. is_error = 1;
  1952. break;
  1953. }
  1954. else
  1955. {
  1956. read_len += result;
  1957. if (read_len >= len)
  1958. {
  1959. break;
  1960. }
  1961. }
  1962. }while(network_socket_receive(ctrl, NULL, len, flags, remote_ip, remote_port) > 0);
  1963. if ( !is_error )
  1964. {
  1965. result = read_len;
  1966. }
  1967. else
  1968. {
  1969. result = -1;
  1970. }
  1971. }
  1972. else
  1973. #endif
  1974. {
  1975. result = network_socket_receive(ctrl, data, len, flags, remote_ip, remote_port);
  1976. }
  1977. }
  1978. else
  1979. {
  1980. #ifdef LUAT_USE_TLS
  1981. if (ctrl->tls_mode)
  1982. {
  1983. read_len = 0;
  1984. do
  1985. {
  1986. result = mbedtls_ssl_read(ctrl->ssl, NULL, 0);
  1987. if ((result < 0) && (result != (MBEDTLS_ERR_SSL_WANT_READ)))
  1988. {
  1989. is_error = 1;
  1990. read_len = 0;
  1991. break;
  1992. }
  1993. else if (!result)
  1994. {
  1995. #if MBEDTLS_VERSION_NUMBER >= 0x03000000
  1996. read_len = ctrl->ssl->MBEDTLS_PRIVATE(in_msglen);
  1997. #else
  1998. read_len = ctrl->ssl->in_msglen;
  1999. #endif
  2000. break;
  2001. }
  2002. else if ((MBEDTLS_ERR_SSL_WANT_READ) == result)
  2003. {
  2004. read_len = 0;
  2005. ctrl->new_rx_flag = 0;
  2006. DBG("socket %d ssl data need more", ctrl->socket_id);
  2007. break;
  2008. }
  2009. }while(network_socket_receive(ctrl, NULL, len, flags, remote_ip, remote_port) > 0);
  2010. if ( !is_error )
  2011. {
  2012. result = read_len;
  2013. }
  2014. else
  2015. {
  2016. result = -1;
  2017. }
  2018. }
  2019. else
  2020. #endif
  2021. {
  2022. result = network_socket_receive(ctrl, data, len, flags, remote_ip, remote_port);
  2023. }
  2024. }
  2025. NW_UNLOCK;
  2026. if (result >= 0)
  2027. {
  2028. *rx_len = result;
  2029. return 0;
  2030. }
  2031. else
  2032. {
  2033. return -1;
  2034. }
  2035. }
  2036. int network_wait_event(network_ctrl_t *ctrl, OS_EVENT *out_event, uint32_t timeout_ms, uint8_t *is_timeout)
  2037. {
  2038. if (ctrl->new_rx_flag)
  2039. {
  2040. ctrl->wait_target_state = NW_WAIT_EVENT;
  2041. if (out_event)
  2042. {
  2043. out_event->ID = 0;
  2044. }
  2045. return 0;
  2046. }
  2047. if ((ctrl->need_close) || (ctrl->socket_id < 0) || (ctrl->state != NW_STATE_ONLINE))
  2048. {
  2049. return -1;
  2050. }
  2051. NW_LOCK;
  2052. ctrl->auto_mode = 1;
  2053. ctrl->wait_target_state = NW_WAIT_EVENT;
  2054. NW_UNLOCK;
  2055. if (!ctrl->task_handle || !timeout_ms)
  2056. {
  2057. return 1;
  2058. }
  2059. *is_timeout = 0;
  2060. uint8_t finish = 0;
  2061. OS_EVENT event;
  2062. int result;
  2063. //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
  2064. platform_start_timer(ctrl->timer, timeout_ms, 0);
  2065. while (!finish)
  2066. {
  2067. platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
  2068. switch (event.ID)
  2069. {
  2070. case EV_NW_RESULT_EVENT:
  2071. result = (int)event.Param1;
  2072. if (result)
  2073. {
  2074. result = -1;
  2075. }
  2076. if (out_event)
  2077. {
  2078. out_event->ID = 0;
  2079. }
  2080. finish = 1;
  2081. break;
  2082. case EV_NW_TIMEOUT:
  2083. *is_timeout = 1;
  2084. result = 0;
  2085. finish = 1;
  2086. break;
  2087. case EV_NW_BREAK_WAIT:
  2088. if (out_event)
  2089. {
  2090. *out_event = event;
  2091. }
  2092. result = 0;
  2093. finish = 1;
  2094. break;
  2095. default:
  2096. if (out_event)
  2097. {
  2098. *out_event = event;
  2099. result = 0;
  2100. finish = 1;
  2101. break;
  2102. }
  2103. else if (ctrl->user_callback)
  2104. {
  2105. ctrl->user_callback((void *)&event, ctrl->user_data);
  2106. }
  2107. break;
  2108. }
  2109. }
  2110. platform_stop_timer(ctrl->timer);
  2111. return result;
  2112. }
  2113. int network_wait_rx(network_ctrl_t *ctrl, uint32_t timeout_ms, uint8_t *is_break, uint8_t *is_timeout)
  2114. {
  2115. *is_timeout = 0;
  2116. *is_break = 0;
  2117. if (ctrl->new_rx_flag)
  2118. {
  2119. ctrl->wait_target_state = NW_WAIT_EVENT;
  2120. return 0;
  2121. }
  2122. if ((ctrl->need_close) || (ctrl->socket_id < 0) || (ctrl->state != NW_STATE_ONLINE))
  2123. {
  2124. return -1;
  2125. }
  2126. NW_LOCK;
  2127. ctrl->auto_mode = 1;
  2128. ctrl->wait_target_state = NW_WAIT_EVENT;
  2129. NW_UNLOCK;
  2130. uint8_t finish = 0;
  2131. OS_EVENT event;
  2132. int result;
  2133. //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
  2134. platform_start_timer(ctrl->timer, timeout_ms, 0);
  2135. while (!finish)
  2136. {
  2137. platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
  2138. switch (event.ID)
  2139. {
  2140. case EV_NW_RESULT_EVENT:
  2141. result = (int)event.Param1;
  2142. if (result)
  2143. {
  2144. result = -1;
  2145. finish = 1;
  2146. }
  2147. else if (ctrl->new_rx_flag)
  2148. {
  2149. result = 0;
  2150. finish = 1;
  2151. }
  2152. break;
  2153. case EV_NW_TIMEOUT:
  2154. *is_timeout = 1;
  2155. result = 0;
  2156. finish = 1;
  2157. break;
  2158. case EV_NW_BREAK_WAIT:
  2159. *is_break = 1;
  2160. result = 0;
  2161. finish = 1;
  2162. break;
  2163. default:
  2164. if (ctrl->user_callback)
  2165. {
  2166. ctrl->user_callback((void *)&event, ctrl->user_data);
  2167. }
  2168. break;
  2169. }
  2170. ctrl->wait_target_state = NW_WAIT_EVENT;
  2171. }
  2172. platform_stop_timer(ctrl->timer);
  2173. return result;
  2174. }
  2175. #else
  2176. #include "dhcp_def.h"
  2177. extern void DBG_Printf(const char* format, ...);
  2178. extern void DBG_HexPrintf(void *Data, unsigned int len);
  2179. //#define DBG(x,y...) DBG_Printf("%s %d:"x"\r\n", __FUNCTION__,__LINE__,##y)
  2180. //#define DBG_ERR(x,y...) DBG_Printf("%s %d:"x"\r\n", __FUNCTION__,__LINE__,##y)
  2181. #define __NW_DEBUG_ENABLE__
  2182. #ifdef __NW_DEBUG_ENABLE__
  2183. #define DBG(x,y...) do {if (ctrl->is_debug) {DBG_Printf("%s %d:"x"\r\n", __FUNCTION__,__LINE__,##y);}} while(0)
  2184. #define DBG_ERR(x,y...) DBG_Printf("%s %d:"x"\r\n", __FUNCTION__,__LINE__,##y)
  2185. #else
  2186. #define DBG(x,y...)
  2187. #define DBG_ERR(x,y...)
  2188. #endif
  2189. //#define NW_LOCK OS_LOCK
  2190. //#define NW_UNLOCK OS_UNLOCK
  2191. #define NW_LOCK platform_lock_mutex(ctrl->mutex)
  2192. #define NW_UNLOCK platform_unlock_mutex(ctrl->mutex)
  2193. #define SOL_SOCKET 0xfff /* options for socket level */
  2194. #define SO_REUSEADDR 0x0004 /* Allow local address reuse */
  2195. #define SO_KEEPALIVE 0x0008 /* keep connections alive */
  2196. #define IPPROTO_ICMP 1
  2197. #define IPPROTO_TCP 6
  2198. #define TCP_NODELAY 0x01 /* don't delay send to coalesce packets */
  2199. #define TCP_KEEPALIVE 0x02 /* send KEEPALIVE probes when idle for pcb->keep_idle milliseconds */
  2200. #define TCP_KEEPIDLE 0x03 /* set pcb->keep_idle - Same as TCP_KEEPALIVE, but use seconds for get/setsockopt */
  2201. #define TCP_KEEPINTVL 0x04 /* set pcb->keep_intvl - Use seconds for get/setsockopt */
  2202. #define TCP_KEEPCNT 0x05 /* set pcb->keep_cnt - Use number of probes sent for get/setsockopt */
  2203. static uint8_t network_check_ip_same(luat_ip_addr_t *ip1, luat_ip_addr_t *ip2)
  2204. {
  2205. if (ip1->is_ipv6 != ip2->is_ipv6)
  2206. {
  2207. return 0;
  2208. }
  2209. if (ip1->is_ipv6)
  2210. {
  2211. return !memcmp(ip1->ipv6_u8_addr, ip2->ipv6_u8_addr, 16);
  2212. }
  2213. else
  2214. {
  2215. return (ip1->ipv4 == ip2->ipv4);
  2216. }
  2217. }
  2218. 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)
  2219. {
  2220. int result = -1;
  2221. if (ctrl->is_tcp)
  2222. {
  2223. result = network_socket_send(ctrl, data, len, flags, NULL, 0);
  2224. }
  2225. else
  2226. {
  2227. if (remote_ip)
  2228. {
  2229. result = network_socket_send(ctrl, data, len, flags, remote_ip, remote_port);
  2230. }
  2231. else
  2232. {
  2233. result = network_socket_send(ctrl, data, len, flags, ctrl->online_ip, ctrl->remote_port);
  2234. }
  2235. }
  2236. if (result >= 0)
  2237. {
  2238. ctrl->tx_size += len;
  2239. }
  2240. else
  2241. {
  2242. ctrl->need_close = 1;
  2243. }
  2244. return result;
  2245. }
  2246. static int32_t tls_shorttimeout(void *data, void *param)
  2247. {
  2248. network_ctrl_t *ctrl = (network_ctrl_t *)param;
  2249. if (!ctrl->tls_mode)
  2250. {
  2251. platform_stop_timer(ctrl->tls_long_timer);
  2252. return 0;
  2253. }
  2254. if (0 == ctrl->tls_timer_state)
  2255. {
  2256. ctrl->tls_timer_state = 1;
  2257. }
  2258. return 0;
  2259. }
  2260. static int32_t tls_longtimeout(void *data, void *param)
  2261. {
  2262. network_ctrl_t *ctrl = (network_ctrl_t *)param;
  2263. platform_stop_timer(ctrl->tls_short_timer);
  2264. if (!ctrl->tls_mode)
  2265. {
  2266. return 0;
  2267. }
  2268. ctrl->tls_timer_state = 2;
  2269. return 0;
  2270. }
  2271. static void tls_settimer( void *data, uint32_t int_ms, uint32_t fin_ms )
  2272. {
  2273. network_ctrl_t *ctrl = (network_ctrl_t *)data;
  2274. if (!ctrl->tls_mode)
  2275. {
  2276. return;
  2277. }
  2278. if (!fin_ms)
  2279. {
  2280. platform_stop_timer(ctrl->tls_short_timer);
  2281. platform_stop_timer(ctrl->tls_long_timer);
  2282. ctrl->tls_timer_state = -1;
  2283. return ;
  2284. }
  2285. platform_start_timer(ctrl->tls_short_timer, int_ms, 0);
  2286. platform_start_timer(ctrl->tls_long_timer, fin_ms, 0);
  2287. ctrl->tls_timer_state = 0;
  2288. }
  2289. static int tls_gettimer( void *data )
  2290. {
  2291. network_ctrl_t *ctrl = (network_ctrl_t *)data;
  2292. if (!ctrl->tls_mode)
  2293. {
  2294. return -ERROR_PARAM_INVALID;
  2295. }
  2296. return ctrl->tls_timer_state;
  2297. }
  2298. static void tls_dbg(void *data, int level,
  2299. const char *file, int line,
  2300. const char *str)
  2301. {
  2302. DBG_Printf("%s %d:%s", file, line, str);
  2303. }
  2304. static int tls_send(void *ctx, const unsigned char *buf, size_t len )
  2305. {
  2306. network_ctrl_t *ctrl = (network_ctrl_t *)ctx;
  2307. if (!ctrl->tls_mode)
  2308. {
  2309. return -ERROR_PERMISSION_DENIED;
  2310. }
  2311. if (network_base_tx(ctrl, buf, len, 0, NULL, 0) != len)
  2312. {
  2313. return -0x004E;
  2314. }
  2315. else
  2316. {
  2317. return len;
  2318. }
  2319. }
  2320. static int tls_recv(void *ctx, unsigned char *buf, size_t len )
  2321. {
  2322. #ifdef LUAT_USE_TLS
  2323. network_ctrl_t *ctrl = (network_ctrl_t *)ctx;
  2324. luat_ip_addr_t remote_ip;
  2325. uint16_t remote_port;
  2326. int result = -1;
  2327. if (!ctrl->tls_mode)
  2328. {
  2329. return -1;
  2330. }
  2331. TLS_RECV:
  2332. result = network_socket_receive(ctrl, buf, len, 0, &remote_ip, &remote_port);
  2333. if (result < 0)
  2334. {
  2335. return -0x004C;
  2336. }
  2337. if (result > 0)
  2338. {
  2339. if (!ctrl->is_tcp)
  2340. {
  2341. if ((remote_port == ctrl->remote_port) && network_check_ip_same(&remote_ip, ctrl->online_ip))
  2342. {
  2343. goto TLS_RECV;
  2344. }
  2345. }
  2346. return result;
  2347. }
  2348. return MBEDTLS_ERR_SSL_WANT_READ;
  2349. #else
  2350. return -1;
  2351. #endif
  2352. }
  2353. static int network_get_host_by_name(network_ctrl_t *ctrl)
  2354. {
  2355. ctrl->remote_ip.is_ipv6 = 0xff;
  2356. if (network_string_is_ipv4(ctrl->domain_name, ctrl->domain_name_len))
  2357. {
  2358. ctrl->remote_ip.is_ipv6 = 0;
  2359. ctrl->remote_ip.ipv4 = network_string_to_ipv4(ctrl->domain_name, ctrl->domain_name_len);
  2360. }
  2361. else
  2362. {
  2363. char *name = zalloc(ctrl->domain_name_len + 1);
  2364. memcpy(name, ctrl->domain_name, ctrl->domain_name_len);
  2365. network_string_to_ipv6(name, &ctrl->remote_ip);
  2366. free(name);
  2367. }
  2368. if (ctrl->remote_ip.is_ipv6 != 0xff)
  2369. {
  2370. return 0;
  2371. }
  2372. else
  2373. {
  2374. return -1;
  2375. }
  2376. }
  2377. static void network_update_dns_cache(network_ctrl_t *ctrl)
  2378. {
  2379. }
  2380. static void network_get_dns_cache(network_ctrl_t *ctrl)
  2381. {
  2382. }
  2383. static int network_base_connect(network_ctrl_t *ctrl, luat_ip_addr_t *remote_ip)
  2384. {
  2385. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  2386. if (ctrl->socket_id >= 0)
  2387. {
  2388. network_force_close_socket(ctrl);
  2389. }
  2390. if (remote_ip)
  2391. {
  2392. if (network_create_soceket(ctrl, remote_ip->is_ipv6) < 0)
  2393. {
  2394. network_clean_invaild_socket(ctrl->adapter_index);
  2395. if (network_create_soceket(ctrl, remote_ip->is_ipv6) < 0)
  2396. {
  2397. return -1;
  2398. }
  2399. }
  2400. if (adapter->opt->is_posix)
  2401. {
  2402. network_setsockopt(ctrl, SOL_SOCKET, SO_KEEPALIVE, (void *)&ctrl->tcp_keep_alive, sizeof(ctrl->tcp_keep_alive));
  2403. if (ctrl->tcp_keep_alive)
  2404. {
  2405. network_setsockopt(ctrl, IPPROTO_TCP, TCP_KEEPIDLE, (void*)&ctrl->tcp_keep_idle, sizeof(ctrl->tcp_keep_idle));
  2406. network_setsockopt(ctrl, IPPROTO_TCP, TCP_KEEPINTVL, (void *)&ctrl->tcp_keep_interval, sizeof(ctrl->tcp_keep_interval));
  2407. network_setsockopt(ctrl, IPPROTO_TCP, TCP_KEEPCNT, (void *)&ctrl->tcp_keep_cnt, sizeof(ctrl->tcp_keep_cnt));
  2408. }
  2409. }
  2410. else
  2411. {
  2412. network_user_cmd(ctrl, NW_CMD_AUTO_HEART_TIME, ctrl->tcp_keep_idle);
  2413. }
  2414. return network_socket_connect(ctrl, remote_ip);
  2415. }
  2416. else
  2417. {
  2418. luat_ip_addr_t local_ip, net_mask, gate_way;
  2419. network_get_local_ip_info(ctrl, &local_ip, &net_mask, &gate_way);
  2420. if (network_create_soceket(ctrl, local_ip.is_ipv6) < 0)
  2421. {
  2422. network_clean_invaild_socket(ctrl->adapter_index);
  2423. if (network_create_soceket(ctrl, local_ip.is_ipv6) < 0)
  2424. {
  2425. return -1;
  2426. }
  2427. }
  2428. return network_socket_listen(ctrl);
  2429. }
  2430. }
  2431. static int network_prepare_connect(network_ctrl_t *ctrl)
  2432. {
  2433. if (ctrl->remote_ip.is_ipv6 != 0xff)
  2434. {
  2435. ;
  2436. }
  2437. else if (ctrl->domain_name)
  2438. {
  2439. if (network_get_host_by_name(ctrl))
  2440. {
  2441. if (network_dns(ctrl))
  2442. {
  2443. network_socket_force_close(ctrl);
  2444. return -1;
  2445. }
  2446. ctrl->state = NW_STATE_WAIT_DNS;
  2447. return 0;
  2448. }
  2449. }
  2450. else
  2451. {
  2452. return -1;
  2453. }
  2454. if (network_base_connect(ctrl, &ctrl->remote_ip))
  2455. {
  2456. network_socket_force_close(ctrl);
  2457. return -1;
  2458. }
  2459. ctrl->state = NW_STATE_CONNECTING;
  2460. return 0;
  2461. }
  2462. static int network_state_link_off(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  2463. {
  2464. if (EV_NW_STATE == event->ID)
  2465. {
  2466. if (event->Param2)
  2467. {
  2468. ctrl->state = NW_STATE_OFF_LINE;
  2469. if (NW_WAIT_LINK_UP == ctrl->wait_target_state)
  2470. {
  2471. return 0;
  2472. }
  2473. else if (NW_WAIT_ON_LINE == ctrl->wait_target_state)
  2474. {
  2475. if (ctrl->is_server_mode)
  2476. {
  2477. if (network_base_connect(ctrl, NULL))
  2478. {
  2479. return -1;
  2480. }
  2481. ctrl->state = NW_STATE_CONNECTING;
  2482. }
  2483. else
  2484. {
  2485. if (network_prepare_connect(ctrl))
  2486. {
  2487. return -1;
  2488. }
  2489. }
  2490. return 1;
  2491. }
  2492. }
  2493. }
  2494. return 1;
  2495. }
  2496. static int network_state_off_line(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  2497. {
  2498. return 1;
  2499. }
  2500. static int network_state_wait_dns(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  2501. {
  2502. if ((ctrl->need_close) || ctrl->wait_target_state != NW_WAIT_ON_LINE) return -1;
  2503. switch(event->ID)
  2504. {
  2505. case EV_NW_RESET:
  2506. case EV_NW_SOCKET_ERROR:
  2507. return -1;
  2508. case EV_NW_STATE:
  2509. if (!event->Param2)
  2510. {
  2511. return -1;
  2512. }
  2513. break;
  2514. case EV_NW_DNS_RESULT:
  2515. if (event->Param1)
  2516. {
  2517. //更新dns cache
  2518. ctrl->dns_ip = event->Param2;
  2519. ctrl->dns_ip_nums = event->Param1;
  2520. network_update_dns_cache(ctrl);
  2521. }
  2522. else
  2523. {
  2524. ctrl->dns_ip_nums = 0;
  2525. network_get_dns_cache(ctrl);
  2526. if (!ctrl->dns_ip_nums)
  2527. {
  2528. return -1;
  2529. }
  2530. }
  2531. ctrl->dns_ip_cnt = 0;
  2532. if (network_base_connect(ctrl, &ctrl->dns_ip[ctrl->dns_ip_cnt].ip))
  2533. {
  2534. network_socket_force_close(ctrl);
  2535. return -1;
  2536. }
  2537. else
  2538. {
  2539. ctrl->state = NW_STATE_CONNECTING;
  2540. return 1;
  2541. }
  2542. default:
  2543. return 1;
  2544. }
  2545. return -1; // TODO 返回值不确定
  2546. }
  2547. static int network_state_connecting(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  2548. {
  2549. if ((ctrl->need_close) || (ctrl->wait_target_state != NW_WAIT_ON_LINE)) return -1;
  2550. switch(event->ID)
  2551. {
  2552. case EV_NW_RESET:
  2553. return -1;
  2554. case EV_NW_SOCKET_ERROR:
  2555. case EV_NW_SOCKET_REMOTE_CLOSE:
  2556. case EV_NW_SOCKET_CLOSE_OK:
  2557. if (ctrl->remote_ip.is_ipv6 != 0xff)
  2558. {
  2559. return -1;
  2560. }
  2561. DBG("dns ip %d no connect!", ctrl->dns_ip_cnt);
  2562. ctrl->dns_ip_cnt++;
  2563. if (ctrl->dns_ip_cnt >= ctrl->dns_ip_nums)
  2564. {
  2565. return -1;
  2566. }
  2567. if (network_base_connect(ctrl, &ctrl->dns_ip[ctrl->dns_ip_cnt].ip))
  2568. {
  2569. network_socket_force_close(ctrl);
  2570. return -1;
  2571. }
  2572. else
  2573. {
  2574. ctrl->state = NW_STATE_CONNECTING;
  2575. return 1;
  2576. }
  2577. break;
  2578. case EV_NW_STATE:
  2579. if (!event->Param2)
  2580. {
  2581. return -1;
  2582. }
  2583. break;
  2584. case EV_NW_SOCKET_LISTEN:
  2585. if (ctrl->is_server_mode)
  2586. {
  2587. ctrl->state = NW_STATE_LISTEN;
  2588. return 1;
  2589. }
  2590. break;
  2591. case EV_NW_SOCKET_CONNECT_OK:
  2592. #ifdef LUAT_USE_TLS
  2593. if (ctrl->tls_mode)
  2594. {
  2595. mbedtls_ssl_free(ctrl->ssl);
  2596. memset(ctrl->ssl, 0, sizeof(mbedtls_ssl_context));
  2597. mbedtls_ssl_setup(ctrl->ssl, ctrl->config);
  2598. ctrl->ssl->f_set_timer = tls_settimer;
  2599. ctrl->ssl->f_get_timer = tls_gettimer;
  2600. ctrl->ssl->p_timer = ctrl;
  2601. ctrl->ssl->p_bio = ctrl;
  2602. ctrl->ssl->f_send = tls_send;
  2603. ctrl->ssl->f_recv = tls_recv;
  2604. // add by wendal
  2605. // cloudflare的https需要设置hostname才能访问
  2606. if (ctrl->domain_name_len > 0 && ctrl->domain_name_len < 256) {
  2607. char host[257] = {0};
  2608. memcpy(host, ctrl->domain_name, ctrl->domain_name_len);
  2609. mbedtls_ssl_set_hostname(ctrl->ssl, host);
  2610. //LLOGD("CALL mbedtls_ssl_set_hostname(%s)", host);
  2611. }
  2612. else {
  2613. //LLOGD("skip mbedtls_ssl_set_hostname");
  2614. }
  2615. ctrl->state = NW_STATE_SHAKEHAND;
  2616. do
  2617. {
  2618. int result = mbedtls_ssl_handshake_step( ctrl->ssl );
  2619. switch(result)
  2620. {
  2621. case MBEDTLS_ERR_SSL_WANT_READ:
  2622. return 1;
  2623. case 0:
  2624. break;
  2625. default:
  2626. DBG_ERR("0x%x, %d", -result, ctrl->ssl->state);
  2627. return -1;
  2628. }
  2629. }while(ctrl->ssl->state != MBEDTLS_SSL_HANDSHAKE_OVER);
  2630. return 0;
  2631. }
  2632. else
  2633. #endif
  2634. {
  2635. ctrl->state = NW_STATE_ONLINE;
  2636. return 0;
  2637. }
  2638. default:
  2639. return 1;
  2640. }
  2641. return 1; // TODO 返回值不确定
  2642. }
  2643. static int network_state_shakehand(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  2644. {
  2645. if ((ctrl->need_close) || ((ctrl->wait_target_state != NW_WAIT_ON_LINE) && (ctrl->wait_target_state != NW_WAIT_TX_OK))) return -1;
  2646. switch(event->ID)
  2647. {
  2648. case EV_NW_RESET:
  2649. case EV_NW_SOCKET_ERROR:
  2650. case EV_NW_SOCKET_REMOTE_CLOSE:
  2651. case EV_NW_SOCKET_CLOSE_OK:
  2652. ctrl->need_close = 1;
  2653. return -1;
  2654. case EV_NW_STATE:
  2655. if (!event->Param2)
  2656. {
  2657. ctrl->need_close = 1;
  2658. return -1;
  2659. }
  2660. break;
  2661. case EV_NW_SOCKET_TX_OK:
  2662. ctrl->ack_size += event->Param2;
  2663. if (ctrl->is_debug)
  2664. {
  2665. DBG("%llu,%llu",ctrl->tx_size, ctrl->ack_size);
  2666. }
  2667. break;
  2668. #ifdef LUAT_USE_TLS
  2669. case EV_NW_SOCKET_RX_NEW:
  2670. do
  2671. {
  2672. int result = mbedtls_ssl_handshake_step( ctrl->ssl );
  2673. switch(result)
  2674. {
  2675. case MBEDTLS_ERR_SSL_WANT_READ:
  2676. return 1;
  2677. case 0:
  2678. break;
  2679. default:
  2680. DBG_ERR("0x%x, %d", -result, ctrl->ssl->state);
  2681. ctrl->need_close = 1;
  2682. return -1;
  2683. }
  2684. }while(ctrl->ssl->state != MBEDTLS_SSL_HANDSHAKE_OVER);
  2685. ctrl->state = NW_STATE_ONLINE;
  2686. if (NW_WAIT_TX_OK == ctrl->wait_target_state)
  2687. {
  2688. if (!ctrl->cache_data)
  2689. {
  2690. ctrl->need_close = 1;
  2691. return -1;
  2692. }
  2693. int result = mbedtls_ssl_write(ctrl->ssl, ctrl->cache_data, ctrl->cache_len);
  2694. free(ctrl->cache_data);
  2695. ctrl->cache_data = NULL;
  2696. ctrl->cache_len = 0;
  2697. if (result < 0)
  2698. {
  2699. DBG("%08x", -result);
  2700. ctrl->need_close = 1;
  2701. return -1;
  2702. }
  2703. return 1;
  2704. }
  2705. return 0;
  2706. #endif
  2707. case EV_NW_SOCKET_CONNECT_OK:
  2708. DBG_ERR("!");
  2709. return 1;
  2710. default:
  2711. return 1;
  2712. }
  2713. return 1;
  2714. }
  2715. static int network_state_on_line(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  2716. {
  2717. if ((ctrl->need_close) || NW_WAIT_OFF_LINE == ctrl->wait_target_state)
  2718. {
  2719. return -1;
  2720. }
  2721. switch(event->ID)
  2722. {
  2723. case EV_NW_RESET:
  2724. case EV_NW_SOCKET_ERROR:
  2725. case EV_NW_SOCKET_REMOTE_CLOSE:
  2726. case EV_NW_SOCKET_CLOSE_OK:
  2727. ctrl->need_close = 1;
  2728. return -1;
  2729. case EV_NW_STATE:
  2730. if (!event->Param2)
  2731. {
  2732. ctrl->need_close = 1;
  2733. return -1;
  2734. }
  2735. break;
  2736. case EV_NW_SOCKET_TX_OK:
  2737. ctrl->ack_size += event->Param2;
  2738. if (NW_WAIT_TX_OK == ctrl->wait_target_state)
  2739. {
  2740. if (ctrl->ack_size == ctrl->tx_size)
  2741. {
  2742. return 0;
  2743. }
  2744. }
  2745. break;
  2746. case EV_NW_SOCKET_RX_NEW:
  2747. ctrl->new_rx_flag = 1;
  2748. if (NW_WAIT_TX_OK != ctrl->wait_target_state)
  2749. {
  2750. return 0;
  2751. }
  2752. break;
  2753. default:
  2754. return 1;
  2755. }
  2756. return 1;
  2757. }
  2758. static int network_state_listen(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  2759. {
  2760. if ((ctrl->need_close) || NW_WAIT_OFF_LINE == ctrl->wait_target_state)
  2761. {
  2762. return -1;
  2763. }
  2764. switch(event->ID)
  2765. {
  2766. case EV_NW_RESET:
  2767. case EV_NW_SOCKET_ERROR:
  2768. case EV_NW_SOCKET_REMOTE_CLOSE:
  2769. case EV_NW_SOCKET_CLOSE_OK:
  2770. ctrl->need_close = 1;
  2771. return -1;
  2772. case EV_NW_STATE:
  2773. if (!event->Param2)
  2774. {
  2775. ctrl->need_close = 1;
  2776. return -1;
  2777. }
  2778. break;
  2779. case EV_NW_SOCKET_NEW_CONNECT:
  2780. case EV_NW_SOCKET_CONNECT_OK:
  2781. ctrl->state = NW_STATE_ONLINE;
  2782. return 0;
  2783. default:
  2784. return 1;
  2785. }
  2786. return 1;
  2787. }
  2788. static int network_state_disconnecting(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  2789. {
  2790. if (ctrl->wait_target_state != NW_WAIT_OFF_LINE)
  2791. {
  2792. return -1;
  2793. }
  2794. switch(event->ID)
  2795. {
  2796. case EV_NW_RESET:
  2797. case EV_NW_SOCKET_ERROR:
  2798. case EV_NW_SOCKET_REMOTE_CLOSE:
  2799. case EV_NW_SOCKET_CLOSE_OK:
  2800. network_force_close_socket(ctrl);
  2801. ctrl->state = NW_STATE_OFF_LINE;
  2802. ctrl->socket_id = -1;
  2803. return 0;
  2804. case EV_NW_STATE:
  2805. if (!event->Param2)
  2806. {
  2807. return -1;
  2808. }
  2809. else
  2810. {
  2811. network_force_close_socket(ctrl);
  2812. ctrl->state = NW_STATE_OFF_LINE;
  2813. ctrl->socket_id = -1;
  2814. }
  2815. break;
  2816. default:
  2817. return 1;
  2818. }
  2819. return -1;
  2820. }
  2821. typedef int (*network_state_fun)(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter);
  2822. static network_state_fun network_state_fun_list[]=
  2823. {
  2824. network_state_link_off,
  2825. network_state_off_line,
  2826. network_state_wait_dns,
  2827. network_state_connecting,
  2828. network_state_shakehand,
  2829. network_state_on_line,
  2830. network_state_listen,
  2831. network_state_disconnecting,
  2832. };
  2833. static void network_default_statemachine(network_ctrl_t *ctrl, OS_EVENT *event, network_adapter_t *adapter)
  2834. {
  2835. int result = 0;
  2836. if (ctrl->state > NW_STATE_DISCONNECTING)
  2837. {
  2838. ctrl->state = NW_STATE_LINK_OFF;
  2839. event->Param1 = -1;
  2840. network_force_close_socket(ctrl);
  2841. event->ID = ctrl->wait_target_state + EV_NW_RESULT_BASE;
  2842. }
  2843. else
  2844. {
  2845. result = network_state_fun_list[ctrl->state](ctrl, event, adapter);
  2846. if (result > 0)
  2847. {
  2848. return ;
  2849. }
  2850. event->ID = (ctrl->wait_target_state?ctrl->wait_target_state:NW_WAIT_EVENT) + EV_NW_RESULT_BASE;
  2851. event->Param1 = result;
  2852. }
  2853. if ((ctrl->state != NW_STATE_LISTEN) || (result < 0))
  2854. {
  2855. ctrl->wait_target_state = NW_WAIT_NONE;
  2856. }
  2857. if (ctrl->task_handle)
  2858. {
  2859. platform_send_event(ctrl->task_handle, event->ID, event->Param1, event->Param2, event->Param3);
  2860. }
  2861. else if (ctrl->user_callback)
  2862. {
  2863. ctrl->user_callback(event, ctrl->user_data);
  2864. }
  2865. }
  2866. static int32_t network_default_socket_callback(void *data, void *param)
  2867. {
  2868. OS_EVENT *event = (OS_EVENT *)data;
  2869. OS_EVENT temp_event;
  2870. luat_network_cb_param_t *cb_param = (luat_network_cb_param_t *)param;
  2871. network_adapter_t *adapter =(network_adapter_t *)(cb_param->param);
  2872. int i = 0;
  2873. network_ctrl_t *ctrl = (network_ctrl_t *)event->Param3;
  2874. if (event->ID > EV_NW_TIMEOUT)
  2875. {
  2876. if (ctrl && ((ctrl->tag == cb_param->tag) || (event->ID == EV_NW_DNS_RESULT)))
  2877. {
  2878. if (ctrl->auto_mode)
  2879. {
  2880. DBG("socket %d,%s,%s,%s", ctrl->socket_id, network_ctrl_event_id_string(event->ID),
  2881. network_ctrl_state_string(ctrl->state),
  2882. network_ctrl_wait_state_string(ctrl->wait_target_state));
  2883. network_default_statemachine(ctrl, event, adapter);
  2884. DBG("%s,%s",
  2885. network_ctrl_state_string(ctrl->state),
  2886. network_ctrl_wait_state_string(ctrl->wait_target_state));
  2887. }
  2888. else if (ctrl->task_handle)
  2889. {
  2890. platform_send_event(ctrl->task_handle, event->ID, event->Param1, event->Param2, event->Param3);
  2891. }
  2892. else if (ctrl->user_callback)
  2893. {
  2894. ctrl->user_callback(event, ctrl->user_data);
  2895. }
  2896. }
  2897. else
  2898. {
  2899. DBG_ERR("cb ctrl invaild %x", ctrl);
  2900. DBG_HexPrintf(&ctrl->tag, 8);
  2901. DBG_HexPrintf(&cb_param->tag, 8);
  2902. }
  2903. }
  2904. else
  2905. {
  2906. for (i = 0; i < adapter->opt->max_socket_num; i++)
  2907. {
  2908. temp_event = *event;
  2909. if (adapter->ctrl_busy[i])
  2910. {
  2911. ctrl = &adapter->ctrl_table[i];
  2912. if (ctrl->auto_mode)
  2913. {
  2914. DBG("socket %d,%s,%s,%s", ctrl->socket_id, network_ctrl_event_id_string(event->ID),
  2915. network_ctrl_state_string(ctrl->state),
  2916. network_ctrl_wait_state_string(ctrl->wait_target_state));
  2917. network_default_statemachine(ctrl, &temp_event, adapter);
  2918. DBG("%s,%s",
  2919. network_ctrl_state_string(ctrl->state),
  2920. network_ctrl_wait_state_string(ctrl->wait_target_state));
  2921. }
  2922. else if (ctrl->task_handle)
  2923. {
  2924. platform_send_event(ctrl->task_handle, event->ID, event->Param1, event->Param2, event->Param3);
  2925. }
  2926. else if (ctrl->user_callback)
  2927. {
  2928. ctrl->user_callback(&temp_event, ctrl->user_data);
  2929. }
  2930. }
  2931. }
  2932. }
  2933. return 0;
  2934. }
  2935. static int32_t network_default_timer_callback(void *data, void *param)
  2936. {
  2937. platform_send_event(param, EV_NW_TIMEOUT, 0, 0, 0);
  2938. return 0;
  2939. }
  2940. uint8_t network_string_is_ipv4(const char *string, uint32_t len)
  2941. {
  2942. int i;
  2943. for(i = 0; i < len; i++)
  2944. {
  2945. if (!isdigit((int)string[i]) && (string[i] != '.'))
  2946. {
  2947. return 0;
  2948. }
  2949. }
  2950. return 1;
  2951. }
  2952. uint32_t network_string_to_ipv4(const char *string, uint32_t len)
  2953. {
  2954. int i;
  2955. int8_t Buf[4][4];
  2956. CmdParam CP;
  2957. PV_Union uIP;
  2958. char temp[32];
  2959. memset(Buf, 0, sizeof(Buf));
  2960. CP.param_max_len = 4;
  2961. CP.param_max_num = 4;
  2962. CP.param_num = 0;
  2963. CP.param_str = (int8_t *)Buf;
  2964. memcpy(temp, string, len);
  2965. temp[len] = 0;
  2966. CmdParseParam((int8_t*)temp, &CP, '.');
  2967. for(i = 0; i < 4; i++)
  2968. {
  2969. uIP.u8[i] = strtol((char *)Buf[i], NULL, 10);
  2970. }
  2971. // DBG("%d.%d.%d.%d", uIP.u8[0], uIP.u8[1], uIP.u8[2], uIP.u8[3]);
  2972. return uIP.u32;
  2973. }
  2974. #define SWAP(x) ((((x) & (uint32_t)0x000000ffUL) << 24) | \
  2975. (((x) & (uint32_t)0x0000ff00UL) << 8) | \
  2976. (((x) & (uint32_t)0x00ff0000UL) >> 8) | \
  2977. (((x) & (uint32_t)0xff000000UL) >> 24))
  2978. int network_string_to_ipv6(const char *string, luat_ip_addr_t *ip_addr)
  2979. {
  2980. uint32_t addr_index, zero_blocks, current_block_index, current_block_value;
  2981. const char *s;
  2982. ip_addr->is_ipv6 = 0xff;
  2983. /* Count the number of colons, to count the number of blocks in a "::" sequence
  2984. zero_blocks may be 1 even if there are no :: sequences */
  2985. zero_blocks = 8;
  2986. for (s = string; *s != 0; s++) {
  2987. if (*s == ':') {
  2988. zero_blocks--;
  2989. } else if (!isxdigit((int)*s)) {
  2990. break;
  2991. }
  2992. }
  2993. /* parse each block */
  2994. addr_index = 0;
  2995. current_block_index = 0;
  2996. current_block_value = 0;
  2997. for (s = string; *s != 0; s++) {
  2998. if (*s == ':') {
  2999. if (current_block_index & 0x1) {
  3000. ip_addr->ipv6_u32_addr[addr_index++] |= current_block_value;
  3001. }
  3002. else {
  3003. ip_addr->ipv6_u32_addr[addr_index] = current_block_value << 16;
  3004. }
  3005. current_block_index++;
  3006. current_block_value = 0;
  3007. if (current_block_index > 7) {
  3008. /* address too long! */
  3009. return 0;
  3010. }
  3011. if (s[1] == ':') {
  3012. if (s[2] == ':') {
  3013. /* invalid format: three successive colons */
  3014. return 0;
  3015. }
  3016. s++;
  3017. /* "::" found, set zeros */
  3018. while (zero_blocks > 0) {
  3019. zero_blocks--;
  3020. if (current_block_index & 0x1) {
  3021. addr_index++;
  3022. } else {
  3023. ip_addr->ipv6_u32_addr[addr_index] = 0;
  3024. }
  3025. current_block_index++;
  3026. if (current_block_index > 7) {
  3027. /* address too long! */
  3028. return 0;
  3029. }
  3030. }
  3031. }
  3032. } else if (isxdigit((int)*s)) {
  3033. /* add current digit */
  3034. current_block_value = (current_block_value << 4) +
  3035. (isxdigit((int)*s) ? (uint32_t)(*s - '0') :
  3036. (uint32_t)(10 + (isxdigit((int)*s) ? *s - 'a' : *s - 'A')));
  3037. } else {
  3038. /* unexpected digit, space? CRLF? */
  3039. break;
  3040. }
  3041. }
  3042. if (current_block_index & 0x1) {
  3043. ip_addr->ipv6_u32_addr[addr_index++] |= current_block_value;
  3044. }
  3045. else {
  3046. ip_addr->ipv6_u32_addr[addr_index] = current_block_value << 16;
  3047. }
  3048. /* convert to network byte order. */
  3049. for (addr_index = 0; addr_index < 4; addr_index++) {
  3050. ip_addr->ipv6_u32_addr[addr_index] = SWAP(ip_addr->ipv6_u32_addr[addr_index]);
  3051. }
  3052. if (current_block_index != 7) {
  3053. return 0;
  3054. }
  3055. ip_addr->is_ipv6 = 1;
  3056. return 1;
  3057. }
  3058. int network_get_last_register_adapter(void)
  3059. {
  3060. return prv_network.last_adapter_index;
  3061. }
  3062. int network_register_adapter(uint8_t adapter_index, network_adapter_info *info, void *user_data)
  3063. {
  3064. prv_adapter_table[adapter_index].opt = info;
  3065. prv_adapter_table[adapter_index].user_data = user_data;
  3066. info->socket_set_callback(network_default_socket_callback, &prv_adapter_table[adapter_index], user_data);
  3067. prv_adapter_table[adapter_index].ctrl_table = zalloc((info->max_socket_num) * sizeof(network_ctrl_t));
  3068. prv_adapter_table[adapter_index].ctrl_busy = zalloc(info->max_socket_num);
  3069. prv_adapter_table[adapter_index].port = 60000;
  3070. if (!prv_network.is_init)
  3071. {
  3072. INIT_LLIST_HEAD(&prv_network.dns_cache_head);
  3073. prv_network.is_init = 0;
  3074. }
  3075. prv_network.last_adapter_index = adapter_index;
  3076. return 0;
  3077. }
  3078. void network_set_dns_server(uint8_t adapter_index, uint8_t server_index, luat_ip_addr_t *ip)
  3079. {
  3080. network_adapter_t *adapter = &prv_adapter_table[adapter_index];
  3081. adapter->opt->set_dns_server(server_index, ip, adapter->user_data);
  3082. }
  3083. /*
  3084. * 申请一个network_ctrl
  3085. */
  3086. network_ctrl_t *network_alloc_ctrl(uint8_t adapter_index)
  3087. {
  3088. int i;
  3089. network_ctrl_t *ctrl = NULL;
  3090. network_adapter_t *adapter = &prv_adapter_table[adapter_index];
  3091. OS_LOCK;
  3092. for (i = 0; i < adapter->opt->max_socket_num; i++)
  3093. {
  3094. if (!adapter->ctrl_busy[i])
  3095. {
  3096. adapter->ctrl_busy[i] = 1;
  3097. ctrl = &adapter->ctrl_table[i];
  3098. ctrl->adapter_index = adapter_index;
  3099. break;
  3100. }
  3101. }
  3102. if (i >= adapter->opt->max_socket_num) {DBG_ERR("adapter no more ctrl!");}
  3103. OS_UNLOCK;
  3104. return ctrl;
  3105. }
  3106. /*
  3107. * 归还一个network_ctrl
  3108. */
  3109. void network_release_ctrl(network_ctrl_t *ctrl)
  3110. {
  3111. int i;
  3112. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3113. NW_UNLOCK;
  3114. OS_LOCK;
  3115. for (i = 0; i < adapter->opt->max_socket_num; i++)
  3116. {
  3117. if (&adapter->ctrl_table[i] == ctrl)
  3118. {
  3119. network_deinit_tls(ctrl);
  3120. if (ctrl->timer)
  3121. {
  3122. platform_release_timer(ctrl->timer);
  3123. }
  3124. if (ctrl->cache_data)
  3125. {
  3126. free(ctrl->cache_data);
  3127. ctrl->cache_data = NULL;
  3128. }
  3129. if (ctrl->dns_ip)
  3130. {
  3131. free(ctrl->dns_ip);
  3132. ctrl->dns_ip = NULL;
  3133. }
  3134. adapter->ctrl_busy[i] = 0;
  3135. platform_release_mutex(ctrl->mutex);
  3136. ctrl->mutex = NULL;
  3137. break;
  3138. }
  3139. }
  3140. OS_UNLOCK;
  3141. // NW_UNLOCK;
  3142. if (i >= adapter->opt->max_socket_num) {DBG_ERR("adapter index maybe error!, %d, %x", ctrl->adapter_index, ctrl);}
  3143. }
  3144. void network_init_ctrl(network_ctrl_t *ctrl, HANDLE task_handle, CBFuncEx_t callback, void *param)
  3145. {
  3146. uint8_t adapter_index = ctrl->adapter_index;
  3147. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3148. if (ctrl->dns_ip)
  3149. {
  3150. free(ctrl->dns_ip);
  3151. ctrl->dns_ip = NULL;
  3152. }
  3153. if (ctrl->cache_data)
  3154. {
  3155. free(ctrl->cache_data);
  3156. ctrl->cache_data = NULL;
  3157. }
  3158. HANDLE sem = ctrl->mutex;
  3159. memset(ctrl, 0, sizeof(network_ctrl_t));
  3160. ctrl->adapter_index = adapter_index;
  3161. ctrl->task_handle = task_handle;
  3162. ctrl->user_callback = callback;
  3163. ctrl->user_data = param;
  3164. ctrl->socket_id = -1;
  3165. ctrl->socket_param = ctrl;
  3166. ctrl->remote_ip.is_ipv6 = 0xff;
  3167. ctrl->mutex = sem;
  3168. if (task_handle)
  3169. {
  3170. ctrl->timer = platform_create_timer(network_default_timer_callback, task_handle, NULL);
  3171. }
  3172. if (!ctrl->mutex)
  3173. {
  3174. ctrl->mutex = platform_create_mutex();
  3175. }
  3176. }
  3177. 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)
  3178. {
  3179. ctrl->is_tcp = is_tcp;
  3180. ctrl->tcp_keep_alive = keep_alive;
  3181. ctrl->tcp_keep_idle = keep_idle;
  3182. ctrl->tcp_keep_interval = keep_interval;
  3183. ctrl->tcp_keep_cnt = keep_cnt;
  3184. ctrl->tcp_timeout_ms = tcp_timeout_ms;
  3185. }
  3186. void network_connect_ipv6_domain(network_ctrl_t *ctrl, uint8_t onoff)
  3187. {
  3188. ctrl->domain_ipv6 = onoff;
  3189. }
  3190. int network_set_local_port(network_ctrl_t *ctrl, uint16_t local_port)
  3191. {
  3192. int i;
  3193. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3194. if (local_port)
  3195. {
  3196. NW_LOCK;
  3197. for (i = 0; i < adapter->opt->max_socket_num; i++)
  3198. {
  3199. if (&adapter->ctrl_table[i] != ctrl)
  3200. {
  3201. if (adapter->ctrl_table[i].local_port == local_port)
  3202. {
  3203. NW_UNLOCK;
  3204. return -1;
  3205. }
  3206. }
  3207. }
  3208. ctrl->local_port = local_port;
  3209. NW_UNLOCK;
  3210. return 0;
  3211. }
  3212. else
  3213. {
  3214. NW_LOCK;
  3215. adapter->port++;
  3216. if (adapter->port < 60000)
  3217. {
  3218. adapter->port = 60000;
  3219. }
  3220. ctrl->local_port = adapter->port;
  3221. NW_UNLOCK;
  3222. return 0;
  3223. }
  3224. }
  3225. uint8_t network_check_ready(network_ctrl_t *ctrl)
  3226. {
  3227. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3228. return adapter->opt->check_ready(adapter->user_data);
  3229. }
  3230. int network_create_soceket(network_ctrl_t *ctrl, uint8_t is_ipv6)
  3231. {
  3232. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3233. ctrl->socket_id = adapter->opt->create_soceket(ctrl->is_tcp, &ctrl->tag, ctrl->socket_param, is_ipv6, adapter->user_data);
  3234. if (ctrl->socket_id < 0)
  3235. {
  3236. ctrl->tag = 0;
  3237. return -1;
  3238. }
  3239. return 0;
  3240. }
  3241. int network_socket_connect(network_ctrl_t *ctrl, luat_ip_addr_t *remote_ip)
  3242. {
  3243. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3244. ctrl->is_server_mode = 0;
  3245. ctrl->online_ip = remote_ip;
  3246. return adapter->opt->socket_connect(ctrl->socket_id, ctrl->tag, ctrl->local_port, remote_ip, ctrl->remote_port, adapter->user_data);
  3247. }
  3248. int network_socket_listen(network_ctrl_t *ctrl)
  3249. {
  3250. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3251. ctrl->is_server_mode = 1;
  3252. return adapter->opt->socket_listen(ctrl->socket_id, ctrl->tag, ctrl->local_port, adapter->user_data);
  3253. }
  3254. uint8_t network_accept_enable(network_ctrl_t *ctrl)
  3255. {
  3256. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3257. return !(adapter->opt->no_accept);
  3258. }
  3259. //作为server接受一个client
  3260. //成功返回0,失败 < 0
  3261. int network_socket_accept(network_ctrl_t *ctrl, network_ctrl_t *accept_ctrl)
  3262. {
  3263. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3264. if (adapter->opt->no_accept)
  3265. {
  3266. // 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);
  3267. adapter->opt->socket_accept(ctrl->socket_id, ctrl->tag, &ctrl->remote_ip, &ctrl->remote_port, adapter->user_data);
  3268. return 0;
  3269. }
  3270. accept_ctrl->socket_id = adapter->opt->socket_accept(ctrl->socket_id, ctrl->tag, &accept_ctrl->remote_ip, &accept_ctrl->remote_port, adapter->user_data);
  3271. if (accept_ctrl->socket_id < 0)
  3272. {
  3273. return -1;
  3274. }
  3275. else
  3276. {
  3277. accept_ctrl->is_tcp = ctrl->is_tcp;
  3278. accept_ctrl->tcp_keep_alive = ctrl->tcp_keep_alive;
  3279. accept_ctrl->tcp_keep_idle = ctrl->tcp_keep_idle;
  3280. accept_ctrl->tcp_keep_interval = ctrl->tcp_keep_interval;
  3281. accept_ctrl->tcp_keep_cnt = ctrl->tcp_keep_cnt;
  3282. accept_ctrl->tcp_timeout_ms = ctrl->tcp_timeout_ms;
  3283. accept_ctrl->local_port = ctrl->local_port;
  3284. accept_ctrl->state = NW_STATE_ONLINE;
  3285. return 0;
  3286. }
  3287. }
  3288. //主动断开一个tcp连接,需要走完整个tcp流程,用户需要接收到close ok回调才能确认彻底断开
  3289. //成功返回0,失败 < 0
  3290. int network_socket_disconnect(network_ctrl_t *ctrl)
  3291. {
  3292. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3293. if (ctrl->socket_id >= 0)
  3294. {
  3295. return adapter->opt->socket_disconnect(ctrl->socket_id, ctrl->tag, adapter->user_data);
  3296. }
  3297. return 0;
  3298. }
  3299. //释放掉socket的控制权
  3300. //成功返回0,失败 < 0
  3301. int network_socket_close(network_ctrl_t *ctrl)
  3302. {
  3303. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3304. if (ctrl->socket_id >= 0)
  3305. {
  3306. return adapter->opt->socket_close(ctrl->socket_id, ctrl->tag, adapter->user_data);
  3307. }
  3308. return 0;
  3309. }
  3310. //强行释放掉socket的控制权
  3311. //成功返回0,失败 < 0
  3312. int network_socket_force_close(network_ctrl_t *ctrl)
  3313. {
  3314. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3315. if (ctrl->socket_id >= 0)
  3316. {
  3317. adapter->opt->socket_force_close(ctrl->socket_id, adapter->user_data);
  3318. }
  3319. ctrl->socket_id = -1;
  3320. return 0;
  3321. }
  3322. //tcp时,不需要remote_ip和remote_port,如果buf为NULL,则返回当前缓存区的数据量,当返回值小于len时说明已经读完了
  3323. //udp时,只返回1个block数据,需要多次读直到没有数据为止
  3324. //成功返回实际读取的值,失败 < 0
  3325. 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)
  3326. {
  3327. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3328. return adapter->opt->socket_receive(ctrl->socket_id, ctrl->tag, buf, len, flags, remote_ip, remote_port, adapter->user_data);
  3329. }
  3330. //tcp时,不需要remote_ip和remote_port
  3331. //成功返回0,失败 < 0
  3332. 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)
  3333. {
  3334. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3335. return adapter->opt->socket_send(ctrl->socket_id, ctrl->tag, buf, len, flags, remote_ip, remote_port, adapter->user_data);
  3336. }
  3337. int network_getsockopt(network_ctrl_t *ctrl, int level, int optname, void *optval, uint32_t *optlen)
  3338. {
  3339. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3340. return adapter->opt->getsockopt(ctrl->socket_id, ctrl->tag, level, optname, optval, optlen, adapter->user_data);
  3341. }
  3342. int network_setsockopt(network_ctrl_t *ctrl, int level, int optname, const void *optval, uint32_t optlen)
  3343. {
  3344. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3345. return adapter->opt->setsockopt(ctrl->socket_id, ctrl->tag, level, optname, optval, optlen, adapter->user_data);
  3346. }
  3347. //非posix的socket,用这个根据实际硬件设置参数
  3348. int network_user_cmd(network_ctrl_t *ctrl, uint32_t cmd, uint32_t value)
  3349. {
  3350. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3351. return adapter->opt->user_cmd(ctrl->socket_id, ctrl->tag, cmd, value, adapter->user_data);
  3352. }
  3353. int network_dns(network_ctrl_t *ctrl)
  3354. {
  3355. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3356. return adapter->opt->dns(ctrl->domain_name, ctrl->domain_name_len, ctrl, adapter->user_data);
  3357. }
  3358. 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)
  3359. {
  3360. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3361. return adapter->opt->get_local_ip_info(ip, submask, gateway, adapter->user_data);
  3362. }
  3363. void network_force_close_socket(network_ctrl_t *ctrl)
  3364. {
  3365. #ifdef LUAT_USE_TLS
  3366. if (ctrl->tls_mode)
  3367. {
  3368. mbedtls_ssl_free(ctrl->ssl);
  3369. }
  3370. #endif
  3371. if (network_socket_close(ctrl))
  3372. {
  3373. network_clean_invaild_socket(ctrl->adapter_index);
  3374. network_socket_force_close(ctrl);
  3375. }
  3376. ctrl->need_close = 0;
  3377. ctrl->socket_id = -1;
  3378. if (ctrl->dns_ip)
  3379. {
  3380. free(ctrl->dns_ip);
  3381. ctrl->dns_ip = NULL;
  3382. }
  3383. ctrl->dns_ip_cnt = 0;
  3384. ctrl->dns_ip_nums = 0;
  3385. }
  3386. void network_clean_invaild_socket(uint8_t adapter_index)
  3387. {
  3388. int i = 0;
  3389. int *list = NULL;
  3390. network_adapter_t *adapter = &prv_adapter_table[adapter_index];
  3391. network_ctrl_t *ctrl = NULL;
  3392. list = malloc(adapter->opt->max_socket_num * sizeof(int));
  3393. NW_LOCK;
  3394. for (i = 0; i < adapter->opt->max_socket_num; i++)
  3395. {
  3396. ctrl = &adapter->ctrl_table[i];
  3397. if (!adapter->opt->socket_check(ctrl->socket_id, ctrl->tag, adapter->user_data))
  3398. {
  3399. list[i] = ctrl->socket_id;
  3400. }
  3401. else
  3402. {
  3403. ctrl->socket_id = -1;
  3404. list[i] = -1;
  3405. }
  3406. DBG("%d,%d", i, list[i]);
  3407. }
  3408. adapter->opt->socket_clean(list, adapter->opt->max_socket_num, adapter->user_data);
  3409. NW_UNLOCK;
  3410. free(list);
  3411. }
  3412. #ifdef LUAT_USE_TLS
  3413. static int tls_verify(void *ctx, mbedtls_x509_crt *crt, int Index, uint32_t *result)
  3414. {
  3415. network_ctrl_t *ctrl = (network_ctrl_t *)ctx;
  3416. DBG("%d, %08x", Index, *result);
  3417. return 0;
  3418. }
  3419. #endif
  3420. int network_set_psk_info(network_ctrl_t *ctrl,
  3421. const unsigned char *psk, size_t psk_len,
  3422. const unsigned char *psk_identity, size_t psk_identity_len)
  3423. {
  3424. #ifdef LUAT_USE_TLS
  3425. if (!ctrl->tls_mode)
  3426. {
  3427. return -ERROR_PERMISSION_DENIED;
  3428. }
  3429. // DBG("%.*s, %.*s", psk_len, psk, psk_identity_len, psk_identity);
  3430. int ret = mbedtls_ssl_conf_psk( ctrl->config,
  3431. psk, psk_len, psk_identity, psk_identity_len );
  3432. if (ret != 0)
  3433. {
  3434. DBG("0x%x", -ret);
  3435. return -ERROR_OPERATION_FAILED;
  3436. }
  3437. return ERROR_NONE;
  3438. #else
  3439. return -1;
  3440. #endif
  3441. }
  3442. int network_set_server_cert(network_ctrl_t *ctrl, const unsigned char *cert, size_t cert_len)
  3443. {
  3444. #ifdef LUAT_USE_TLS
  3445. int ret;
  3446. if (!ctrl->tls_mode)
  3447. {
  3448. return -ERROR_PERMISSION_DENIED;
  3449. }
  3450. ret = mbedtls_x509_crt_parse( ctrl->ca_cert, cert, cert_len);
  3451. if (ret != 0)
  3452. {
  3453. DBG("%08x", -ret);
  3454. return -ERROR_OPERATION_FAILED;
  3455. }
  3456. return ERROR_NONE;
  3457. #else
  3458. return -1;
  3459. #endif
  3460. }
  3461. int network_set_client_cert(network_ctrl_t *ctrl,
  3462. const unsigned char *cert, size_t certLen,
  3463. const unsigned char *key, size_t keylen,
  3464. const unsigned char *pwd, size_t pwdlen)
  3465. {
  3466. #ifdef LUAT_USE_TLS
  3467. int ret;
  3468. mbedtls_x509_crt *client_cert = NULL;
  3469. mbedtls_pk_context *pkey = NULL;
  3470. if (!ctrl->tls_mode)
  3471. {
  3472. return -ERROR_PERMISSION_DENIED;
  3473. }
  3474. client_cert = zalloc(sizeof(mbedtls_x509_crt));
  3475. pkey = zalloc(sizeof(mbedtls_pk_context));
  3476. if (!client_cert || !pkey)
  3477. {
  3478. goto ERROR_OUT;
  3479. }
  3480. ret = mbedtls_x509_crt_parse( client_cert, cert, certLen );
  3481. if (ret != 0)
  3482. {
  3483. DBG("%08x", -ret);
  3484. goto ERROR_OUT;
  3485. }
  3486. ret = mbedtls_pk_parse_key( pkey, key, keylen, pwd, pwdlen );
  3487. if (ret != 0)
  3488. {
  3489. DBG("%08x", -ret);
  3490. goto ERROR_OUT;
  3491. }
  3492. ret = mbedtls_ssl_conf_own_cert( ctrl->config, client_cert, pkey );
  3493. if (ret != 0)
  3494. {
  3495. DBG("%08x", -ret);
  3496. goto ERROR_OUT;
  3497. }
  3498. return ERROR_NONE;
  3499. ERROR_OUT:
  3500. if (client_cert) free(client_cert);
  3501. if (pkey) free(pkey);
  3502. return -1;
  3503. #else
  3504. return -1;
  3505. #endif
  3506. }
  3507. int network_cert_verify_result(network_ctrl_t *ctrl)
  3508. {
  3509. #ifdef LUAT_USE_TLS
  3510. if (!ctrl->tls_mode)
  3511. {
  3512. return -1;
  3513. }
  3514. return mbedtls_ssl_get_verify_result(ctrl->ssl);
  3515. #else
  3516. return -1;
  3517. #endif
  3518. }
  3519. static int tls_random( void *p_rng,
  3520. unsigned char *output, size_t output_len )
  3521. {
  3522. platform_random((char*)output, output_len);
  3523. return 0;
  3524. }
  3525. void network_init_tls(network_ctrl_t *ctrl, int verify_mode)
  3526. {
  3527. #ifdef LUAT_USE_TLS
  3528. ctrl->tls_mode = 1;
  3529. if (!ctrl->ssl)
  3530. {
  3531. ctrl->ssl = zalloc(sizeof(mbedtls_ssl_context));
  3532. ctrl->ca_cert = zalloc(sizeof(mbedtls_x509_crt));
  3533. ctrl->config = zalloc(sizeof(mbedtls_ssl_config));
  3534. 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);
  3535. ctrl->config->authmode = verify_mode;
  3536. ctrl->config->hs_timeout_min = 20000;
  3537. ctrl->config->f_rng = tls_random;
  3538. ctrl->config->p_rng = NULL;
  3539. ctrl->config->f_dbg = tls_dbg;
  3540. ctrl->config->p_dbg = NULL;
  3541. ctrl->config->f_vrfy = tls_verify;
  3542. ctrl->config->p_vrfy = ctrl;
  3543. ctrl->config->ca_chain = ctrl->ca_cert;
  3544. ctrl->config->read_timeout = 20000;
  3545. ctrl->tls_long_timer = platform_create_timer(tls_longtimeout, ctrl, NULL);
  3546. ctrl->tls_short_timer = platform_create_timer(tls_shorttimeout, ctrl, NULL);
  3547. }
  3548. ctrl->tls_timer_state = -1;
  3549. #endif
  3550. }
  3551. void network_deinit_tls(network_ctrl_t *ctrl)
  3552. {
  3553. #ifdef LUAT_USE_TLS
  3554. if (ctrl->ssl)
  3555. {
  3556. mbedtls_ssl_free(ctrl->ssl);
  3557. free(ctrl->ssl);
  3558. ctrl->ssl = NULL;
  3559. }
  3560. if (ctrl->config)
  3561. {
  3562. mbedtls_ssl_config_free(ctrl->config);
  3563. free(ctrl->config);
  3564. ctrl->config = NULL;
  3565. }
  3566. if (ctrl->ca_cert)
  3567. {
  3568. mbedtls_x509_crt_free(ctrl->ca_cert);
  3569. free(ctrl->ca_cert);
  3570. ctrl->ca_cert = NULL;
  3571. }
  3572. ctrl->tls_mode = 0;
  3573. ctrl->tls_timer_state = -1;
  3574. if (ctrl->tls_short_timer)
  3575. {
  3576. platform_release_timer(ctrl->tls_short_timer);
  3577. ctrl->tls_short_timer = NULL;
  3578. }
  3579. if (ctrl->tls_long_timer)
  3580. {
  3581. platform_release_timer(ctrl->tls_long_timer);
  3582. ctrl->tls_long_timer = NULL;
  3583. }
  3584. #endif
  3585. }
  3586. int network_wait_link_up(network_ctrl_t *ctrl, uint32_t timeout_ms)
  3587. {
  3588. NW_LOCK;
  3589. ctrl->auto_mode = 1;
  3590. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3591. if (network_check_ready(ctrl))
  3592. {
  3593. ctrl->state = NW_STATE_OFF_LINE;
  3594. ctrl->wait_target_state = NW_WAIT_NONE;
  3595. NW_UNLOCK;
  3596. return 0;
  3597. }
  3598. ctrl->state = NW_STATE_LINK_OFF;
  3599. ctrl->wait_target_state = NW_WAIT_LINK_UP;
  3600. NW_UNLOCK;
  3601. if (!ctrl->task_handle || !timeout_ms)
  3602. {
  3603. return 1;
  3604. }
  3605. uint8_t finish = 0;
  3606. OS_EVENT event;
  3607. int result;
  3608. //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
  3609. platform_start_timer(ctrl->timer, timeout_ms, 0);
  3610. while (!finish)
  3611. {
  3612. platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
  3613. switch (event.ID)
  3614. {
  3615. case EV_NW_RESULT_LINK:
  3616. result = (int)event.Param1;
  3617. finish = 1;
  3618. break;
  3619. case EV_NW_TIMEOUT:
  3620. result = -1;
  3621. finish = 1;
  3622. break;
  3623. default:
  3624. if (ctrl->user_callback)
  3625. {
  3626. ctrl->user_callback((void *)&event, ctrl->user_data);
  3627. }
  3628. break;
  3629. }
  3630. }
  3631. platform_stop_timer(ctrl->timer);
  3632. return result;
  3633. }
  3634. /*
  3635. * 1.进行ready检测和等待ready
  3636. * 2.有remote_ip则开始连接服务器并等待连接结果
  3637. * 3.没有remote_ip则开始对url进行dns解析,解析完成后对所有ip进行尝试连接直到有个成功或者全部失败
  3638. * 4.如果是加密模式,还要走握手环节,等到握手环节完成后才能返回结果
  3639. * local_port如果为0则api内部自动生成一个
  3640. */
  3641. 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)
  3642. {
  3643. if (ctrl->socket_id >= 0)
  3644. {
  3645. return -1;
  3646. }
  3647. NW_LOCK;
  3648. ctrl->is_server_mode = 0;
  3649. if (ctrl->dns_ip)
  3650. {
  3651. free(ctrl->dns_ip);
  3652. ctrl->dns_ip = NULL;
  3653. }
  3654. if (ctrl->cache_data)
  3655. {
  3656. free(ctrl->cache_data);
  3657. ctrl->cache_data = NULL;
  3658. }
  3659. ctrl->need_close = 0;
  3660. ctrl->domain_name = domain_name;
  3661. ctrl->domain_name_len = domain_name_len;
  3662. if (remote_ip)
  3663. {
  3664. ctrl->remote_ip = *remote_ip;
  3665. }
  3666. else
  3667. {
  3668. ctrl->remote_ip.is_ipv6 = 0xff;
  3669. }
  3670. ctrl->auto_mode = 1;
  3671. ctrl->remote_port = remote_port;
  3672. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3673. ctrl->wait_target_state = NW_WAIT_ON_LINE;
  3674. if (!network_check_ready(ctrl))
  3675. {
  3676. ctrl->state = NW_STATE_LINK_OFF;
  3677. goto NETWORK_CONNECT_WAIT;
  3678. }
  3679. if (network_prepare_connect(ctrl))
  3680. {
  3681. ctrl->state = NW_STATE_OFF_LINE;
  3682. ctrl->wait_target_state = NW_WAIT_NONE;
  3683. NW_UNLOCK;
  3684. return -1;
  3685. }
  3686. NETWORK_CONNECT_WAIT:
  3687. NW_UNLOCK;
  3688. if (!ctrl->task_handle || !timeout_ms)
  3689. {
  3690. return 1;
  3691. }
  3692. uint8_t finish = 0;
  3693. OS_EVENT event;
  3694. int result;
  3695. //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
  3696. platform_start_timer(ctrl->timer, timeout_ms, 0);
  3697. while (!finish)
  3698. {
  3699. platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
  3700. switch (event.ID)
  3701. {
  3702. case EV_NW_RESULT_CONNECT:
  3703. result = (int)event.Param1;
  3704. finish = 1;
  3705. break;
  3706. case EV_NW_TIMEOUT:
  3707. result = -1;
  3708. finish = 1;
  3709. break;
  3710. default:
  3711. if (ctrl->user_callback)
  3712. {
  3713. ctrl->user_callback((void *)&event, ctrl->user_data);
  3714. }
  3715. break;
  3716. }
  3717. }
  3718. platform_stop_timer(ctrl->timer);
  3719. return result;
  3720. }
  3721. int network_listen(network_ctrl_t *ctrl, uint32_t timeout_ms)
  3722. {
  3723. if (NW_STATE_LISTEN == ctrl->state)
  3724. {
  3725. DBG("socket %d is listen", ctrl->socket_id);
  3726. return 0;
  3727. }
  3728. if (ctrl->socket_id >= 0)
  3729. {
  3730. return -1;
  3731. }
  3732. NW_LOCK;
  3733. ctrl->is_server_mode = 1;
  3734. ctrl->auto_mode = 1;
  3735. ctrl->need_close = 0;
  3736. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3737. ctrl->wait_target_state = NW_WAIT_ON_LINE;
  3738. if (!network_check_ready(ctrl))
  3739. {
  3740. ctrl->state = NW_STATE_LINK_OFF;
  3741. goto NETWORK_LISTEN_WAIT;
  3742. }
  3743. if (network_base_connect(ctrl, NULL))
  3744. {
  3745. ctrl->state = NW_STATE_OFF_LINE;
  3746. ctrl->wait_target_state = NW_WAIT_NONE;
  3747. NW_UNLOCK;
  3748. return -1;
  3749. }
  3750. ctrl->state = NW_STATE_CONNECTING;
  3751. NETWORK_LISTEN_WAIT:
  3752. NW_UNLOCK;
  3753. if (!ctrl->task_handle || !timeout_ms)
  3754. {
  3755. return 1;
  3756. }
  3757. uint8_t finish = 0;
  3758. OS_EVENT event;
  3759. int result;
  3760. //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
  3761. platform_start_timer(ctrl->timer, timeout_ms, 0);
  3762. while (!finish)
  3763. {
  3764. platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
  3765. switch (event.ID)
  3766. {
  3767. case EV_NW_RESULT_CONNECT:
  3768. result = (int)event.Param1;
  3769. finish = 1;
  3770. break;
  3771. case EV_NW_TIMEOUT:
  3772. result = -1;
  3773. finish = 1;
  3774. break;
  3775. default:
  3776. if (ctrl->user_callback)
  3777. {
  3778. ctrl->user_callback((void *)&event, ctrl->user_data);
  3779. }
  3780. break;
  3781. }
  3782. }
  3783. platform_stop_timer(ctrl->timer);
  3784. return result;
  3785. }
  3786. int network_close(network_ctrl_t *ctrl, uint32_t timeout_ms)
  3787. {
  3788. NW_LOCK;
  3789. if (ctrl->cache_data)
  3790. {
  3791. free(ctrl->cache_data);
  3792. ctrl->cache_data = NULL;
  3793. }
  3794. uint8_t old_state = ctrl->state;
  3795. ctrl->auto_mode = 1;
  3796. ctrl->need_close = 0;
  3797. network_adapter_t *adapter = &prv_adapter_table[ctrl->adapter_index];
  3798. #ifdef LUAT_USE_TLS
  3799. if (ctrl->tls_mode)
  3800. {
  3801. mbedtls_ssl_free(ctrl->ssl);
  3802. }
  3803. #endif
  3804. if (ctrl->socket_id < 0)
  3805. {
  3806. ctrl->state = NW_STATE_OFF_LINE;
  3807. ctrl->wait_target_state = NW_WAIT_NONE;
  3808. NW_UNLOCK;
  3809. return 0;
  3810. }
  3811. ctrl->state = NW_STATE_DISCONNECTING;
  3812. ctrl->wait_target_state = NW_WAIT_OFF_LINE;
  3813. if ((NW_STATE_ONLINE == old_state) && ctrl->is_tcp)
  3814. {
  3815. if (network_socket_disconnect(ctrl))
  3816. {
  3817. network_force_close_socket(ctrl);
  3818. ctrl->state = NW_STATE_OFF_LINE;
  3819. ctrl->wait_target_state = NW_WAIT_NONE;
  3820. NW_UNLOCK;
  3821. return 0;
  3822. }
  3823. }
  3824. else
  3825. {
  3826. network_force_close_socket(ctrl);
  3827. ctrl->state = NW_STATE_OFF_LINE;
  3828. ctrl->wait_target_state = NW_WAIT_NONE;
  3829. NW_UNLOCK;
  3830. return 0;
  3831. }
  3832. NW_UNLOCK;
  3833. if (!ctrl->task_handle || !timeout_ms)
  3834. {
  3835. return 1;
  3836. }
  3837. uint8_t finish = 0;
  3838. OS_EVENT event;
  3839. int result;
  3840. //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
  3841. platform_start_timer(ctrl->timer, timeout_ms, 0);
  3842. while (!finish)
  3843. {
  3844. platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
  3845. switch (event.ID)
  3846. {
  3847. case EV_NW_RESULT_CLOSE:
  3848. result = 0;
  3849. finish = 1;
  3850. break;
  3851. case EV_NW_TIMEOUT:
  3852. result = 0;
  3853. finish = 1;
  3854. break;
  3855. default:
  3856. if (ctrl->user_callback)
  3857. {
  3858. ctrl->user_callback((void *)&event, ctrl->user_data);
  3859. }
  3860. break;
  3861. }
  3862. }
  3863. platform_stop_timer(ctrl->timer);
  3864. network_socket_force_close(ctrl);
  3865. return result;
  3866. }
  3867. /*
  3868. * timeout_ms=0时,为非阻塞接口
  3869. */
  3870. 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)
  3871. {
  3872. if ((ctrl->need_close) || (ctrl->socket_id < 0) || (ctrl->state != NW_STATE_ONLINE))
  3873. {
  3874. return -1;
  3875. }
  3876. NW_LOCK;
  3877. int result;
  3878. ctrl->auto_mode = 1;
  3879. #ifdef LUAT_USE_TLS
  3880. if (ctrl->tls_mode)
  3881. {
  3882. if (ctrl->tls_need_reshakehand)
  3883. {
  3884. ctrl->tls_need_reshakehand = 0;
  3885. if (ctrl->cache_data)
  3886. {
  3887. free(ctrl->cache_data);
  3888. ctrl->cache_data = NULL;
  3889. }
  3890. ctrl->cache_data = malloc(len);
  3891. memcpy(ctrl->cache_data, data, len);
  3892. ctrl->cache_len = len;
  3893. mbedtls_ssl_session_reset(ctrl->ssl);
  3894. do
  3895. {
  3896. result = mbedtls_ssl_handshake_step( ctrl->ssl );
  3897. switch(result)
  3898. {
  3899. case MBEDTLS_ERR_SSL_WANT_READ:
  3900. ctrl->state = NW_STATE_SHAKEHAND;
  3901. goto NETWORK_TX_WAIT;
  3902. case 0:
  3903. break;
  3904. default:
  3905. DBG_ERR("0x%x, %d", -result, ctrl->ssl->state);
  3906. ctrl->need_close = 1;
  3907. NW_UNLOCK;
  3908. return -1;
  3909. }
  3910. }while(ctrl->ssl->state != MBEDTLS_SSL_HANDSHAKE_OVER);
  3911. }
  3912. result = mbedtls_ssl_write(ctrl->ssl, data, len);
  3913. if (result < 0)
  3914. {
  3915. DBG("%08x", -result);
  3916. ctrl->need_close = 1;
  3917. NW_UNLOCK;
  3918. return -1;
  3919. }
  3920. *tx_len = result;
  3921. }
  3922. else
  3923. #endif
  3924. {
  3925. result = network_base_tx(ctrl, data, len, flags, remote_ip, remote_port);
  3926. if (result < 0)
  3927. {
  3928. ctrl->need_close = 1;
  3929. NW_UNLOCK;
  3930. return -1;
  3931. }
  3932. *tx_len = result;
  3933. if (!result && len)
  3934. {
  3935. NW_UNLOCK;
  3936. return 0;
  3937. }
  3938. }
  3939. #ifdef LUAT_USE_TLS
  3940. NETWORK_TX_WAIT:
  3941. #endif
  3942. ctrl->wait_target_state = NW_WAIT_TX_OK;
  3943. NW_UNLOCK;
  3944. if (!ctrl->task_handle || !timeout_ms)
  3945. {
  3946. return 1;
  3947. }
  3948. uint8_t finish = 0;
  3949. OS_EVENT event;
  3950. //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
  3951. platform_start_timer(ctrl->timer, timeout_ms, 0);
  3952. while (!finish)
  3953. {
  3954. platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
  3955. switch (event.ID)
  3956. {
  3957. case EV_NW_RESULT_TX:
  3958. result = (int)event.Param1;
  3959. finish = 1;
  3960. break;
  3961. case EV_NW_TIMEOUT:
  3962. result = -1;
  3963. finish = 1;
  3964. break;
  3965. default:
  3966. if (ctrl->user_callback)
  3967. {
  3968. ctrl->user_callback((void *)&event, ctrl->user_data);
  3969. }
  3970. break;
  3971. }
  3972. }
  3973. platform_stop_timer(ctrl->timer);
  3974. return result;
  3975. }
  3976. /*
  3977. * 实际读到的数据量在rx_len里,如果是UDP模式且为server时,需要看remote_ip和remote_port
  3978. */
  3979. 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)
  3980. {
  3981. if (!ctrl->new_rx_flag && ((ctrl->need_close) || (ctrl->socket_id < 0) || (ctrl->state != NW_STATE_ONLINE)))
  3982. {
  3983. return -1;
  3984. }
  3985. NW_LOCK;
  3986. int result = -1;
  3987. ctrl->auto_mode = 1;
  3988. uint32_t read_len = 0;
  3989. uint8_t is_error = 0;
  3990. if (data)
  3991. {
  3992. ctrl->new_rx_flag = 0;
  3993. #ifdef LUAT_USE_TLS
  3994. if (ctrl->tls_mode)
  3995. {
  3996. do
  3997. {
  3998. result = mbedtls_ssl_read(ctrl->ssl, data + read_len, len - read_len);
  3999. if (result < 0 && (result != MBEDTLS_ERR_SSL_WANT_READ))
  4000. {
  4001. is_error = 1;
  4002. break;
  4003. }
  4004. else
  4005. {
  4006. read_len += result;
  4007. if (read_len >= len)
  4008. {
  4009. break;
  4010. }
  4011. }
  4012. }while(network_socket_receive(ctrl, NULL, len, flags, remote_ip, remote_port) > 0);
  4013. if ( !is_error )
  4014. {
  4015. result = read_len;
  4016. }
  4017. else
  4018. {
  4019. result = -1;
  4020. }
  4021. }
  4022. else
  4023. #endif
  4024. {
  4025. result = network_socket_receive(ctrl, data, len, flags, remote_ip, remote_port);
  4026. }
  4027. }
  4028. else
  4029. {
  4030. #ifdef LUAT_USE_TLS
  4031. if (ctrl->tls_mode)
  4032. {
  4033. read_len = 0;
  4034. do
  4035. {
  4036. result = mbedtls_ssl_read(ctrl->ssl, NULL, 0);
  4037. if ((result < 0) && (result != (MBEDTLS_ERR_SSL_WANT_READ)))
  4038. {
  4039. is_error = 1;
  4040. read_len = 0;
  4041. break;
  4042. }
  4043. else if (!result)
  4044. {
  4045. read_len = ctrl->ssl->in_msglen;
  4046. break;
  4047. }
  4048. else if ((MBEDTLS_ERR_SSL_WANT_READ) == result)
  4049. {
  4050. read_len = 0;
  4051. ctrl->new_rx_flag = 0;
  4052. DBG("socket %d ssl data need more", ctrl->socket_id);
  4053. break;
  4054. }
  4055. }while(network_socket_receive(ctrl, NULL, len, flags, remote_ip, remote_port) > 0);
  4056. if ( !is_error )
  4057. {
  4058. result = read_len;
  4059. }
  4060. else
  4061. {
  4062. result = -1;
  4063. }
  4064. }
  4065. else
  4066. #endif
  4067. {
  4068. result = network_socket_receive(ctrl, data, len, flags, remote_ip, remote_port);
  4069. }
  4070. }
  4071. NW_UNLOCK;
  4072. if (result >= 0)
  4073. {
  4074. *rx_len = result;
  4075. return 0;
  4076. }
  4077. else
  4078. {
  4079. return -1;
  4080. }
  4081. }
  4082. int network_wait_event(network_ctrl_t *ctrl, OS_EVENT *out_event, uint32_t timeout_ms, uint8_t *is_timeout)
  4083. {
  4084. if (ctrl->new_rx_flag)
  4085. {
  4086. ctrl->wait_target_state = NW_WAIT_EVENT;
  4087. if (out_event)
  4088. {
  4089. out_event->ID = 0;
  4090. }
  4091. return 0;
  4092. }
  4093. if ((ctrl->need_close) || (ctrl->socket_id < 0) || (ctrl->state != NW_STATE_ONLINE))
  4094. {
  4095. return -1;
  4096. }
  4097. NW_LOCK;
  4098. ctrl->auto_mode = 1;
  4099. ctrl->wait_target_state = NW_WAIT_EVENT;
  4100. NW_UNLOCK;
  4101. if (!ctrl->task_handle || !timeout_ms)
  4102. {
  4103. return 1;
  4104. }
  4105. *is_timeout = 0;
  4106. uint8_t finish = 0;
  4107. OS_EVENT event;
  4108. int result;
  4109. //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
  4110. platform_start_timer(ctrl->timer, timeout_ms, 0);
  4111. while (!finish)
  4112. {
  4113. platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
  4114. switch (event.ID)
  4115. {
  4116. case EV_NW_RESULT_EVENT:
  4117. result = (int)event.Param1;
  4118. if (result)
  4119. {
  4120. result = -1;
  4121. }
  4122. if (out_event)
  4123. {
  4124. out_event->ID = 0;
  4125. }
  4126. finish = 1;
  4127. break;
  4128. case EV_NW_TIMEOUT:
  4129. *is_timeout = 1;
  4130. result = 0;
  4131. finish = 1;
  4132. break;
  4133. case EV_NW_BREAK_WAIT:
  4134. if (out_event)
  4135. {
  4136. *out_event = event;
  4137. }
  4138. result = 0;
  4139. finish = 1;
  4140. break;
  4141. default:
  4142. if (out_event)
  4143. {
  4144. *out_event = event;
  4145. result = 0;
  4146. finish = 1;
  4147. break;
  4148. }
  4149. else if (ctrl->user_callback)
  4150. {
  4151. ctrl->user_callback((void *)&event, ctrl->user_data);
  4152. }
  4153. break;
  4154. }
  4155. }
  4156. platform_stop_timer(ctrl->timer);
  4157. return result;
  4158. }
  4159. int network_wait_rx(network_ctrl_t *ctrl, uint32_t timeout_ms, uint8_t *is_break, uint8_t *is_timeout)
  4160. {
  4161. *is_timeout = 0;
  4162. *is_break = 0;
  4163. if (ctrl->new_rx_flag)
  4164. {
  4165. ctrl->wait_target_state = NW_WAIT_EVENT;
  4166. return 0;
  4167. }
  4168. if ((ctrl->need_close) || (ctrl->socket_id < 0) || (ctrl->state != NW_STATE_ONLINE))
  4169. {
  4170. return -1;
  4171. }
  4172. NW_LOCK;
  4173. ctrl->auto_mode = 1;
  4174. ctrl->wait_target_state = NW_WAIT_EVENT;
  4175. NW_UNLOCK;
  4176. uint8_t finish = 0;
  4177. OS_EVENT event;
  4178. int result;
  4179. //DBG_INFO("%s wait for active!,%u,%x", Net->Tag, To * SYS_TICK, Net->hTask);
  4180. if (timeout_ms)
  4181. {
  4182. platform_start_timer(ctrl->timer, timeout_ms, 0);
  4183. }
  4184. while (!finish)
  4185. {
  4186. platform_wait_event(ctrl->task_handle, 0, &event, NULL, 0);
  4187. switch (event.ID)
  4188. {
  4189. case EV_NW_RESULT_EVENT:
  4190. result = (int)event.Param1;
  4191. if (result)
  4192. {
  4193. result = -1;
  4194. finish = 1;
  4195. }
  4196. else if (ctrl->new_rx_flag)
  4197. {
  4198. result = 0;
  4199. finish = 1;
  4200. }
  4201. break;
  4202. case EV_NW_TIMEOUT:
  4203. *is_timeout = 1;
  4204. result = 0;
  4205. finish = 1;
  4206. break;
  4207. case EV_NW_BREAK_WAIT:
  4208. *is_break = 1;
  4209. result = 0;
  4210. finish = 1;
  4211. break;
  4212. default:
  4213. if (ctrl->user_callback)
  4214. {
  4215. ctrl->user_callback((void *)&event, ctrl->user_data);
  4216. }
  4217. break;
  4218. }
  4219. ctrl->wait_target_state = NW_WAIT_EVENT;
  4220. }
  4221. platform_stop_timer(ctrl->timer);
  4222. return result;
  4223. }
  4224. #endif
  4225. #endif