RyanW5500Socket.c 57 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706
  1. #define rlogEnable // 是否使能日志
  2. #define rlogColorEnable // 是否使能日志颜色
  3. #define rlogLevel (rlogLvlWarning) // 日志打印等级
  4. #define rlogTag "W5500Socket" // 日志tag
  5. #include "RyanW5500Store.h"
  6. #define RyanW5500LinkCheck(value) \
  7. do \
  8. { \
  9. uint8_t linkState = 0; \
  10. ctlwizchip(CW_GET_PHYLINK, &linkState); \
  11. if (PHY_LINK_ON != linkState) \
  12. { \
  13. rt_set_errno(EHOSTUNREACH); \
  14. return value; \
  15. } \
  16. } while (0);
  17. // 可用套接字的全局数组
  18. static RyanW5500Socket RyanW5500Sockets[RyanW5500MaxSocketNum] = {0};
  19. static uint16_t wiz_port = 15500; // 用户可以自定义
  20. static int _isalpha(char ch)
  21. {
  22. return (ch >= 'a' && ch <= 'z') || (ch >= 'A' && ch <= 'Z');
  23. }
  24. /**
  25. * @brief
  26. *
  27. * @param s_addr
  28. * @param ipStrArr
  29. */
  30. static void inAddrToipStrArr(in_addr_t *s_addr, uint8_t *ipStrArr)
  31. {
  32. RT_ASSERT(NULL != s_addr);
  33. RT_ASSERT(NULL != ipStrArr);
  34. // inet_pton(AF_INET, inet_ntoa(sin->sin_addr), &ipStrArr); // 效率有点低
  35. uint8_t *p = (uint8_t *)s_addr;
  36. ipStrArr[0] = *p;
  37. ipStrArr[1] = *(p + 1);
  38. ipStrArr[2] = *(p + 2);
  39. ipStrArr[3] = *(p + 3);
  40. }
  41. /**
  42. * @brief
  43. *
  44. * @param ipStrArr
  45. * @return in_addr_t
  46. */
  47. static in_addr_t ipStrArrToinAddr(uint8_t *ipStrArr)
  48. {
  49. RT_ASSERT(NULL != ipStrArr);
  50. // 效率有点低
  51. // char remote_ipaddr[16] = {0};
  52. // snprintf(remote_ipaddr, sizeof(remote_ipaddr), "%d.%d.%d.%d", ipStrArr[0], ipStrArr[1], ipStrArr[2], ipStrArr[3]);
  53. // return inet_addr((const char *)remote_ipaddr);
  54. in_addr_t p2 = 0;
  55. uint8_t *p = (uint8_t *)&p2;
  56. *p = ipStrArr[0];
  57. *(p + 1) = ipStrArr[1];
  58. *(p + 2) = ipStrArr[2];
  59. *(p + 3) = ipStrArr[3];
  60. return p2;
  61. }
  62. /**
  63. * @brief 新的listen客户端连接
  64. *
  65. * @param serviceSock
  66. * @param clientSock
  67. */
  68. static void RyanW5500DoEventChanges(RyanW5500Socket *sock, RyanW5500Event_e event, rt_bool_t is_plus)
  69. {
  70. RT_ASSERT(NULL != sock);
  71. switch (event)
  72. {
  73. case WIZ_EVENT_SEND:
  74. {
  75. if (is_plus)
  76. {
  77. sock->sendevent = 1;
  78. #ifdef SAL_USING_POSIX
  79. rt_wqueue_wakeup(&sock->wait_head, (void *)POLLOUT);
  80. #endif
  81. }
  82. else if (sock->sendevent)
  83. {
  84. sock->sendevent = 0;
  85. }
  86. break;
  87. }
  88. case WIZ_EVENT_RECV:
  89. {
  90. if (is_plus)
  91. {
  92. sock->rcvevent++;
  93. #ifdef SAL_USING_POSIX
  94. rt_wqueue_wakeup(&sock->wait_head, (void *)POLLIN);
  95. #endif
  96. }
  97. else if (sock->rcvevent)
  98. {
  99. sock->rcvevent--;
  100. }
  101. break;
  102. }
  103. case WIZ_EVENT_ERROR:
  104. {
  105. if (is_plus)
  106. {
  107. sock->errevent++;
  108. #ifdef SAL_USING_POSIX
  109. rt_wqueue_wakeup(&sock->wait_head, (void *)POLLERR);
  110. #endif
  111. }
  112. else if (sock->errevent)
  113. {
  114. sock->errevent--;
  115. }
  116. break;
  117. }
  118. default:
  119. rlog_e("Not supported event (%d)", event);
  120. }
  121. }
  122. static void wiz_do_event_clean(RyanW5500Socket *sock, RyanW5500Event_e event)
  123. {
  124. switch (event)
  125. {
  126. case WIZ_EVENT_SEND:
  127. {
  128. sock->sendevent = 0;
  129. break;
  130. }
  131. case WIZ_EVENT_RECV:
  132. {
  133. sock->rcvevent = 0;
  134. break;
  135. }
  136. case WIZ_EVENT_ERROR:
  137. {
  138. sock->errevent = 0;
  139. break;
  140. }
  141. default:
  142. rlog_e("Not supported event (%d)", event);
  143. }
  144. }
  145. /**
  146. * @brief 中断接收到数据回调函数
  147. *
  148. * @param socket
  149. * @return int
  150. */
  151. int RyanW5500RecvDataCallback(int socket)
  152. {
  153. RyanW5500Socket *sock = NULL;
  154. sock = RyanW5500GetSock(socket);
  155. RyanW5500CheckCode(NULL != sock, EBADF, rlog_d, { return -1; });
  156. rt_event_send(RyanW5500Entry.W5500EventHandle, 1 << sock->socket);
  157. RyanW5500DoEventChanges(sock, WIZ_EVENT_RECV, RT_TRUE);
  158. return 0;
  159. }
  160. /**
  161. * @brief 中断接收到close信号回调函数
  162. *
  163. * @param socket
  164. * @return int
  165. */
  166. int RyanW5500CloseCallback(int socket)
  167. {
  168. RyanW5500Socket *sock = NULL;
  169. sock = RyanW5500GetSock(socket);
  170. RyanW5500CheckCode(NULL != sock, EBADF, rlog_d, { return -1; });
  171. sock->state = RyanW5500SocketClose;
  172. RyanW5500DoEventChanges(sock, WIZ_EVENT_RECV, RT_TRUE);
  173. RyanW5500DoEventChanges(sock, WIZ_EVENT_ERROR, RT_TRUE);
  174. rt_event_send(RyanW5500Entry.W5500EventHandle, 1 << sock->socket);
  175. return 0;
  176. }
  177. /**
  178. * @brief 根据socket获取RyanW5500Socket结构体
  179. *
  180. * @param socket
  181. * @return RyanW5500Socket*
  182. */
  183. RyanW5500Socket *RyanW5500GetSock(int socket)
  184. {
  185. if (socket < 0 || socket >= RyanW5500MaxSocketNum)
  186. {
  187. rlog_d("socket 越界");
  188. return NULL;
  189. }
  190. // 检查套接字结构是否有效
  191. if (RyanW5500Sockets[socket].magic != WIZ_SOCKET_MAGIC)
  192. {
  193. rlog_d("套接字结构无效");
  194. return NULL;
  195. }
  196. return &RyanW5500Sockets[socket];
  197. }
  198. /**
  199. * @brief 给listen服务器套接字创建client客户端
  200. *
  201. * @param serviceSock
  202. * @return RyanW5500Socket*
  203. */
  204. static RyanW5500Socket *RyanW5500CreateListenClient(RyanW5500Socket *serviceSock)
  205. {
  206. RT_ASSERT(NULL != serviceSock);
  207. RyanW5500Socket *clientSock = NULL;
  208. RyanW5500ClientInfo *clientInfo = NULL;
  209. clientSock = RyanW5500SocketCreate(serviceSock->type, serviceSock->port);
  210. RyanW5500CheckCodeNoReturn(NULL != clientSock, EMFILE, rlog_d, { goto err; });
  211. clientSock->serviceSocket = serviceSock->socket;
  212. // 创建客户端信息并将客户端添加到服务器clientList
  213. clientInfo = (RyanW5500ClientInfo *)malloc(sizeof(RyanW5500ClientInfo));
  214. RyanW5500CheckCodeNoReturn(NULL != clientInfo, ENOMEM, rlog_d, { goto err; });
  215. memset(clientInfo, 0, sizeof(RyanW5500ClientInfo));
  216. clientInfo->sock = clientSock;
  217. RyanListAddTail(&clientInfo->list, &serviceSock->serviceInfo->clientList);
  218. RyanW5500CheckCodeNoReturn(SOCK_OK == wizchip_listen(clientSock->socket), EPROTO, rlog_d, { wiz_closesocket(clientSock->socket); goto err; });
  219. clientSock->state = RyanW5500SocketListen;
  220. return clientSock;
  221. err:
  222. if (NULL != clientInfo)
  223. wiz_closesocket(clientSock->socket);
  224. if (NULL != clientInfo)
  225. free(clientInfo);
  226. return NULL;
  227. }
  228. static void RyanW5500ListenSocketDestory(RyanW5500Socket *sock)
  229. {
  230. RyanW5500Socket *serviceSock = RyanW5500GetSock(sock->serviceSocket);
  231. RT_ASSERT(NULL != serviceSock);
  232. RT_ASSERT(NULL != serviceSock->serviceInfo);
  233. RT_ASSERT(sock->port == serviceSock->port);
  234. // 分配并初始化新的客户端套接字
  235. RyanList_t *curr = NULL,
  236. *next = NULL;
  237. RyanW5500ClientInfo *clientInfo = NULL;
  238. RyanListForEachSafe(curr, next, &serviceSock->serviceInfo->clientList)
  239. {
  240. // 获取此节点的结构体
  241. clientInfo = RyanListEntry(curr, RyanW5500ClientInfo, list);
  242. RT_ASSERT(NULL != clientInfo);
  243. if (clientInfo->sock->socket != sock->socket)
  244. continue;
  245. RyanListDel(&clientInfo->list);
  246. // 增加listen客户端数
  247. if (0 == serviceSock->serviceInfo->backlog)
  248. {
  249. RyanW5500Socket *sock = RyanW5500CreateListenClient(serviceSock);
  250. if (NULL == sock)
  251. {
  252. int8_t socket = -1;
  253. rt_mq_send(serviceSock->serviceInfo->clientInfoQueueHandle, &socket, sizeof(int8_t));
  254. }
  255. }
  256. // 添加服务器套接字监听数
  257. serviceSock->serviceInfo->backlog++;
  258. free(clientInfo);
  259. break;
  260. }
  261. }
  262. static int RyanW5500SocketDestory(RyanW5500Socket *sock)
  263. {
  264. RT_ASSERT(NULL != sock);
  265. rlog_d("销毁套接字");
  266. rt_mutex_take(RyanW5500Entry.socketMutexHandle, RT_WAITING_FOREVER); //
  267. if (sock->magic != WIZ_SOCKET_MAGIC)
  268. {
  269. rt_mutex_release(RyanW5500Entry.socketMutexHandle); // 释放互斥锁
  270. return -1;
  271. }
  272. if (sock->recvLock)
  273. rt_mutex_delete(sock->recvLock);
  274. if (sock->remoteAddr)
  275. free(sock->remoteAddr);
  276. if (-1 != sock->serviceSocket) // 根据记录的serviceSocket判断是否为listen客户端
  277. RyanW5500ListenSocketDestory(sock);
  278. // listen服务器套接字,释放服务器信息并关闭所有客户端套接字clientList
  279. else if (NULL != sock->serviceInfo)
  280. {
  281. RyanW5500ClientInfo *clientInfo = NULL;
  282. // 分配并初始化新的客户端套接字
  283. RyanList_t *curr = NULL,
  284. *next = NULL;
  285. RyanListForEachSafe(curr, next, &sock->serviceInfo->clientList)
  286. {
  287. // 获取此节点的结构体
  288. clientInfo = RyanListEntry(curr, RyanW5500ClientInfo, list);
  289. if (NULL == clientInfo)
  290. continue;
  291. RyanListDel(&clientInfo->list);
  292. wiz_closesocket(clientInfo->sock->socket);
  293. free(clientInfo);
  294. }
  295. if (sock->serviceInfo->clientInfoQueueHandle)
  296. rt_mq_delete(sock->serviceInfo->clientInfoQueueHandle);
  297. if (sock->serviceInfo)
  298. free(sock->serviceInfo);
  299. }
  300. setSn_IR(sock->socket, 0xff); // 清空套接字中断
  301. setSn_IMR(sock->socket, 0); // 设置套接字ISR状态支持
  302. memset(sock, 0, sizeof(RyanW5500Socket));
  303. rt_mutex_release(RyanW5500Entry.socketMutexHandle); // 释放互斥锁
  304. return 0;
  305. }
  306. /**
  307. * @brief 创建socket
  308. *
  309. * @param type
  310. * @param port
  311. * @return RyanW5500Socket*
  312. */
  313. RyanW5500Socket *RyanW5500SocketCreate(int type, int port)
  314. {
  315. RyanW5500Socket *sock = NULL;
  316. uint8_t idx = 0;
  317. rt_mutex_take(RyanW5500Entry.socketMutexHandle, RT_WAITING_FOREVER); // 获取互斥锁
  318. // 找到一个空的 WIZnet 套接字条目
  319. for (idx = 0; idx < RyanW5500MaxSocketNum && WIZ_SOCKET_MAGIC == RyanW5500Sockets[idx].magic; idx++)
  320. ;
  321. // 没有空闲socket
  322. RyanW5500CheckCodeNoReturn(RyanW5500MaxSocketNum != idx, EMFILE, rlog_d, {
  323. rt_mutex_release(RyanW5500Entry.socketMutexHandle); // 释放互斥锁
  324. goto err; });
  325. sock = &(RyanW5500Sockets[idx]);
  326. sock->magic = WIZ_SOCKET_MAGIC;
  327. sock->socket = idx;
  328. sock->type = type;
  329. sock->port = port;
  330. sock->state = RyanW5500SocketInit;
  331. sock->remoteAddr = NULL;
  332. sock->recvTimeout = 0;
  333. sock->sendTimeout = 0;
  334. sock->serviceInfo = NULL; // 用于服务器套接字
  335. sock->serviceSocket = -1;
  336. // 创建接收锁
  337. char name[32] = {0};
  338. rt_snprintf(name, sizeof(name), "%s%d", "w5500_sr", idx);
  339. sock->recvLock = rt_mutex_create(name, RT_IPC_FLAG_FIFO);
  340. if (RT_NULL == sock->recvLock)
  341. {
  342. rt_mutex_release(RyanW5500Entry.socketMutexHandle); // 释放互斥锁
  343. goto err;
  344. }
  345. #ifdef SAL_USING_POSIX
  346. rt_wqueue_init(&sock->wait_head);
  347. #endif
  348. setSn_IR(sock->socket, 0xff); // 清空套接字中断
  349. setSn_IMR(sock->socket, RyanW5500SnIMR); // 设置套接字ISR状态支持
  350. wiz_recv_ignore(sock->socket, UINT16_MAX); // 清空之前的接收缓存
  351. int8_t result = wizchip_socket(sock->socket, sock->type, sock->port, 0); // wizchip_socket内部会先close一次
  352. if (result != sock->socket)
  353. {
  354. rlog_e("RyanW5500 socket(%d) create failed! result: %d", sock->socket, result);
  355. rt_mutex_release(RyanW5500Entry.socketMutexHandle); // 释放互斥锁
  356. goto err;
  357. }
  358. sock->state = RyanW5500SocketInit;
  359. rt_mutex_release(RyanW5500Entry.socketMutexHandle); // 释放互斥锁
  360. return sock;
  361. err:
  362. wiz_closesocket(sock->socket);
  363. return NULL;
  364. }
  365. /**
  366. * @brief 新的listen客户端连接
  367. *
  368. * @param serviceSock
  369. * @param clientSock
  370. */
  371. void RyanListenServiceAddClient(RyanW5500Socket *serviceSock, RyanW5500Socket *clientSock)
  372. {
  373. RT_ASSERT(NULL != serviceSock);
  374. RT_ASSERT(NULL != clientSock);
  375. RT_ASSERT(serviceSock->serviceInfo->backlog > 0);
  376. int8_t socket = -1;
  377. if (serviceSock->serviceInfo->backlog < 0)
  378. {
  379. goto __exit;
  380. }
  381. serviceSock->serviceInfo->backlog--;
  382. clientSock->state = RyanW5500SocketEstablished;
  383. if (serviceSock->serviceInfo->backlog > 0)
  384. {
  385. rlog_w("listen回调连接");
  386. RyanW5500Socket *sock = RyanW5500CreateListenClient(serviceSock);
  387. if (NULL == sock)
  388. {
  389. goto __exit;
  390. }
  391. }
  392. socket = clientSock->socket;
  393. rt_mq_send(serviceSock->serviceInfo->clientInfoQueueHandle, &socket, sizeof(int8_t));
  394. return;
  395. __exit:
  396. rt_mq_send(serviceSock->serviceInfo->clientInfoQueueHandle, &socket, sizeof(int8_t));
  397. RyanW5500DoEventChanges(serviceSock, WIZ_EVENT_ERROR, RT_TRUE);
  398. }
  399. /**
  400. * @brief 创建套接字
  401. *
  402. * @param domain 协议族类型
  403. * @param type 协议类型
  404. * @param protocol 实际使用的运输层协议
  405. * @return int 返回一个代表套接字描述符的整数
  406. */
  407. int wiz_socket(int domain, int type, int protocol)
  408. {
  409. RyanW5500LinkCheck(-1);
  410. RyanW5500Socket *sock = NULL;
  411. // 该实现不支持指定的协议族类型。
  412. RyanW5500CheckCode(AF_INET == domain || AF_WIZ == domain, EAFNOSUPPORT, rlog_d, { return -1; });
  413. // 不支持特定的运输层协议
  414. // RyanW5500CheckCode(0 == protocol, EPROTONOSUPPORT, rlog_d, { return -1; });
  415. switch (type)
  416. {
  417. case SOCK_STREAM:
  418. case SOCK_DGRAM:
  419. case SOCK_RAW:
  420. break;
  421. default:
  422. // 协议不支持套接字类型
  423. RyanW5500CheckCode(NULL, EPROTOTYPE, rlog_d, { return -1; });
  424. }
  425. // 分配并初始化一个新的 WIZnet 套接字
  426. wiz_port++;
  427. sock = RyanW5500SocketCreate(type, wiz_port);
  428. RyanW5500CheckCode(NULL != sock, EMFILE, rlog_d, { return -1; });
  429. return sock->socket;
  430. }
  431. /**
  432. * @brief 该函数用于将端口号和 IP 地址绑定带指定套接字上。
  433. *
  434. * @param socket 套接字描述符
  435. * @param name 指向 sockaddr 结构体的指针
  436. * @param namelen sockaddr 结构体的长度
  437. * @return int
  438. */
  439. int wiz_bind(int socket, const struct sockaddr *name, socklen_t namelen)
  440. {
  441. RyanW5500LinkCheck(-1);
  442. RyanW5500Socket *sock = NULL;
  443. uint16_t port = 0;
  444. RyanW5500CheckCode(NULL != name && 0 != namelen, EAFNOSUPPORT, rlog_d, { return -1; }); // 非法地址
  445. sock = RyanW5500GetSock(socket);
  446. RyanW5500CheckCode(NULL != sock, EBADF, rlog_d, { return -1; });
  447. RyanW5500CheckCode(WIZ_SOCKET_MAGIC == sock->magic, EIO, rlog_d, { return -1; });
  448. // PRASE IP 地址和端口
  449. const struct sockaddr_in *sin = (const struct sockaddr_in *)name;
  450. port = (uint16_t)htons(sin->sin_port);
  451. if (sock->port == port) // 相等,不需要重新进行绑定
  452. return 0;
  453. #ifdef SAL_USING_POSIX
  454. rt_wqueue_init(&sock->wait_head); // 重置一下wqueue
  455. #endif
  456. sock->port = port;
  457. // wizchip_socket内部会先close一次
  458. int8_t result = wizchip_socket(sock->socket, sock->type, sock->port, 0);
  459. RyanW5500CheckCode(result == sock->socket, EMFILE, rlog_d, { return -1; });
  460. sock->state = RyanW5500SocketInit;
  461. return 0;
  462. }
  463. /**
  464. * @brief 建立连接
  465. *
  466. * @param socket 套接字描述符
  467. * @param name 服务器地址信息
  468. * @param namelen 服务器地址结构体的长度
  469. * @return int
  470. */
  471. int wiz_connect(int socket, const struct sockaddr *name, socklen_t namelen)
  472. {
  473. RyanW5500LinkCheck(-1);
  474. RyanW5500CheckCode(NULL != name && 0 != namelen, EAFNOSUPPORT, rlog_d, { return -1; }); // 非法地址
  475. RyanW5500Socket *sock = NULL;
  476. sock = RyanW5500GetSock(socket);
  477. RyanW5500CheckCode(NULL != sock, EBADF, rlog_d, { return -1; });
  478. RyanW5500CheckCode(RyanW5500SocketInit == sock->state, EADDRINUSE, rlog_d, { return -1; }); // 尝试建立 已在使用的 / 未初始化 socket的连接。
  479. switch (getSn_SR(socket))
  480. {
  481. case SOCK_UDP: // udp
  482. case SOCK_IPRAW:
  483. if (NULL == sock->remoteAddr)
  484. {
  485. sock->remoteAddr = (struct sockaddr *)malloc(sizeof(struct sockaddr));
  486. RyanW5500CheckCode(NULL != sock->remoteAddr, ENOMEM, rlog_d, { return -1; });
  487. memset(sock->remoteAddr, 0, sizeof(struct sockaddr));
  488. }
  489. // 复制远程端口,ip,协议族
  490. sock->remoteAddr->sa_len = name->sa_len;
  491. sock->remoteAddr->sa_family = name->sa_family;
  492. memcpy(sock->remoteAddr->sa_data, name->sa_data, sizeof(sock->remoteAddr->sa_data));
  493. break;
  494. case SOCK_INIT: // tcp
  495. {
  496. uint8_t ipStrArr[4] = {0};
  497. struct sockaddr_in *sin = (struct sockaddr_in *)name;
  498. inAddrToipStrArr(&sin->sin_addr.s_addr, ipStrArr);
  499. int result = wizchip_connect(socket, ipStrArr, htons(sin->sin_port));
  500. if (SOCK_OK != result)
  501. {
  502. RyanW5500DoEventChanges(sock, WIZ_EVENT_ERROR, RT_TRUE);
  503. RyanW5500CheckCode(SOCKERR_IPINVALID != result, EAFNOSUPPORT, rlog_d, { return -1; }); // 无效的 IP 地址
  504. RyanW5500CheckCode(SOCKERR_TIMEOUT != result, ETIMEDOUT, rlog_d, { return -1; }); // 连接超时
  505. return -1;
  506. }
  507. }
  508. break;
  509. default: // 无效
  510. RyanW5500DoEventChanges(sock, WIZ_EVENT_ERROR, RT_TRUE);
  511. break;
  512. }
  513. sock->state = RyanW5500SocketEstablished;
  514. RyanW5500DoEventChanges(sock, WIZ_EVENT_SEND, RT_TRUE);
  515. return 0;
  516. }
  517. /**
  518. * @brief 监听套接字
  519. *
  520. * @param socket 套接字描述符
  521. * @param backlog 表示一次能够等待的最大连接数目
  522. * @return int
  523. */
  524. int wiz_listen(int socket, int backlog)
  525. {
  526. RyanW5500LinkCheck(-1);
  527. RyanW5500Socket *sock = RyanW5500GetSock(socket);
  528. RyanW5500CheckCode(NULL != sock, EBADF, rlog_d, { return -1; }); // 套接字参数不是有效的文件描述符。
  529. RyanW5500CheckCode(WIZ_SOCKET_MAGIC == sock->magic, EIO, rlog_d, { return -1; });
  530. RyanW5500CheckCode(RyanW5500SocketEstablished != sock->state, EINVAL, rlog_d, { return -1; }); // 套接字已连接
  531. RyanW5500CheckCode(Sn_MR_IPRAW != sock->type, EOPNOTSUPP, rlog_d, { return -1; }); // 套接字协议不支持 listen()
  532. RyanW5500CheckCode(NULL == sock->serviceInfo, EINVAL, rlog_d, { return -1; }); // 套接字第一次调用listen
  533. sock->serviceInfo = (RyanW5500ServiceInfo *)malloc(sizeof(RyanW5500ServiceInfo));
  534. RyanW5500CheckCode(NULL != sock->serviceInfo, ENOMEM, rlog_d, { goto err; });
  535. memset(sock->serviceInfo, 0, sizeof(RyanW5500ServiceInfo));
  536. // 创建listen客户端连接消息队列
  537. char name[32] = {0};
  538. snprintf(name, 32, "wiz_mb%d", sock->socket);
  539. sock->serviceInfo->backlog = backlog > 0 ? backlog : 0;
  540. // BSD socket允许listen为0,这里消息队列创建1大小,为了appect可以阻塞
  541. sock->serviceInfo->clientInfoQueueHandle = rt_mq_create(name, sizeof(int8_t), backlog > 0 ? backlog : 1, RT_IPC_FLAG_FIFO);
  542. RyanW5500CheckCode(NULL != sock->serviceInfo->clientInfoQueueHandle, ENOMEM, rlog_d, { goto err; });
  543. RyanListInit(&sock->serviceInfo->clientList);
  544. sock->state = RyanW5500SocketListen;
  545. rlog_w("listen连接");
  546. if (NULL == RyanW5500CreateListenClient(sock))
  547. goto err;
  548. return 0;
  549. err:
  550. if (NULL != sock->serviceInfo)
  551. {
  552. if (NULL != sock->serviceInfo->clientInfoQueueHandle)
  553. rt_mq_delete(sock->serviceInfo->clientInfoQueueHandle);
  554. free(sock->serviceInfo);
  555. sock->serviceInfo = NULL;
  556. }
  557. return -1;
  558. }
  559. /**
  560. * @brief 接收连接
  561. *
  562. * @param socket 套接字描述符
  563. * @param addr 存储客户端设备地址结构体
  564. * @param addrlen 客户端设备地址结构体的长度
  565. * @return int
  566. */
  567. int wiz_accept(int socket, struct sockaddr *addr, socklen_t *addrlen)
  568. {
  569. RyanW5500LinkCheck(-1);
  570. RyanW5500CheckCode(NULL != addr && 0 != addrlen, EAFNOSUPPORT, rlog_d, { return -1; }); // 非法地址
  571. RyanW5500Socket *serviceSock = RyanW5500GetSock(socket);
  572. RyanW5500CheckCode(NULL != serviceSock, EBADF, rlog_d, { return -1; }); // 套接字参数不是有效的文件描述符。
  573. RyanW5500CheckCode(WIZ_SOCKET_MAGIC == serviceSock->magic, EIO, rlog_d, { return -1; });
  574. RyanW5500CheckCode(RyanW5500SocketListen == serviceSock->state, EINVAL, rlog_d, { return -1; }); // 套接字不接受连接,没有listen
  575. RyanW5500CheckCode(NULL != serviceSock->serviceInfo, ENOBUFS, rlog_d, { return -1; }); // 没有可用的缓冲区空间,理论上不会出现
  576. while (1)
  577. {
  578. int8_t clientSocket = -1;
  579. // 接收客户端连接消息
  580. #if (RT_VER_NUM < 0x50001)
  581. if (rt_mq_recv(serviceSock->serviceInfo->clientInfoQueueHandle, (void *)&clientSocket, sizeof(int8_t), RT_WAITING_FOREVER) != RT_EOK)
  582. continue;
  583. #else
  584. if (rt_mq_recv(serviceSock->serviceInfo->clientInfoQueueHandle, (void *)&clientSocket, sizeof(int8_t), RT_WAITING_FOREVER) > 0)
  585. continue;
  586. #endif
  587. RyanW5500CheckCode(-1 != clientSocket, EPROTO, rlog_d, { return -1; });
  588. // 检查连接消息类型
  589. if (SOCK_ESTABLISHED != getSn_SR(clientSocket))
  590. {
  591. // 清除服务套接字错误
  592. wiz_do_event_clean(serviceSock, WIZ_EVENT_ERROR);
  593. // 错误按摩,关闭客户端套接字
  594. wiz_closesocket(clientSocket);
  595. RyanW5500CheckCode(NULL, EPROTO, rlog_d, { return -1; }); // 发生协议错误,客户端套接字不处于连接状态,极端情况才会出现
  596. }
  597. // ?连接成功事件,service触发recv事件
  598. // RyanW5500DoEventChanges(socket, WIZ_EVENT_RECV, RT_TRUE);
  599. // 获取新的客户端套接字信息
  600. struct sockaddr_in *sin = (struct sockaddr_in *)addr;
  601. uint8_t ipStrArr[4] = {0};
  602. uint16_t remotePort = getSn_DPORT(clientSocket);
  603. getSn_DIPR(clientSocket, ipStrArr); // 获取远程 IP 地址和端口
  604. memset(sin, 0, sizeof(struct sockaddr_in));
  605. sin->sin_port = htons(remotePort);
  606. sin->sin_addr.s_addr = ipStrArrToinAddr(ipStrArr);
  607. *addrlen = sizeof(struct sockaddr);
  608. rlog_d("accept remote ip: %s, remote port: %d", inet_ntoa(sin->sin_addr.s_addr), remotePort);
  609. // clean server socket receive event and status
  610. wiz_do_event_clean(serviceSock, WIZ_EVENT_RECV);
  611. return clientSocket;
  612. }
  613. }
  614. /**
  615. * @brief
  616. *
  617. * @param socket 套接字描述符
  618. * @param data 发送的消息的缓冲区
  619. * @param size 以字节为单位消息的大小
  620. * @param flags 消息传输的类型,一般为 0
  621. * @param to 指向包含目标地址的sockaddr结构
  622. * @param tolen 指向的sockaddr结构的长度
  623. * @return int 返回发送的字节数
  624. */
  625. int wiz_sendto(int socket, const void *data, size_t size, int flags, const struct sockaddr *to, socklen_t tolen)
  626. {
  627. RyanW5500LinkCheck(-1);
  628. RyanW5500CheckCode(NULL != data && 0 != size, EFAULT, rlog_d, { return -1; });
  629. RyanW5500Socket *sock = NULL;
  630. uint8_t socketState = 0;
  631. int32_t sendLen = 0;
  632. sock = RyanW5500GetSock(socket);
  633. RyanW5500CheckCode(NULL != sock, EBADF, rlog_d, { return -1; });
  634. RyanW5500CheckCode(WIZ_SOCKET_MAGIC == sock->magic, EIO, rlog_d, { return -1; });
  635. // tcp 时由 getSn_SR 校验连接状态
  636. // RyanW5500CheckCode(RyanW5500SocketEstablished == sock->state, EIO, rlog_d, { return -1; });
  637. switch (sock->type)
  638. {
  639. case Sn_MR_TCP:
  640. socketState = getSn_SR(socket);
  641. RyanW5500CheckCode(SOCK_ESTABLISHED == socketState, EDESTADDRREQ, rlog_d, { return -1; }); // 套接字不是连接模式,没有设置其对等地址,也没有指定目标地址。
  642. // 如果发送数据比剩余缓冲区大,则分片发送,否则缓冲区数据会被清空
  643. sendLen = getSn_TX_FSR(sock->socket);
  644. RyanW5500CheckCode(sendLen > 0, EWOULDBLOCK, rlog_d, { return -1; }); // 发送缓冲区已满,直接返回
  645. if (sendLen > size)
  646. sendLen = size;
  647. sendLen = wizchip_send(socket, (uint8_t *)data, sendLen);
  648. if (sendLen < 0)
  649. {
  650. rlog_e("tcp send fail, result: %d", sendLen);
  651. RyanW5500DoEventChanges(sock, WIZ_EVENT_ERROR, RT_TRUE);
  652. RyanW5500CheckCode(SOCKERR_SOCKCLOSED == sendLen, EPIPE, rlog_d, { return 0; }); // 套接字被关闭
  653. return -1;
  654. }
  655. return sendLen;
  656. case Sn_MR_UDP:
  657. case Sn_MR_IPRAW:
  658. {
  659. socketState = getSn_SR(socket);
  660. RyanW5500CheckCode(SOCK_UDP == socketState || SOCK_IPRAW == socketState, EDESTADDRREQ, rlog_d, { return -1; });
  661. struct sockaddr_in *sin = NULL;
  662. // 获取目标地址结构体指针
  663. if (RyanW5500SocketEstablished == sock->state && NULL != sock->remoteAddr)
  664. sin = (struct sockaddr_in *)sock->remoteAddr;
  665. else
  666. sin = (struct sockaddr_in *)to;
  667. // 查看是否是广播
  668. if (sin->sin_addr.s_addr == -1) // inet_addr("255.255.255.255")
  669. RyanW5500CheckCode(0 != (sock->soOptionsFlag & SO_BROADCAST), EACCES, rlog_d, { return -1; });
  670. // 将发送ip转换为W5500识别的
  671. uint8_t ipStrArr[4] = {0};
  672. inAddrToipStrArr(&sin->sin_addr.s_addr, ipStrArr);
  673. // 如果发送数据比剩余缓冲区大,则分片发送,否则缓冲区数据会被清空
  674. sendLen = getSn_TX_FSR(sock->socket);
  675. RyanW5500CheckCode(sendLen > 0, EWOULDBLOCK, rlog_d, { return -1; }); // 发送缓冲区已满
  676. if (sendLen > size)
  677. sendLen = size;
  678. sendLen = wizchip_sendto(sock->socket, (uint8_t *)data, sendLen, ipStrArr, htons(sin->sin_port));
  679. if (sendLen < 0)
  680. {
  681. RyanW5500DoEventChanges(sock, WIZ_EVENT_ERROR, RT_TRUE);
  682. rlog_e("udp send fail, result: %d", sendLen);
  683. RyanW5500CheckCode(SOCKERR_SOCKCLOSED == sendLen, EPIPE, rlog_d, { return 0; }); // 套接字被关闭
  684. return -1;
  685. }
  686. return sendLen;
  687. }
  688. default:
  689. rlog_e("socket (%d) type %d is not support.", socket, sock->type);
  690. return -1;
  691. }
  692. }
  693. /**
  694. * @brief
  695. *
  696. * @param socket 套接字描述符
  697. * @param mem 接收消息的缓冲区
  698. * @param len 接收消息的缓冲区长度
  699. * @param flags 消息接收的类型,一般为 0
  700. * @param from 接收地址结构体指针,可以为NULL
  701. * @param fromlen 接收地址结构体长度
  702. * @return int 返回接收的数据的长度
  703. */
  704. int wiz_recvfrom(int socket, void *mem, size_t len, int flags, struct sockaddr *from, socklen_t *fromlen)
  705. {
  706. RyanW5500LinkCheck(-1);
  707. RyanW5500CheckCode(NULL != mem && 0 != len, EFAULT, rlog_d, { return -1; });
  708. RyanW5500Socket *sock = NULL;
  709. uint8_t socketState = 0;
  710. int32_t recvLen = 0,
  711. timeout = 0,
  712. result = 0;
  713. platformTimer_t recvTimer = {0};
  714. sock = RyanW5500GetSock(socket);
  715. RyanW5500CheckCode(NULL != sock, EBADF, rlog_d, { return -1; });
  716. RyanW5500CheckCode(WIZ_SOCKET_MAGIC == sock->magic, EIO, rlog_d, { return -1; });
  717. // 设置 WIZNnet 套接字接收超时
  718. timeout = sock->recvTimeout;
  719. if (timeout <= 0)
  720. timeout = RT_WAITING_FOREVER;
  721. platformTimerCutdown(&recvTimer, timeout);
  722. again:
  723. // 判断是否超时
  724. timeout = platformTimerRemain(&recvTimer);
  725. RyanW5500CheckCode(0 != timeout, EAGAIN, rlog_d, { result = -1; goto __exit; });
  726. switch (sock->type)
  727. {
  728. case Sn_MR_TCP:
  729. {
  730. socketState = getSn_SR(socket);
  731. RyanW5500CheckCode(SOCK_ESTABLISHED == socketState, ENOTCONN, rlog_d, { result = -1; goto __exit; }); // 在未连接的连接模式套接字上尝试接收。
  732. // 先读取之前存在的数据
  733. recvLen = getSn_RX_RSR(socket);
  734. if (recvLen > 0)
  735. {
  736. rt_mutex_take(sock->recvLock, RT_WAITING_FOREVER);
  737. recvLen = wizchip_recv(socket, mem, len);
  738. rt_mutex_release(sock->recvLock);
  739. if (recvLen > 0)
  740. goto __exit;
  741. }
  742. result = rt_event_recv(RyanW5500Entry.W5500EventHandle, (1 << sock->socket),
  743. RT_EVENT_FLAG_AND | RT_EVENT_FLAG_CLEAR,
  744. timeout, NULL);
  745. RyanW5500CheckCode(RT_EOK == result, EAGAIN, rlog_d, { result = -1; goto __exit; }); // 判断是否超时
  746. RyanW5500CheckCode(RyanW5500SocketClose != sock->state, ECONNRESET, rlog_d, { result = 0; goto __exit; }); // 连接被对等方关闭。
  747. // 获取数据寄存器,没有数据就重新接收
  748. if (getSn_RX_RSR(socket) <= 0)
  749. goto again;
  750. rt_mutex_take(sock->recvLock, RT_WAITING_FOREVER);
  751. recvLen = wizchip_recv(socket, mem, len);
  752. rt_mutex_release(sock->recvLock);
  753. if (recvLen < 0)
  754. {
  755. rlog_e("recv error, result: %d", recvLen);
  756. RyanW5500CheckCode(SOCKERR_SOCKCLOSED == recvLen, ECONNRESET, rlog_d, { result = 0; goto __exit; }); // 套接字被关闭
  757. result = -1;
  758. goto __exit;
  759. }
  760. }
  761. break;
  762. case Sn_MR_UDP:
  763. case Sn_MR_IPRAW:
  764. {
  765. socketState = getSn_SR(socket);
  766. RyanW5500CheckCode(SOCK_UDP == socketState || SOCK_IPRAW == socketState, ENOTCONN, rlog_d, { result = -1; goto __exit; }); // 在未连接的连接模式套接字上尝试接收。
  767. result = rt_event_recv(RyanW5500Entry.W5500EventHandle, (1 << sock->socket),
  768. RT_EVENT_FLAG_AND | RT_EVENT_FLAG_CLEAR,
  769. timeout, NULL);
  770. RyanW5500CheckCode(RT_EOK == result, EAGAIN, rlog_d, { result = -1; goto __exit; }); // 判断是否超时
  771. RyanW5500CheckCode(RyanW5500SocketClose != sock->state, ECONNRESET, rlog_d, { result = 0; goto __exit; });
  772. // 获取数据寄存器,没有数据就重新接收
  773. recvLen = getSn_RX_RSR(socket);
  774. if (recvLen <= 0)
  775. goto again;
  776. uint16_t remotePort = 0;
  777. uint8_t remoteIp[4] = {0};
  778. // 如果数据比缓冲区大,则调整至缓冲区大小
  779. if (recvLen > len)
  780. recvLen = len;
  781. // rt_mutex_take(sock->recvLock, RT_WAITING_FOREVER);
  782. recvLen = wizchip_recvfrom(socket, mem, recvLen, remoteIp, &remotePort);
  783. // rt_mutex_release(sock->recvLock);
  784. if (recvLen < 0)
  785. {
  786. RyanW5500CheckCode(SOCKERR_SOCKCLOSED == recvLen, ECONNRESET, rlog_d, { result = 0; goto __exit; }); // 套接字被关闭
  787. rlog_e("recvfrom error, result: %d", recvLen);
  788. result = -1;
  789. goto __exit;
  790. }
  791. // 将消息信息写入from结构体
  792. *fromlen = sizeof(struct sockaddr_in);
  793. struct sockaddr_in *sin = (struct sockaddr_in *)from;
  794. memset(sin, 0, sizeof(struct sockaddr_in));
  795. sin->sin_port = htons(remotePort);
  796. sin->sin_addr.s_addr = ipStrArrToinAddr(remoteIp);
  797. }
  798. break;
  799. default:
  800. rlog_e("socket (%d) type %d is not support.", socket, sock->type);
  801. return -1;
  802. }
  803. __exit:
  804. if (recvLen > 0)
  805. {
  806. // 每接收一次就清除一次recv计数,当缓冲区为0时清空recv
  807. errno = 0;
  808. RyanW5500DoEventChanges(sock, WIZ_EVENT_RECV, RT_FALSE);
  809. result = recvLen;
  810. if (getSn_RX_RSR(socket) == 0)
  811. wiz_do_event_clean(sock, WIZ_EVENT_RECV);
  812. }
  813. else
  814. {
  815. RyanW5500DoEventChanges(sock, WIZ_EVENT_ERROR, RT_TRUE);
  816. }
  817. return result;
  818. }
  819. int wiz_send(int socket, const void *data, size_t size, int flags)
  820. {
  821. return wiz_sendto(socket, data, size, flags, NULL, 0);
  822. }
  823. /**
  824. * @brief TCP 数据接收
  825. *
  826. * @param socket 套接字描述符
  827. * @param mem 接收消息的缓冲区
  828. * @param len 接收消息的缓冲区长度
  829. * @param flags 消息接收的类型,一般为 0
  830. * @return int
  831. */
  832. int wiz_recv(int socket, void *mem, size_t len, int flags)
  833. {
  834. return wiz_recvfrom(socket, mem, len, flags, NULL, NULL);
  835. }
  836. /**
  837. * @brief 关闭套接字
  838. *
  839. * @param socket 套接字描述符
  840. * @return int
  841. */
  842. int wiz_closesocket(int socket)
  843. {
  844. RyanW5500Socket *sock = NULL;
  845. sock = RyanW5500GetSock(socket);
  846. RyanW5500CheckCode(NULL != sock, EBADF, rlog_d, { return -1; });
  847. if (SOCK_CLOSED == getSn_SR(sock->socket))
  848. {
  849. RyanW5500SocketDestory(sock);
  850. return -1;
  851. }
  852. int8_t result = SOCK_FATAL;
  853. // 属于tcp套接字,但又不是listen服务器套接字
  854. if (Sn_MR_TCP == sock->type && RyanW5500SocketEstablished == sock->state && NULL == sock->serviceInfo)
  855. result = wizchip_disconnect(sock->socket);
  856. if (SOCK_OK != result)
  857. {
  858. result = wizchip_close(sock->socket);
  859. if (SOCK_OK != result)
  860. {
  861. rlog_e("socket(%d) close failed.", sock->socket);
  862. RyanW5500SocketDestory(sock);
  863. return -1;
  864. }
  865. }
  866. return RyanW5500SocketDestory(sock);
  867. }
  868. /**
  869. * @brief 按设置关闭套接字
  870. *
  871. * @param socket 套接字描述符
  872. * @param how 套接字控制的方式
  873. * @return int
  874. */
  875. int wiz_shutdown(int socket, int how)
  876. {
  877. RyanW5500Socket *sock = NULL;
  878. switch (how)
  879. {
  880. case SHUT_RD: // 禁用进一步的接收操作。
  881. case SHUT_WR: // 禁用进一步的发送操作。
  882. case SHUT_RDWR: // 禁用进一步的发送和接收操作。
  883. break;
  884. default:
  885. RyanW5500CheckCode(NULL, EINVAL, rlog_d, return -1;)
  886. }
  887. sock = RyanW5500GetSock(socket);
  888. RyanW5500CheckCode(NULL != sock, EBADF, rlog_d, { return -1; });
  889. return wiz_closesocket(sock->socket);
  890. }
  891. /**
  892. * @brief 设置套接字选项
  893. *
  894. * @param socket 套接字描述符
  895. * @param level 协议栈配置选项
  896. * @param optname 需要设置的选项名
  897. * @param optval 设置选项值的缓冲区地址
  898. * @param optlen 设置选项值的缓冲区长度
  899. * @return int
  900. */
  901. int wiz_setsockopt(int socket, int level, int optname, const void *optval, socklen_t optlen)
  902. {
  903. RyanW5500LinkCheck(-1);
  904. RyanW5500CheckCode(NULL != optval && 0 != optlen, EFAULT, rlog_d, { return -1; });
  905. RyanW5500Socket *sock = NULL;
  906. sock = RyanW5500GetSock(socket);
  907. RyanW5500CheckCode(NULL != sock, EBADF, rlog_d, { return -1; }); // 指定的选项在指定的套接字级别无效或套接字已关闭。
  908. if (SOL_SOCKET == level)
  909. {
  910. switch (optname)
  911. {
  912. case SO_BROADCAST: // 配置用于发送广播数据的套接字。此选项仅适用于支持广播 IP 和 UDP 的协议 布尔类型
  913. RyanW5500CheckCode(Sn_MR_TCP != sock->type, EINVAL, rlog_d, { return -1; }); // 指定的选项在指定的套接字级别无效或套接字已关闭。
  914. RyanW5500CheckCode(sizeof(int) == optlen, EFAULT, rlog_d, { return -1; });
  915. if (1 == *(int *)optval)
  916. sock->soOptionsFlag |= optname;
  917. else
  918. sock->soOptionsFlag &= ~optname;
  919. break;
  920. case SO_KEEPALIVE: // 为套接字连接启用保持连接。 仅适用于支持保持连接的协议,比如tcp, 对于 TCP,默认保持连接超时为 2 小时,保持连接间隔为 1 秒 布尔类型
  921. {
  922. if (sock->type != Sn_MR_TCP) // 对于w5500非tcp连接忽略
  923. break;
  924. uint32_t keepalive = 2 * 60 * 60 / 5; // w5500 Sn_KPALVTR单位时间为5
  925. if (1 == *(int *)optval)
  926. RyanW5500CheckCode(SOCK_OK == wizchip_setsockopt(sock->socket, SO_KEEPALIVEAUTO, (void *)&keepalive), EINVAL, rlog_d, { return -1; });
  927. break;
  928. }
  929. case SO_RCVTIMEO: // 阻止接收调用的超时(以毫秒为单位)。 此选项的默认值为零,表示接收操作不会超时 类型struct timeval
  930. RyanW5500CheckCode(sizeof(struct timeval) == optlen, EFAULT, rlog_d, { return -1; });
  931. sock->recvTimeout = ((const struct timeval *)optval)->tv_sec * 1000 +
  932. ((const struct timeval *)optval)->tv_usec / 1000;
  933. break;
  934. case SO_SNDTIMEO: // 阻止发送调用的超时(以毫秒为单位)。 此选项的默认值为零,表示发送操作不会超时。类型struct timeval
  935. RyanW5500CheckCode(sizeof(struct timeval) == optlen, EFAULT, rlog_d, { return -1; });
  936. sock->sendTimeout = ((const struct timeval *)optval)->tv_sec * 1000 +
  937. ((const struct timeval *)optval)->tv_usec / 1000;
  938. break;
  939. // 以下参数没有实现的原因是因为W5500 本身发送 / 接收缓冲区各16K是8个socket共享的
  940. // 用户可能不知道他修改的缓冲区是否是正确,如果超过16k则有可能造成个别socket通道发送 / 接收失败
  941. case SO_SNDBUF: // 设置发送缓冲区大小。此选项采用 int 值。
  942. case SO_RCVBUF: // 设置接收缓冲区大小。此选项采用 int 值。
  943. case SO_RCVLOWAT: // 此选项设置套接字输入操作要处理的最小字节数
  944. case SO_SNDLOWAT: // 此选项设置套接字输出操作要处理的最小字节数
  945. case SO_DEBUG: // 启用调试输出
  946. case SO_REUSEADDR: // 允许套接字绑定到已使用的地址和端口。布尔类型
  947. case SO_LINGER: // 如果存在数据,则停留在 close()上等待缓存数据发送完毕。 布尔类型
  948. case SO_OOBINLINE: // 指示应随常规数据一起返回超出边界的数据。 此选项仅适用于支持带外数据的面向连接的协议。 布尔类型
  949. case SO_DONTROUTE: // 指示应在套接字绑定到的任何接口上发送传出数据,而不是在其他接口上路由。 布尔类型
  950. return -1;
  951. default:
  952. RyanW5500CheckCode(NULL, ENOPROTOOPT, rlog_d, { return -1; });
  953. }
  954. }
  955. else if (IPPROTO_IP == level)
  956. {
  957. switch (optname)
  958. {
  959. case IP_TOS: // 本选项是int型的数值选项,允许对TCP、UDP的IP头中的tos字段进行设置
  960. RyanW5500CheckCode(SOCK_OK == wizchip_setsockopt(sock->socket, SO_TOS, (void *)optval), EINVAL, rlog_d, { return -1; });
  961. break;
  962. case IP_TTL: // 本选项是int型的数值选项,允许对单播报文的TTL默认值进行设置。
  963. RyanW5500CheckCode(SOCK_OK == wizchip_setsockopt(sock->socket, SO_TTL, (void *)optval), EINVAL, rlog_d, { return -1; });
  964. break;
  965. // UDP 多播的选项和类型
  966. case IP_ADD_MEMBERSHIP: // 加入一个组播组 struct ip_mreq
  967. {
  968. RyanW5500CheckCode(Sn_MR_TCP != sock->type, EINVAL, rlog_d, { return -1; }); // 指定的选项在指定的套接字级别无效或套接字已关闭。
  969. RyanW5500CheckCode(sizeof(struct ip_mreq) == optlen, EFAULT, rlog_d, { return -1; });
  970. struct ip_mreq *mreq = (struct ip_mreq *)optval;
  971. // 需要在Sn_CR命令之前,分开配置组播 IP 地址及端口号。
  972. wizchip_close(sock->socket);
  973. // 组播MAC地址的高24bit为0x01005e,高位第25bit为0,即高25bit为固定值。
  974. // MAC地址的低23bit为组播IP地址的低23bit
  975. uint8_t ipStrArr[4] = {0};
  976. inAddrToipStrArr(&mreq->imr_multiaddr.s_addr, ipStrArr);
  977. uint8_t DHAR[6] = {0x01, 0x00, 0x5e};
  978. memcpy(DHAR + 3, ipStrArr + 1, 3);
  979. DHAR[2] &= ~(1 << 7); // 高位第25bit为0
  980. setSn_DHAR(sock->socket, DHAR);
  981. setSn_DIPR(sock->socket, ipStrArr);
  982. setSn_DPORT(sock->socket, sock->port);
  983. int8_t result = wizchip_socket(sock->socket, sock->type, sock->port, Sn_MR_MULTI);
  984. RyanW5500CheckCode(result == sock->socket, EMFILE, rlog_d, { return -1; });
  985. break;
  986. }
  987. case IP_DROP_MEMBERSHIP: // 退出组播组 struct ip_mreq
  988. {
  989. int8_t result = wizchip_socket(sock->socket, sock->type, sock->port, 0);
  990. RyanW5500CheckCode(result == sock->socket, EMFILE, rlog_d, { return -1; });
  991. break;
  992. }
  993. case IP_MULTICAST_TTL: // 设置多播组数据的TTL值, 范围为0~255之间的任何值
  994. case IP_MULTICAST_IF: // 设置用于发送 IPv4 多播流量的传出接口。 struct in_addr
  995. case IP_MULTICAST_LOOP: // 对于已加入一个或多个多播组的套接字,此套接字控制它是否将收到通过所选多播接口发送到这些多播组的 传出 数据包的副本。布尔类型
  996. return -1;
  997. // case IP_RECVDSTADDR: // 本选项是标志位,置上之后,允许对UDP socket调用recvfrom的时候,能够以辅助数据的形式获取到客户端报文的目的地址。
  998. // case IP_RECVIF: // 本选项是标志位,置上之后,允许对UDP socket调用recvfrom的时候,能够以辅助数据的形式获取到客户端报文的目的接口。
  999. default:
  1000. RyanW5500CheckCode(NULL, ENOPROTOOPT, rlog_d, { return -1; });
  1001. }
  1002. }
  1003. else if (IPPROTO_TCP == level)
  1004. {
  1005. switch (optname)
  1006. {
  1007. case TCP_NODELAY: // 启用或禁用 TCP 套接字的 Nagle 算法。
  1008. case TCP_KEEPIDLE: // 获取或设置 TCP 连接在发送到远程之前保持空闲状态的秒数。
  1009. case TCP_KEEPINTVL: // 获取或设置 TCP 连接在发送另一个保留探测之前等待保持响应的秒数
  1010. case TCP_KEEPCNT: // 获取或设置将在连接终止之前发送的 TCP 保持活动探测数。 将TCP_KEEPCNT设置为大于 255 的值是非法的。
  1011. return -1;
  1012. default:
  1013. RyanW5500CheckCode(NULL, ENOPROTOOPT, rlog_d, { return -1; });
  1014. }
  1015. }
  1016. else
  1017. RyanW5500CheckCode(NULL, EINVAL, rlog_d, { return -1; });
  1018. return 0;
  1019. }
  1020. /**
  1021. * @brief 获取套接字选项
  1022. *
  1023. * @param socket 套接字描述符
  1024. * @param level 协议栈配置选项
  1025. * @param optname 需要设置的选项名
  1026. * @param optval 获取选项值的缓冲区地址
  1027. * @param optlen 获取选项值的缓冲区长度地址
  1028. * 如果选项值的大小大于option_len ,存储在option_value参数指向的对象中的值将被静默截断。
  1029. * 否则, option_len参数指向的对象 应被修改以指示值的实际长度。
  1030. * @return int
  1031. */
  1032. int wiz_getsockopt(int socket, int level, int optname, void *optval, socklen_t *optlen)
  1033. {
  1034. RyanW5500LinkCheck(-1);
  1035. RyanW5500CheckCode(NULL != optval && NULL != optlen, EFAULT, rlog_d, { return -1; });
  1036. RyanW5500Socket *sock = NULL;
  1037. sock = RyanW5500GetSock(socket);
  1038. RyanW5500CheckCode(NULL != sock, EBADF, rlog_d, { return -1; });
  1039. if (SOL_SOCKET == level)
  1040. {
  1041. switch (optname)
  1042. {
  1043. case SO_BROADCAST: // 报告是否支持广播消息的传输 布尔类型
  1044. RyanW5500CheckCode(sizeof(int) <= *optlen, EINVAL, rlog_d, { return -1; });
  1045. *(int *)optval = Sn_MR_TCP != sock->type ? 1 : 0;
  1046. *optlen = sizeof(int);
  1047. break;
  1048. // 暂不实现
  1049. case SO_KEEPALIVE: // 报告连接是否通过定期传输消息保持活动状态 布尔类型
  1050. return -1;
  1051. case SO_RCVTIMEO: // 报告阻止接收调用的超时。 类型struct timeval
  1052. RyanW5500CheckCode(sizeof(struct timeval) <= *optlen, EINVAL, rlog_d, { return -1; });
  1053. ((struct timeval *)(optval))->tv_sec = sock->recvTimeout / 1000U;
  1054. ((struct timeval *)(optval))->tv_usec = (sock->recvTimeout % 1000U) * 1000U;
  1055. *optlen = sizeof(struct timeval);
  1056. break;
  1057. case SO_SNDTIMEO: // 报告阻止发送调用的超时。 此选项的默认值为零,表示发送操作不会超时。类型struct timeval
  1058. RyanW5500CheckCode(sizeof(struct timeval) <= *optlen, EINVAL, rlog_d, { return -1; });
  1059. ((struct timeval *)optval)->tv_sec = sock->sendTimeout / 1000U;
  1060. ((struct timeval *)optval)->tv_usec = (sock->sendTimeout % 1000U) * 1000U;
  1061. *optlen = sizeof(struct timeval);
  1062. break;
  1063. case SO_SNDBUF: // 报告发送缓冲区大小。此选项采用 int 值。
  1064. RyanW5500CheckCode(sizeof(int) <= *optlen, EINVAL, rlog_d, { return -1; });
  1065. wizchip_getsockopt(socket, (sockopt_type)SO_SENDBUF, (void *)optval);
  1066. *optlen = sizeof(int);
  1067. break;
  1068. case SO_RCVBUF: // 报告接收缓冲区大小。此选项采用 int 值。
  1069. RyanW5500CheckCode(sizeof(int) <= *optlen, EINVAL, rlog_d, { return -1; });
  1070. wizchip_getsockopt(socket, (sockopt_type)SO_RECVBUF, (void *)optval);
  1071. *optlen = sizeof(int);
  1072. break;
  1073. case SO_ACCEPTCONN: // 报告是否启用套接字侦听 布尔类型
  1074. RyanW5500CheckCode(sizeof(int) <= *optlen, EINVAL, rlog_d, { return -1; });
  1075. if (RyanW5500SocketListen == sock->state && NULL != sock->serviceInfo)
  1076. *(int *)optval = 1;
  1077. else
  1078. *(int *)optval = 0;
  1079. *optlen = sizeof(int);
  1080. break;
  1081. case SO_ERROR: // 报告有关错误状态的信息并将其清除。 int
  1082. RyanW5500CheckCode(sizeof(int) <= *optlen, EINVAL, rlog_d, { return -1; });
  1083. *(int *)optval = errno;
  1084. *optlen = sizeof(int);
  1085. errno = 0;
  1086. break;
  1087. case SO_TYPE: // 报告套接字类型 int
  1088. RyanW5500CheckCode(sizeof(int) <= *optlen, EINVAL, rlog_d, { return -1; });
  1089. *(int *)optval = sock->type;
  1090. *optlen = sizeof(int);
  1091. break;
  1092. case SO_DEBUG: // 报告是否正在记录调试信息 布尔类型
  1093. case SO_REUSEADDR: // 报告是否允许套接字绑定到已使用的地址和端口。布尔类型
  1094. case SO_LINGER: // 报告套接字是否停留在 close()上
  1095. case SO_OOBINLINE: // 报告套接字是否保留接收到的带外数据 此选项仅适用于支持带外数据的面向连接的协议。 布尔类型
  1096. case SO_DONTROUTE: // 报告是否允许指示应在套接字绑定到的任何接口上发送传出数据,而不是在其他接口上路由。 布尔类型
  1097. case SO_RCVLOWAT: // 报告套接字输入操作要处理的最小字节数 int
  1098. case SO_SNDLOWAT: // 报告套接字输出操作要处理的最小字节数 int
  1099. return -1;
  1100. default:
  1101. RyanW5500CheckCode(NULL, ENOPROTOOPT, rlog_d, { return -1; });
  1102. }
  1103. }
  1104. else if (IPPROTO_IP == level)
  1105. {
  1106. switch (optname)
  1107. {
  1108. case IP_TOS: // 本选项是int型的数值选项,允许对TCP、UDP的IP头中的tos字段进行获取
  1109. RyanW5500CheckCode(sizeof(int) <= *optlen, EINVAL, rlog_d, { return -1; });
  1110. wizchip_getsockopt(sock->socket, (sockopt_type)SO_TOS, optval);
  1111. *optlen = sizeof(int);
  1112. break;
  1113. case IP_TTL: // 本选项是int型的数值选项,允许对单播报文的TTL默认值进行获取
  1114. RyanW5500CheckCode(sizeof(int) <= *optlen, EINVAL, rlog_d, { return -1; });
  1115. wizchip_getsockopt(sock->socket, (sockopt_type)SO_TTL, optval);
  1116. *optlen = sizeof(int);
  1117. break;
  1118. // UDP 多播的选项和类型
  1119. case IP_MULTICAST_TTL: // 获取多播组数据的TTL值, 范围为0~255之间的任何值
  1120. case IP_MULTICAST_IF: // 获取用于发送 IPv4 多播流量的传出接口。 struct in_addr
  1121. return -1;
  1122. // case IP_RECVDSTADDR: // 本选项是标志位,置上之后,允许对UDP socket调用recvfrom的时候,能够以辅助数据的形式获取到客户端报文的目的地址。
  1123. // case IP_RECVIF: // 本选项是标志位,置上之后,允许对UDP socket调用recvfrom的时候,能够以辅助数据的形式获取到客户端报文的目的接口。
  1124. // return -1;
  1125. default:
  1126. RyanW5500CheckCode(NULL, ENOPROTOOPT, rlog_d, { return -1; });
  1127. }
  1128. }
  1129. else if (IPPROTO_TCP == level)
  1130. {
  1131. switch (optname)
  1132. {
  1133. case TCP_NODELAY: // 获取或设置 TCP 套接字的 Nagle 算法。
  1134. case TCP_KEEPIDLE: // 获取或设置 TCP 连接在发送到远程之前保持空闲状态的秒数。
  1135. case TCP_KEEPINTVL: // 获取或设置 TCP 连接在发送另一个保留探测之前等待保持响应的秒数
  1136. case TCP_KEEPCNT: // 获取或设置将在连接终止之前发送的 TCP 保持活动探测数。 将TCP_KEEPCNT设置为大于 255 的值是非法的。
  1137. return -1;
  1138. default:
  1139. RyanW5500CheckCode(NULL, ENOPROTOOPT, rlog_d, { return -1; });
  1140. }
  1141. }
  1142. else
  1143. RyanW5500CheckCode(NULL, EINVAL, rlog_d, { return -1; });
  1144. return 0;
  1145. return 0;
  1146. }
  1147. static int RyanW5500_gethostbyname(const char *name, ip_addr_t *addr)
  1148. {
  1149. int idx = 0;
  1150. int nameLen = strlen(name);
  1151. char ipStrArr[16] = {0};
  1152. // 检查域名 / ip地址
  1153. for (idx = 0; idx < nameLen && !_isalpha(name[idx]); idx++)
  1154. ;
  1155. // 输入名称为ip地址
  1156. if (idx == strlen(name))
  1157. strncpy(ipStrArr, name, nameLen);
  1158. // 输入名称为域名需要调用dns
  1159. else
  1160. {
  1161. int8_t ret = 0;
  1162. uint8_t remote_ip[4] = {0};
  1163. uint8_t data_buffer[512];
  1164. // rlog_w("开始获取dns");
  1165. // DNS客户端处理
  1166. rt_mutex_take(RyanW5500Entry.dnsMutexHandle, RT_WAITING_FOREVER); // 获取互斥锁
  1167. ret = DNS_run(gWIZNETINFO.dns, (uint8_t *)name, remote_ip, data_buffer);
  1168. rt_mutex_release(RyanW5500Entry.dnsMutexHandle); // 释放互斥锁
  1169. if (1 != ret)
  1170. {
  1171. if (-1 == ret)
  1172. {
  1173. rlog_e("MAX_DOMAIN_NAME is too small, should be redefined it.");
  1174. return -1;
  1175. }
  1176. if (-2 == ret)
  1177. {
  1178. rlog_e("DNS failed, socket number is full.");
  1179. return -2;
  1180. }
  1181. return -1;
  1182. }
  1183. // 域解析失败
  1184. if (remote_ip[0] == 0)
  1185. return -1;
  1186. snprintf(ipStrArr, 16, "%u.%u.%u.%u", remote_ip[0], remote_ip[1], remote_ip[2], remote_ip[3]);
  1187. }
  1188. inet_aton(ipStrArr, addr);
  1189. return 0;
  1190. }
  1191. /**
  1192. * @brief 根据主机名解析地址信息 DNS
  1193. *
  1194. * @param name 主机名,dns规定长度限制为255
  1195. * @return struct hostent*
  1196. */
  1197. struct hostent *wiz_gethostbyname(const char *name)
  1198. {
  1199. // 检查WIZnet初始化状态
  1200. RyanW5500LinkCheck(NULL);
  1201. ip_addr_t addr = {0};
  1202. int err = 0;
  1203. // 用于 gethostbyname() 的缓冲区变量
  1204. static struct hostent s_hostent;
  1205. static char s_hostname[DNS_MAX_NAME_LENGTH + 1]; // 主机名称
  1206. static char *s_aliases; // 主机别名
  1207. static ip_addr_t s_hostent_addr; // 临时主机IP
  1208. static ip_addr_t *s_phostent_addr[2]; // 主机IP
  1209. if (NULL == name)
  1210. {
  1211. rlog_e("gethostbyname input name err!");
  1212. return NULL;
  1213. }
  1214. if (strlen(name) > DNS_MAX_NAME_LENGTH)
  1215. return NULL;
  1216. err = RyanW5500_gethostbyname(name, &addr);
  1217. if (0 != err)
  1218. {
  1219. // 不支持h_errno
  1220. // RyanW5500CheckCode(-1 == result, HOST_NOT_FOUND, return NULL);
  1221. // RyanW5500CheckCode(-2 == result, TRY_AGAIN, return NULL);
  1222. return NULL;
  1223. }
  1224. // 填充主机结构
  1225. s_hostent_addr = addr;
  1226. s_phostent_addr[0] = &s_hostent_addr;
  1227. s_phostent_addr[1] = NULL;
  1228. strncpy(s_hostname, name, strlen(name));
  1229. s_hostname[strlen(name)] = 0;
  1230. s_aliases = NULL;
  1231. s_hostent.h_name = s_hostname;
  1232. s_hostent.h_aliases = &s_aliases;
  1233. s_hostent.h_addrtype = AF_INET;
  1234. s_hostent.h_length = sizeof(ip_addr_t);
  1235. s_hostent.h_addr_list = (char **)&s_phostent_addr;
  1236. return &s_hostent;
  1237. }
  1238. /**
  1239. * @brief wiz_gethostbyname 线程安全版本
  1240. *
  1241. * @param name 主机名,dns规定长度限制为255
  1242. * @param ret 预先分配的结构体,用于存储结果
  1243. * @param buf 预先分配的缓冲区,用于存储额外的数据
  1244. * @param buflen 缓冲区大小
  1245. * @param result 指向主机指针的指针,成功时设置为 ret,错误时设置为零
  1246. * @param h_errnop 指向存储错误的 int 的指针(而不是修改全局 h_errno)
  1247. * @return int 0成功,非0错误
  1248. */
  1249. int wiz_gethostbyname_r(const char *name, struct hostent *ret, char *buf, size_t buflen, struct hostent **result, int *h_errnop)
  1250. {
  1251. // 检查WIZnet初始化状态
  1252. RyanW5500LinkCheck(-1);
  1253. // gethostbyname_r 的辅助结构,用于访问hostent中的 char*缓冲区
  1254. struct gethostbyname_r_helper
  1255. {
  1256. ip_addr_t *addr_list[2];
  1257. ip_addr_t addr;
  1258. char *aliases;
  1259. };
  1260. char *hostname = NULL;
  1261. int lh_errno = 0,
  1262. err = 0;
  1263. size_t namelen = 0;
  1264. struct gethostbyname_r_helper *h = NULL;
  1265. if (h_errnop == NULL)
  1266. h_errnop = &lh_errno; // 确保 h_errnop 永远不会为 NULL
  1267. if ((name == NULL) || (ret == NULL) || (buf == NULL))
  1268. {
  1269. *h_errnop = EINVAL; // 参数不足
  1270. return -1;
  1271. }
  1272. if (result == NULL)
  1273. {
  1274. *h_errnop = EINVAL; // 参数不足
  1275. return -1;
  1276. }
  1277. // 第一件事:将 *result 设置为空
  1278. *result = NULL;
  1279. namelen = strlen(name);
  1280. if (buflen < (sizeof(struct gethostbyname_r_helper) + (namelen + 1))) // 计算buf是否足够存储hostent信息
  1281. {
  1282. *h_errnop = ERANGE; // buf 不能保存所需的数据 + 名称的副本
  1283. return -1;
  1284. }
  1285. h = (struct gethostbyname_r_helper *)buf; // 从buf中分配用于hostent中的 char*缓冲区
  1286. hostname = ((char *)h) + sizeof(struct gethostbyname_r_helper); // 从buf中分配用于hostent中 h_name缓冲区
  1287. err = RyanW5500_gethostbyname(name, &h->addr);
  1288. if (0 != err)
  1289. {
  1290. *h_errnop = HOST_NOT_FOUND;
  1291. return -1;
  1292. }
  1293. // 拷贝hostname信息
  1294. memcpy(hostname, name, namelen);
  1295. hostname[namelen] = 0;
  1296. // 填充主机结构
  1297. h->addr_list[0] = &h->addr;
  1298. h->addr_list[1] = NULL;
  1299. h->aliases = NULL;
  1300. ret->h_name = hostname;
  1301. ret->h_aliases = &h->aliases;
  1302. ret->h_addrtype = AF_INET;
  1303. ret->h_length = sizeof(ip_addr_t);
  1304. ret->h_addr_list = (char **)&h->addr_list;
  1305. // 设置结果
  1306. *result = ret;
  1307. return 0;
  1308. }
  1309. /**
  1310. * @brief 获取地址信息,应该是线程安全的
  1311. * !没有ipv6实现,只支持ipv4 地址
  1312. *
  1313. * @param nodename 主机名,可以是域名,也可以是地址的点分十进制字符串
  1314. * @param servname 服务名可以是十进制的端口号("8080")字符串,(也可以是已定义的服务名称,如"ftp"、"http"等,当前接口不支持)
  1315. * @param hints 用户设定的 struct addrinfo 结构体
  1316. * @param res 该参数获取一个指向存储结果的 struct addrinfo 结构体列表,使用完成后调用 freeaddrinfo() 释放存储结果空间。
  1317. * @return int
  1318. */
  1319. int wiz_getaddrinfo(const char *nodename, const char *servname, const struct addrinfo *hints, struct addrinfo **res)
  1320. {
  1321. // 检查WIZnet初始化状态
  1322. RyanW5500LinkCheck(-1);
  1323. int err = 0,
  1324. port_nr = 0,
  1325. ai_family = 0;
  1326. size_t total_size = 0;
  1327. size_t namelen = 0;
  1328. ip_addr_t addr = {0};
  1329. struct addrinfo *ai = NULL;
  1330. struct sockaddr_storage *sa = NULL;
  1331. if (NULL == res)
  1332. return EAI_FAIL;
  1333. *res = NULL;
  1334. // nodename 和 servname 可以设置为NULL,但同时只能有一个为NUL。
  1335. if ((NULL == nodename) && (NULL == servname))
  1336. return EAI_NONAME;
  1337. if (NULL == hints)
  1338. ai_family = AF_UNSPEC;
  1339. else
  1340. {
  1341. ai_family = hints->ai_family; // 套接字的地址族
  1342. if (ai_family != AF_WIZ &&
  1343. ai_family != AF_INET &&
  1344. ai_family != AF_UNSPEC)
  1345. return EAI_FAMILY;
  1346. }
  1347. // 指定的服务名称: 转换为端口号
  1348. if (servname != NULL)
  1349. {
  1350. port_nr = atoi(servname);
  1351. if ((port_nr <= 0) || (port_nr > 0xffff))
  1352. return EAI_SERVICE;
  1353. }
  1354. if (NULL == nodename)
  1355. {
  1356. inet_aton("127.0.0.1", &addr); // 没有指定服务位置,请使用回送地址
  1357. }
  1358. else if (nodename != NULL)
  1359. {
  1360. // 指定了服务位置,尝试解析
  1361. if ((hints != NULL) && (hints->ai_flags & AI_NUMERICHOST))
  1362. {
  1363. if (AF_INET != ai_family)
  1364. return EAI_NONAME;
  1365. // 解析地址字符串
  1366. if (!inet_aton(nodename, (ip4_addr_t *)&addr))
  1367. return EAI_NONAME;
  1368. }
  1369. else
  1370. {
  1371. err = RyanW5500_gethostbyname(nodename, &addr);
  1372. if (0 != err)
  1373. return EAI_FAIL;
  1374. }
  1375. }
  1376. total_size = sizeof(struct addrinfo) + sizeof(struct sockaddr_storage);
  1377. if (NULL != nodename)
  1378. {
  1379. namelen = strlen(nodename);
  1380. if (namelen > DNS_MAX_NAME_LENGTH)
  1381. return EAI_FAIL;
  1382. total_size += namelen + 1;
  1383. }
  1384. if (total_size > sizeof(struct addrinfo) + sizeof(struct sockaddr_storage) + DNS_MAX_NAME_LENGTH + 1)
  1385. return EAI_FAIL;
  1386. ai = (struct addrinfo *)malloc(total_size);
  1387. if (ai == NULL)
  1388. return ENOMEM;
  1389. memset(ai, 0, total_size);
  1390. // 通过void强制转换 * 以消除对齐警告
  1391. sa = (struct sockaddr_storage *)(void *)((uint8_t *)ai + sizeof(struct addrinfo));
  1392. struct sockaddr_in *sa4 = (struct sockaddr_in *)sa;
  1393. /* set up sockaddr */
  1394. sa4->sin_addr.s_addr = addr.addr;
  1395. sa4->sin_family = AF_INET;
  1396. sa4->sin_len = sizeof(struct sockaddr_in);
  1397. sa4->sin_port = htons((uint16_t)port_nr);
  1398. ai->ai_family = AF_INET;
  1399. // 设置addrinfo
  1400. if (hints != NULL)
  1401. {
  1402. // 如果指定,从指定中复制套接字类型和协议
  1403. ai->ai_socktype = hints->ai_socktype;
  1404. ai->ai_protocol = hints->ai_protocol;
  1405. }
  1406. if (nodename != NULL)
  1407. {
  1408. // 如果指定,将节点名称复制到canonname
  1409. ai->ai_canonname = ((char *)ai + sizeof(struct addrinfo) + sizeof(struct sockaddr_storage));
  1410. memcpy(ai->ai_canonname, nodename, namelen);
  1411. ai->ai_canonname[namelen] = 0;
  1412. }
  1413. ai->ai_addrlen = sizeof(struct sockaddr_storage);
  1414. ai->ai_addr = (struct sockaddr *)sa;
  1415. *res = ai;
  1416. return 0;
  1417. }
  1418. void wiz_freeaddrinfo(struct addrinfo *ai)
  1419. {
  1420. struct addrinfo *next = NULL;
  1421. while (ai != NULL)
  1422. {
  1423. next = ai->ai_next;
  1424. free(ai);
  1425. ai = next;
  1426. }
  1427. }