rt_socket.c 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416
  1. /******************************************************************************
  2. 版权所有:
  3. 文件名称: rt_socket.c
  4. 文件版本: 01.01
  5. 创建作者: sunxi
  6. 创建日期: 2020-06-18
  7. 功能说明: 实时微系统网络接口.所有接口必须在线程中调用,不能在中断中调用。
  8. 其它说明:
  9. 修改记录:
  10. */
  11. /*------------------------------- 头文件 --------------------------------------
  12. */
  13. #include <rt.h>
  14. #include "rt_socket.h"
  15. #include "rt_printf.h"
  16. #include <linux/route.h>
  17. #include "bspconfig.h"
  18. /*------------------------------- 宏定义 --------------------------------------
  19. */
  20. /*------------------------------ 类型结构 -------------------------------------
  21. */
  22. /*------------------------------ 全局变量 -------------------------------------
  23. */
  24. char * g_if_name[] =
  25. {
  26. "eth0",
  27. "eth1",
  28. "eth2",
  29. "eth3",
  30. "eth4",
  31. };
  32. #define IF_NUM ((int)(sizeof(g_if_name)/4))
  33. int g_net_num = CFG_ETH_MAX_PHY; // 正常的网口数量
  34. /*------------------------------ 函数声明 -------------------------------------
  35. */
  36. /*------------------------------ 外部函数 -------------------------------------
  37. 外部函数供其它实体文件引用,必须仔细检查传入参数的合法性.
  38. */
  39. int rt_socket_init(void)
  40. {
  41. unsigned int ret;
  42. // 目前工厂主板标准硬件版本为0,2个网口。
  43. // 以后如果使用3个网口,需修改硬件版本号。
  44. ret = 3;// gpio_get_version();
  45. if(ret < 3)
  46. {
  47. g_net_num = 2;
  48. }
  49. return 0;
  50. }
  51. // 打开关闭
  52. SOCKET rt_socket (int domain, int type, int protocol)
  53. {
  54. SOCKET s;
  55. s = socket(domain, type, protocol);
  56. return s;
  57. }
  58. int rt_socket_close (SOCKET s)
  59. {
  60. if(s > 0)
  61. {
  62. close(s);
  63. }
  64. return 0;
  65. }
  66. //连接
  67. SOCKET rt_accept (SOCKET s, struct sockaddr * addr, unsigned int * addr_len)
  68. {
  69. SOCKET newsock = 0;
  70. newsock = accept(s, addr, addr_len);
  71. return newsock;
  72. }
  73. int rt_bind (SOCKET s, struct sockaddr * addr, unsigned int len)
  74. {
  75. return bind(s, addr, len);
  76. }
  77. int rt_connect (SOCKET s, struct sockaddr * addr, unsigned int len)
  78. {
  79. return connect(s, addr, len);
  80. }
  81. int rt_listen (SOCKET s, int n)
  82. {
  83. return listen(s, n);
  84. }
  85. int rt_shutdown (SOCKET s, int how)
  86. {
  87. return shutdown(s, how);
  88. }
  89. //接收发送
  90. int rt_send(SOCKET s, void *buf, int n, int flags)
  91. {
  92. // flags加上MSG_NOSIGNAL标志,解决主循环在此断开的socket上继续发送数据,引起系统异常的问题。
  93. // return rt_sendto(s,buf,n,flags,NULL,0);
  94. return sendto(s, buf, n, flags | MSG_NOSIGNAL, NULL, 0);
  95. }
  96. int rt_recv (SOCKET s, void *buf, int n, int flags)
  97. {
  98. return rt_recvfrom(s, buf, n, flags, NULL, 0);
  99. }
  100. int rt_sendto (SOCKET s, void *buf, int n,int flags, struct sockaddr * addr, unsigned int addr_len)
  101. {
  102. return sendto(s, buf, n, flags, addr, addr_len);
  103. }
  104. int rt_recvfrom (SOCKET s, void * buf, int n, int flags, struct sockaddr * addr, unsigned int * addr_len)
  105. {
  106. return recvfrom(s, buf, n, flags, addr, addr_len);
  107. }
  108. //控制
  109. int rt_getsockname (SOCKET s, struct sockaddr * addr, unsigned int * len)
  110. {
  111. return getsockname(s, addr, len);
  112. }
  113. int rt_getpeername (SOCKET s, struct sockaddr * addr,unsigned int * len)
  114. {
  115. return getpeername(s, addr, len);
  116. }
  117. int rt_getsockopt (SOCKET s, int level, int optname,void * optval,unsigned int * optlen)
  118. {
  119. return getsockopt(s, level, optname, optval, optlen);
  120. }
  121. int rt_setsockopt (SOCKET s, int level, int optname, void *optval, unsigned int optlen)
  122. {
  123. return setsockopt(s, level, optname, optval, optlen);
  124. }
  125. int rt_sockioctl(SOCKET s, int cmd, uint32_t arg)
  126. {
  127. return ioctl(s, cmd, arg);
  128. }
  129. int set_net_param_fun(char* NetNum,char* para,char* cmd)
  130. {
  131. int ret;
  132. char argv[256] = {0x00};
  133. //sprintf(argv, "/bin/ifconfig %s %s %s", NetNum,para,cmd);
  134. sprintf(argv, "ifconfig %s %s %d.%d.%d.%d", NetNum,para,cmd[0],cmd[1],cmd[2],cmd[3]);
  135. rt_printf("sh cmd=%s\r\n",argv);
  136. ret = system(argv);
  137. return ret;
  138. }
  139. int set_gw_param_fun(char* ip,char* gw)
  140. {
  141. int ret;
  142. char argv[256] = {0x00};
  143. //sprintf(argv, "/bin/ifconfig %s %s %s", NetNum,para,cmd);
  144. sprintf(argv, "route add %s gw %s",ip,gw);
  145. rt_printf("sh cmd=%s\r\n",argv);
  146. ret = system(argv);
  147. return ret;
  148. }
  149. #if 0 // 没有用到
  150. int rt_sockconnected(SOCKET s)
  151. {
  152. struct tcp_info info;
  153. int len=sizeof(info);
  154. int ret=-1;
  155. if(!s)
  156. return ret;
  157. memset(&info, 0, sizeof(info));
  158. ret=rt_getsockopt(s, IPPROTO_TCP, TCP_INFO, (char *)&info, &len);
  159. if(info.tcpi_state==1)
  160. {
  161. //处于链接状态
  162. ret = 0;
  163. }
  164. else
  165. {
  166. //处于断开状态
  167. ret = -1;
  168. }
  169. return ret;
  170. }
  171. int rt_net_set_KeepAlive(SOCKET s, int interval)
  172. {
  173. int val = interval;
  174. //开启keepalive机制
  175. if (rt_setsockopt(s, SOL_SOCKET, SO_KEEPALIVE, &val, sizeof(val)) == -1)
  176. {
  177. // printf("setsockopt SO_KEEPALIVE: %s", strerror(errno));
  178. return -1;
  179. }
  180. /* Default settings are more or less garbage, with the keepalive time
  181. * set to 7200 by default on Linux. Modify settings to make the feature
  182. * actually useful. */
  183. /* Send first probe after interval. */
  184. val = interval;
  185. if (rt_setsockopt(s, IPPROTO_TCP, TCP_KEEPIDLE, &val, sizeof(val)) < 0) {
  186. // printf("setsockopt TCP_KEEPIDLE: %s\n", strerror(errno));
  187. return -1;
  188. }
  189. /* Send next probes after the specified interval. Note that we set the
  190. * delay as interval / 3, as we send three probes before detecting
  191. * an error (see the next setsockopt call). */
  192. val = interval / 3;
  193. if (val == 0) val = 1;
  194. if (rt_setsockopt(s, IPPROTO_TCP, TCP_KEEPINTVL, &val, sizeof(val)) < 0) {
  195. // printf("setsockopt TCP_KEEPINTVL: %s\n", strerror(errno));
  196. return -1;
  197. }
  198. /* Consider the socket in error state after three we send three ACK
  199. * probes without getting a reply. */
  200. val = 3;
  201. if (rt_setsockopt(s, IPPROTO_TCP, TCP_KEEPCNT, &val, sizeof(val)) < 0) {
  202. // printf("setsockopt TCP_KEEPCNT: %s\n", strerror(errno));
  203. return -1;
  204. }
  205. return 0;
  206. }
  207. #endif
  208. /******************************************************************************
  209. 函数名称:rt_if_up
  210. 函数版本:01.01
  211. 创建作者:sunxi
  212. 创建日期:2020-06-18
  213. 函数说明:UP一个接口。
  214. 参数说明:
  215. if_name: 接口名称,如eth0,fei1等。
  216. 返回值 :
  217. >=0 : 成功。
  218. -1 : 失败。
  219. 更新记录:
  220. */
  221. #if 0
  222. int rt_if_up(int index)
  223. {
  224. struct ifreq ifr;
  225. SOCKET sock;
  226. if(index >= IF_NUM)
  227. {
  228. return -1;
  229. }
  230. sock = rt_socket(AF_INET, SOCK_DGRAM, 0);
  231. if (sock == RT_SOCKET_ERR)
  232. {
  233. rt_printf("rt_if_up:: rt_socket::error\n");
  234. return -1;
  235. }
  236. memset(&ifr,0,sizeof(ifr));
  237. strcpy(ifr.ifr_name,g_if_name[index]);
  238. /* 得到标志*/
  239. if(rt_sockioctl(sock,SIOCGIFFLAGS,(uint32_t)&ifr) != 0)
  240. {
  241. rt_socket_close(sock);
  242. return -1;
  243. }
  244. /* 修改标志*/
  245. ifr.ifr_ifru.ifru_flags |= IFF_UP;
  246. /* 写回标志*/
  247. if(rt_sockioctl(sock,SIOCSIFFLAGS,(uint32_t)&ifr) != 0)
  248. {
  249. rt_socket_close(sock);
  250. return -1;
  251. }
  252. rt_socket_close(sock);
  253. return 0;
  254. }
  255. #endif
  256. int rt_if_up(int index)
  257. {
  258. int ret;
  259. char argv[256] = {0x00};
  260. sprintf(argv, "/sbin/ifconfig eth%d up", index);
  261. ret = system(argv);
  262. return ret;
  263. }
  264. /******************************************************************************
  265. 函数名称:rt_if_down
  266. 函数版本:01.01
  267. 创建作者:sunxi
  268. 创建日期:2020-06-18
  269. 函数说明:DOWN一个接口。
  270. 参数说明:
  271. if_name: 接口名称,如eth0,fei1等。
  272. 返回值 :
  273. >=0 : 成功。
  274. -1 : 失败。
  275. 更新记录:
  276. */
  277. int rt_if_down(int index)
  278. {
  279. int ret;
  280. char argv[256] = {0x00};
  281. sprintf(argv, "/sbin/ifconfig eth%d down", index);
  282. ret = system(argv);
  283. return ret;
  284. }
  285. int c2n(char c)
  286. {
  287. if(c>='0' && c <= '9')
  288. {
  289. return c-'0';
  290. }
  291. if(c>='A' && c <= 'F')
  292. {
  293. return c-'A'+10;
  294. }
  295. if(c>='a' && c <= 'f')
  296. {
  297. return c-'a'+10;
  298. }
  299. return -1;
  300. }
  301. // 将标准MAC地址字符串转换为6字节MAC地址
  302. //80-81-00-b0-32-81
  303. int rt_if_sz2mac(char *sz, unsigned char *mac)
  304. {
  305. int i;
  306. int c0,c1;
  307. if(sz == NULL || mac == NULL)
  308. {
  309. return -1;
  310. }
  311. if(strlen(sz) != 17)
  312. {
  313. return -1;
  314. }
  315. i=0;
  316. while(i<17)
  317. {
  318. if((i%3 == 2))
  319. {
  320. if(sz[i++] != '-')
  321. {
  322. return -2;
  323. }
  324. }
  325. c0 = c2n(sz[i++]);
  326. if(c0<0)
  327. {
  328. return -3;
  329. }
  330. c1 = c2n(sz[i++]);
  331. if(c1<0)
  332. {
  333. return -4;
  334. }
  335. *mac++ = (c0<<4) | c1;
  336. }
  337. return 0;
  338. }
  339. /*
  340. int rt_if_mac_set(int index, unsigned char *mac)
  341. {
  342. int ret;
  343. struct ifreq ifr;
  344. int sock;
  345. if((unsigned int)index >= IF_NUM)
  346. {
  347. return -1;
  348. }
  349. if(mac == NULL)
  350. {
  351. return -1;
  352. }
  353. memset(&ifr, 0, sizeof(struct ifreq));
  354. //创建socket
  355. sock = rt_socket(AF_INET, SOCK_DGRAM, 0);
  356. if(sock < 0)
  357. {
  358. rt_printf("Not create network rt_socket connection\\n");
  359. return (-1);
  360. }
  361. strncpy(ifr.ifr_name,g_if_name[index],IFNAMSIZ);
  362. ifr.ifr_name[IFNAMSIZ - 1] = 0;
  363. //关闭网卡
  364. ifr.ifr_flags &= ~IFF_UP;
  365. if(ioctl(sock,SIOCSIFFLAGS,(uint32_t)&ifr) < 0)
  366. {
  367. //rt_printf("SIOCSIFFLAGS");
  368. close(sock);
  369. return (-1);
  370. }
  371. //获取Mac地址
  372. ret = ioctl(sock,SIOCGIFHWADDR,(uint32_t)&ifr);
  373. if( ret < 0)
  374. {
  375. //rt_printf("Not setup SIOCGIFHWADDR(ret=%d)\n",ret);
  376. close(sock);
  377. return (-1);
  378. }
  379. //填充Mac地址
  380. ifr.ifr_hwaddr.sa_data[0]=mac[0];
  381. ifr.ifr_hwaddr.sa_data[1]=mac[1];
  382. ifr.ifr_hwaddr.sa_data[2]=mac[2];
  383. ifr.ifr_hwaddr.sa_data[3]=mac[3];
  384. ifr.ifr_hwaddr.sa_data[4]=mac[4];
  385. ifr.ifr_hwaddr.sa_data[5]=mac[5];
  386. //修改Mac地址
  387. ret = ioctl(sock, SIOCSIFHWADDR, (uint32_t)&ifr);
  388. if( ret < 0)
  389. {
  390. rt_printf("Not setup SIOCSIFHWADDR(ret=%d)\n",ret);
  391. close(sock);
  392. return (-1);
  393. }
  394. //启动网卡
  395. ifr.ifr_flags |= IFF_UP|IFF_RUNNING;
  396. if(ioctl(sock,SIOCSIFFLAGS,(uint32_t)&ifr)<0)
  397. {
  398. rt_printf("SIOCSIFFLAGS");
  399. close(sock);
  400. return (-1);
  401. }
  402. close(sock);
  403. return 0;
  404. }
  405. */
  406. int rt_if_mac_set(int index, unsigned char *mac)
  407. {
  408. char argv[256] = {0x00};
  409. int ret;
  410. char NetNum[20]="";
  411. sprintf(NetNum,"eth%d", index);
  412. sprintf(argv, "ifconfig %s hw ether %02X:%02X:%02X:%02X:%02X:%02X", NetNum,mac[0],mac[1],mac[2],mac[3],mac[4],mac[5]);
  413. rt_printf("sh cmd=%s\r\n",argv);
  414. ret = system(argv);
  415. return ret;
  416. }
  417. int rt_if_mac_get(int index, unsigned char *mac)
  418. {
  419. int ret;
  420. struct ifreq ifr;
  421. int sock;
  422. if((unsigned int)index >= IF_NUM)
  423. {
  424. return -1;
  425. }
  426. if(mac == NULL)
  427. {
  428. return -1;
  429. }
  430. memset(&ifr, 0, sizeof(struct ifreq));
  431. //创建socket
  432. sock = rt_socket(AF_INET, SOCK_DGRAM, 0);
  433. if(sock < 0)
  434. {
  435. rt_printf("Not create network rt_socket connection\\n");
  436. return (-1);
  437. }
  438. strncpy(ifr.ifr_name,g_if_name[index],IFNAMSIZ);
  439. ifr.ifr_name[IFNAMSIZ - 1] = 0;
  440. //获取Mac地址
  441. ret = ioctl(sock,SIOCGIFHWADDR,(uint32_t)&ifr);
  442. if( ret < 0)
  443. {
  444. // rt_printf("Not setup SIOCGIFHWADDR(ret=%d)\n",ret);
  445. close(sock);
  446. return (-1);
  447. }
  448. //填充Mac地址
  449. mac[0] = ifr.ifr_hwaddr.sa_data[0];
  450. mac[1] = ifr.ifr_hwaddr.sa_data[1];
  451. mac[2] = ifr.ifr_hwaddr.sa_data[2];
  452. mac[3] = ifr.ifr_hwaddr.sa_data[3];
  453. mac[4] = ifr.ifr_hwaddr.sa_data[4];
  454. mac[5] = ifr.ifr_hwaddr.sa_data[5];
  455. close(sock);
  456. return 0;
  457. }
  458. /******************************************************************************
  459. 函数名称:rt_if_ip_set
  460. 函数版本:01.01
  461. 创建作者:sunxi
  462. 创建日期:2020-06-18
  463. 函数说明:设置一个接口的IP地址。
  464. 参数说明:
  465. if_name: 接口名称,如eth0,fei1等。
  466. ip : IP地址。
  467. 返回值 :
  468. >=0 : 成功。
  469. -1 : 失败。
  470. 更新记录:
  471. */
  472. int rt_if_ip_set(int index, unsigned char * ip)
  473. {
  474. int ret;
  475. char NetNum[20]="";
  476. if((unsigned int)index >= IF_NUM)
  477. {
  478. return -1;
  479. }
  480. sprintf(NetNum,"eth%d",index);
  481. ret = set_net_param_fun(NetNum," ",ip); //设置ip
  482. return ret;
  483. }
  484. /******************************************************************************
  485. 函数名称:rt_if_ip_get
  486. 函数版本:01.01
  487. 创建作者:sunxi
  488. 创建日期:2020-06-18
  489. 函数说明:得到一个接口的IP地址。
  490. 参数说明:
  491. if_name: 接口名称,如eth0,fei1等。
  492. ip : 返回的IP地址在*if_addr中。
  493. 返回值 :
  494. >=0 : 成功。
  495. -1 : 失败。
  496. 更新记录:
  497. */
  498. int rt_if_ip_get(int index, unsigned char *ip)
  499. {
  500. unsigned int addr;
  501. struct ifreq ifr;
  502. SOCKET sock;
  503. if((unsigned int)index >= IF_NUM)
  504. {
  505. return -1;
  506. }
  507. if(ip == NULL)
  508. {
  509. return -1;
  510. }
  511. sock = rt_socket(AF_INET, SOCK_DGRAM, 0);
  512. if (sock == RT_SOCKET_ERR)
  513. {
  514. rt_printf("%s:: rt_socket [%d]::error\n", __FUNCTION__, index);
  515. return -1;
  516. }
  517. memset(&ifr,0,sizeof(ifr));
  518. strcpy(ifr.ifr_name,g_if_name[index]);
  519. if(rt_sockioctl(sock,SIOCGIFADDR,(uint32_t)&ifr) != 0)
  520. {
  521. rt_socket_close(sock);
  522. return -1;
  523. }
  524. addr = ((struct sockaddr_in *)&ifr.ifr_addr)->sin_addr.s_addr;
  525. addr = ntohl(addr);
  526. memcpy(ip,&addr,4);
  527. rt_socket_close(sock);
  528. return 0;
  529. }
  530. /******************************************************************************
  531. 函数名称:rt_if_mask_set
  532. 函数版本:01.01
  533. 创建作者:sunxi
  534. 创建日期:2020-06-18
  535. 函数说明:设置一个接口的子网屏蔽。
  536. 参数说明:
  537. if_name: 接口名称,如eth0,fei1等。
  538. mask : 子网屏蔽
  539. 返回值 :
  540. >=0 : 成功。
  541. -1 : 失败。
  542. 更新记录:
  543. */
  544. int rt_if_mask_set(int index, unsigned char * mask)
  545. {
  546. int ret;
  547. char NetNum[20]="";
  548. if((unsigned int)index >= IF_NUM)
  549. {
  550. return -1;
  551. }
  552. sprintf(NetNum,"eth%d",index);
  553. ret = set_net_param_fun(NetNum,"netmask",mask); //设置子网掩码
  554. return ret;
  555. }
  556. /******************************************************************************
  557. 函数名称:rt_if_mask_get
  558. 函数版本:01.01
  559. 创建作者:sunxi
  560. 创建日期:2020-06-18
  561. 函数说明:得到一个接口的子网屏蔽。
  562. 参数说明:
  563. if_name: 接口名称,如eth0,fei1等。
  564. mask : 返回的子网屏蔽在*if_mask中。
  565. 返回值 :
  566. >=0 : 成功。
  567. -1 : 失败。
  568. 更新记录:
  569. */
  570. int rt_if_mask_get(int index, unsigned char *mask)
  571. {
  572. unsigned int addr;
  573. struct ifreq ifr;
  574. SOCKET sock;
  575. if((unsigned int)index >= IF_NUM)
  576. {
  577. return -1;
  578. }
  579. if(mask == NULL)
  580. {
  581. return -1;
  582. }
  583. sock = rt_socket(AF_INET, SOCK_DGRAM, 0);
  584. if (sock == RT_SOCKET_ERR)
  585. {
  586. rt_printf("rt_if_mask_get:: rt_socket::error\n");
  587. return -1;
  588. }
  589. memset(&ifr,0,sizeof(ifr));
  590. strcpy(ifr.ifr_name,g_if_name[index]);
  591. if(rt_sockioctl(sock,SIOCGIFNETMASK,(uint32_t)&ifr) != 0)
  592. {
  593. rt_socket_close(sock);
  594. return -1;
  595. }
  596. addr = ((struct sockaddr_in *)&ifr.ifr_netmask)->sin_addr.s_addr;
  597. addr = ntohl(addr);
  598. memcpy(mask,&addr,4);
  599. rt_socket_close(sock);
  600. return 0;
  601. }
  602. /******************************************************************************
  603. 函数名称:rt_route_add
  604. 函数版本:01.01
  605. 创建作者:xxxxxx
  606. 创建日期:2008-04-09
  607. 函数说明:增加一个接口的路由。
  608. 参数说明:
  609. destination : 目标网络。
  610. gateway : 网关
  611. 返回值 :
  612. >=0 : 成功。
  613. -1 : 失败。
  614. 更新记录:
  615. */
  616. int rt_route_add(int index,unsigned char * destination,unsigned char *gateway)
  617. {
  618. int sock;
  619. unsigned int destination2,gateway2;
  620. struct rtentry rt;
  621. if(destination == NULL || gateway == NULL)
  622. {
  623. return -1;
  624. }
  625. memcpy(&destination2,destination,4);
  626. memcpy(&gateway2,gateway,4);
  627. sock = rt_socket(AF_INET, SOCK_DGRAM, 0);
  628. if( sock < 0 )
  629. {
  630. rt_printf("Create Socket Erro\r\n" );
  631. return -1;
  632. }
  633. memset(&rt,0,sizeof(struct rtentry));
  634. rt.rt_dev = g_if_name[index];
  635. rt.rt_flags = RTF_UP | RTF_GATEWAY;
  636. ((struct sockaddr_in *) &rt.rt_gateway)->sin_family = AF_INET;
  637. ((struct sockaddr_in *) &rt.rt_gateway)->sin_addr.s_addr = htonl(gateway2);
  638. ((struct sockaddr_in *) &rt.rt_dst)->sin_family = AF_INET;
  639. ((struct sockaddr_in *) &rt.rt_dst)->sin_addr.s_addr = htonl(destination2);
  640. /* The netmask for the destination net; ’255.255.255.255’ for a
  641. host destination and ’0.0.0.0’ for the default route.
  642. */
  643. ((struct sockaddr_in *) &rt.rt_genmask)->sin_family = AF_INET;
  644. ((struct sockaddr_in *) &rt.rt_genmask)->sin_addr.s_addr = 0 ;
  645. if( -1 == ioctl(sock,SIOCADDRT,(uint32_t)&rt))
  646. {
  647. rt_printf("rt_route_add fail");
  648. close(sock);
  649. return -1;
  650. }
  651. close(sock);
  652. return 0;
  653. }
  654. /******************************************************************************
  655. 函数名称:rt_route_delete
  656. 函数版本:01.01
  657. 创建作者:sunxi
  658. 创建日期:2020-6-19
  659. 函数说明:删除一个接口的路由。
  660. 参数说明:
  661. if_name_prefic : 接口名称前缀,如eth,fei等。
  662. unit : 网卡编号,如,0,1,2等
  663. 返回值 :
  664. >=0 : 成功。
  665. -1 : 失败。
  666. 更新记录:
  667. */
  668. int rt_route_del(int index,unsigned char * destination,unsigned char *gateway)
  669. {
  670. int sock;
  671. unsigned int destination2,gateway2;
  672. struct rtentry rt;
  673. if(destination == NULL || gateway == NULL)
  674. {
  675. return -1;
  676. }
  677. memcpy(&destination2,destination,4);
  678. memcpy(&gateway2,gateway,4);
  679. sock = rt_socket(AF_INET, SOCK_DGRAM, 0);
  680. if( sock < 0 )
  681. {
  682. rt_printf("Create Socket Erro\r\n" );
  683. return -1;
  684. }
  685. memset(&rt,0,sizeof(struct rtentry));
  686. rt.rt_dev = g_if_name[index];
  687. rt.rt_flags = RTF_UP | RTF_GATEWAY;
  688. // ((struct sockaddr_in *) &rt.rt_gateway)->sin_family = AF_INET;
  689. // ((struct sockaddr_in *) &rt.rt_gateway)->sin_addr.s_addr = htonl(gateway2);
  690. ((struct sockaddr_in *) &rt.rt_dst)->sin_family = AF_INET;
  691. ((struct sockaddr_in *) &rt.rt_dst)->sin_addr.s_addr = htonl(destination2);
  692. /* The netmask for the destination net; ’255.255.255.255’ for a
  693. host destination and ’0.0.0.0’ for the default route.
  694. */
  695. ((struct sockaddr_in *) &rt.rt_genmask)->sin_family = AF_INET;
  696. ((struct sockaddr_in *) &rt.rt_genmask)->sin_addr.s_addr = 0 ;
  697. if( -1 == ioctl(sock,SIOCDELRT,(uint32_t)&rt))
  698. {
  699. rt_printf("rt_route_del fail");
  700. close(sock);
  701. return -1;
  702. }
  703. close(sock);
  704. return 0;
  705. }
  706. int net_gateway_set(unsigned char *gateway)
  707. {
  708. int ret;
  709. char argv_del[] = "/sbin/route del default";
  710. char argv_add[256] = {0x00};
  711. if(gateway == 0)
  712. {
  713. return -1;
  714. }
  715. sprintf(argv_add, "/sbin/route add default gw %d.%d.%d.%d", gateway[0], gateway[1], gateway[2], gateway[3]);
  716. ret = system(argv_del);
  717. ret = system(argv_add);
  718. return ret;
  719. }
  720. int net_if_set(int if_index,unsigned char *mac,unsigned char *ip,unsigned char *mask)
  721. {
  722. if((unsigned int)if_index >= IF_NUM)
  723. {
  724. return -1;
  725. }
  726. if(mac)
  727. {
  728. rt_if_mac_set(if_index,mac);
  729. }
  730. if(ip)
  731. {
  732. rt_if_ip_set(if_index,ip);
  733. }
  734. if(mask)
  735. {
  736. rt_if_mask_set(if_index,mask);
  737. }
  738. return 0;
  739. }
  740. int net_if_get(int if_index,unsigned char *mac,unsigned char *ip,unsigned char *mask)
  741. {
  742. int ret;
  743. char buf[64];
  744. if((unsigned int)if_index >= IF_NUM)
  745. {
  746. return -1;
  747. }
  748. if(mac)
  749. {
  750. ret = rt_if_mac_get(if_index,mac);
  751. if(ret != 0)
  752. {
  753. return -1;
  754. }
  755. sprintf(buf,"%02x:%02x:%02x:%02x:%02x:%02x",mac[0],mac[1],mac[2],mac[3],mac[4],mac[5]);
  756. }
  757. if(ip)
  758. {
  759. ret = rt_if_ip_get(if_index,ip);
  760. if(ret != 0)
  761. {
  762. return -2;
  763. }
  764. sprintf(buf,"%d.%d.%d.%d",ip[0],ip[1],ip[2],ip[3]);
  765. }
  766. if(mask)
  767. {
  768. ret = rt_if_mask_get(if_index,mask);
  769. if(ret != 0)
  770. {
  771. return -3;
  772. }
  773. sprintf(buf,"%d.%d.%d.%d",mask[0],mask[1],mask[2],mask[3]);
  774. }
  775. return 0;
  776. }
  777. int rt_get_net_linkstatus(int if_index)
  778. {
  779. int skfd = -1;
  780. struct ifreq ifr;
  781. unsigned short *data, mii_val;
  782. unsigned phy_id;
  783. if(if_index < 0)
  784. {
  785. return -1;
  786. }
  787. if (( skfd = rt_socket( AF_INET, SOCK_DGRAM, 0 ) ) < 0 )
  788. {
  789. return -1;
  790. }
  791. /* Get the vitals from the interface. */
  792. strncpy(ifr.ifr_name, g_if_name[if_index], IFNAMSIZ);
  793. if (ioctl(skfd, SIOCGMIIPHY, (uint32_t)&ifr) < 0)
  794. {
  795. rt_printf( "SIOCGMIIPHY on %s failed!\n", ifr.ifr_name);
  796. close(skfd);
  797. return -2;
  798. }
  799. data = (unsigned short *)(&ifr.ifr_data);
  800. phy_id = data[0];
  801. data[1] = 1;
  802. if (ioctl(skfd, SIOCGMIIREG, (uint32_t)&ifr) < 0)
  803. {
  804. rt_printf( "SIOCGMIIREG on %s failed!\n", ifr.ifr_name);
  805. close(skfd);
  806. return -3;
  807. }
  808. close(skfd);
  809. mii_val = data[3];
  810. return(((mii_val & 0x0016) == 0x0004) ? 0 : 1);
  811. }
  812. #define NET_NUM_CNT 3
  813. int rt_get_netcard_id_from_socket(SOCKET s)
  814. {
  815. struct sockaddr_in local_addr;
  816. unsigned int len = sizeof(local_addr);
  817. char buf[32]={0};
  818. unsigned int ip=0;
  819. int i ,j;
  820. int index;
  821. rt_getsockname(s, (struct sockaddr *)&local_addr, &len);
  822. index = -1;
  823. for(i=0;i<g_net_num;i++)
  824. {
  825. //获取网卡IP
  826. rt_if_ip_get(i, buf);
  827. memcpy(&ip, buf, 4);
  828. if(ip==local_addr.sin_addr.s_addr)
  829. {
  830. index = i;
  831. break;
  832. }
  833. else
  834. {
  835. for(j=1;j<=NET_NUM_CNT;j++)
  836. {
  837. rt_if_ip_get2(i, j, buf);
  838. memcpy(&ip, buf, 4);
  839. if(ip==local_addr.sin_addr.s_addr)
  840. {
  841. return i;
  842. }
  843. }
  844. }
  845. }
  846. return index;
  847. }
  848. int net_route_set(unsigned char *routenet, unsigned char *routenetmask, unsigned char *routegate)
  849. {
  850. int ret;
  851. char buf_routenet[32];
  852. char buf_routenetmask[32];
  853. char buf_routegate[32];
  854. char argv_add[512] = {0x00};
  855. char argv_add2[512] = {0x00};
  856. if((routenet == 0) || (routenetmask==0) || (routegate==0))
  857. {
  858. return -1;
  859. }
  860. if((routenet[0]==0) && (routenet[1]==0) && (routenet[2]==0) && (routenet[3]==0))
  861. {
  862. return 0;
  863. }
  864. if((routegate[0]==0) && (routegate[1]==0) && (routegate[2]==0) && (routegate[3]==0))
  865. {
  866. return 0;
  867. }
  868. sprintf(buf_routenet,"%d.%d.%d.%d",routenet[0],routenet[1],routenet[2],routenet[3]);
  869. sprintf(buf_routenetmask,"%d.%d.%d.%d",routenetmask[0],routenetmask[1],routenetmask[2],routenetmask[3]);
  870. sprintf(buf_routegate,"%d.%d.%d.%d",routegate[0],routegate[1],routegate[2],routegate[3]);
  871. sprintf(argv_add, "/sbin/route add -net %s netmask %s gw %s", buf_routenet, buf_routenetmask, buf_routegate);
  872. sprintf(argv_add2, "/sbin/route add -host %s gw %s", buf_routenet, buf_routegate);
  873. if((routenetmask[0]==0) && (routenetmask[1]==0) && (routenetmask[2]==0) && (routenetmask[3]==0))
  874. {
  875. ret = system(argv_add2);
  876. }
  877. else
  878. {
  879. ret = system(argv_add);
  880. }
  881. return ret;
  882. }
  883. int rt_if_ip_set2(int eth_id, int index, unsigned char *ip)
  884. {
  885. int ret;
  886. char NetNum[20]="";
  887. if((unsigned int)index >= IF_NUM)
  888. {
  889. return -1;
  890. }
  891. sprintf(NetNum,"eth%d:%d", eth_id, index);
  892. ret = set_net_param_fun(NetNum," ",ip); //设置ip
  893. return ret;
  894. }
  895. int rt_if_mask_set2(int eth_id, int index, unsigned char *mask)
  896. {
  897. int ret;
  898. char NetNum[20]="";
  899. if((unsigned int)index >= IF_NUM)
  900. {
  901. return -1;
  902. }
  903. sprintf(NetNum,"eth%d:%d", eth_id, index);
  904. ret = set_net_param_fun(NetNum,"netmask",mask); //设置子网掩码
  905. return ret;
  906. }
  907. int net_if_set2(int eth_id, int if_index,unsigned char *mac,unsigned char *ip,unsigned char *mask)
  908. {
  909. if((unsigned int)if_index > IF_NUM)
  910. {
  911. return -1;
  912. }
  913. if(mac)
  914. {
  915. //rt_if_mac_set(if_index,mac);
  916. mac=mac;
  917. }
  918. if(ip)
  919. {
  920. rt_if_ip_set2(eth_id, if_index, ip);
  921. }
  922. if(mask)
  923. {
  924. rt_if_mask_set2(eth_id, if_index, mask);
  925. }
  926. return 0;
  927. }
  928. int rt_if_ip_get2(int eth_id, int index, unsigned char *ip)
  929. {
  930. unsigned int addr;
  931. char buf[32];
  932. struct ifreq ifr;
  933. SOCKET sock;
  934. if((unsigned int)eth_id >= IF_NUM)
  935. {
  936. return -1;
  937. }
  938. if(ip == NULL)
  939. {
  940. return -1;
  941. }
  942. sock = rt_socket(AF_INET, SOCK_DGRAM, 0);
  943. if (sock == RT_SOCKET_ERR)
  944. {
  945. rt_printf("%s:: rt_socket [%d][%d]::error\n", __FUNCTION__, eth_id, index);
  946. return -1;
  947. }
  948. sprintf(buf, "eth%d:%d", eth_id, index);
  949. memset(&ifr,0,sizeof(ifr));
  950. strcpy(ifr.ifr_name,buf);
  951. if(rt_sockioctl(sock,SIOCGIFADDR,(uint32_t)&ifr) != 0)
  952. {
  953. rt_socket_close(sock);
  954. return -1;
  955. }
  956. addr = ((struct sockaddr_in *)&ifr.ifr_addr)->sin_addr.s_addr;
  957. addr = ntohl(addr);
  958. memcpy(ip,&addr,4);
  959. rt_socket_close(sock);
  960. return 0;
  961. }
  962. int rt_if_down2(int eth_id, int index)
  963. {
  964. int ret;
  965. char argv[256] = {0x00};
  966. sprintf(argv, "/sbin/ifconfig eth%d:%d down", eth_id, index);
  967. ret = system(argv);
  968. return ret;
  969. }
  970. /*------------------------------ 内部函数 -------------------------------------
  971. 内部函数以下划线‘_’开头,不需要检查参数的合法性.
  972. */
  973. /*------------------------------ 测试函数 -------------------------------------
  974. 一个实体文件必须带一个本模块的测试函数来进行单元测试,如果的确不方便在本模块中
  975. 进行单元测试,必须在此注明实际的测试位置(例如在哪个实体文件中使用哪个测试函数).
  976. */
  977. #define IF_INDEX 1
  978. int net_if_test(void)
  979. {
  980. //uint32_t us0;
  981. unsigned char mac[6];
  982. unsigned char ip[4];
  983. unsigned char mask[4];
  984. unsigned char gateway[4];
  985. rt_printf("net_if_test begin...\r\n");
  986. // 得到并打印旧的网络参数
  987. net_if_get(IF_INDEX,mac,ip,mask);
  988. rt_printf("old mac: %02x:%02x:%02x:%02x:%02x:%02x\r\n",mac[0],mac[1],mac[2],mac[3],mac[4],mac[5]);
  989. rt_printf("old ip:%d.%d.%d.%d\r\n",ip[0],ip[1],ip[2],ip[3]);
  990. rt_printf("old mask:%d.%d.%d.%d\r\n",mask[0],mask[1],mask[2],mask[3]);
  991. rt_printf("old gateway:%d.%d.%d.%d\r\n",gateway[0],gateway[1],gateway[2],gateway[3]);
  992. // 设置接口MAC地址
  993. mac[0] = 'H';
  994. mac[1] = 'G';
  995. mac[2] = 192;
  996. mac[3] = 168;
  997. mac[4] = 1;
  998. mac[5] = 240;
  999. // 设置IP
  1000. memcpy(ip,mac+2,4);
  1001. // 设置MASK
  1002. mask[0] = 255;
  1003. mask[1] = 255;
  1004. mask[2] = 0;
  1005. mask[3] = 0;
  1006. // 设置网关
  1007. gateway[0] = 192;
  1008. gateway[1] = 168;
  1009. gateway[2] = 1;
  1010. gateway[3] = 254;
  1011. // 设置网络参数
  1012. //us0 = ustimer_get_origin();
  1013. net_if_set(IF_INDEX,mac,ip,mask);
  1014. net_gateway_set(gateway);
  1015. //rt_printf("net_if_set duration:%dus.\r\n",ustimer_get_duration(us0));
  1016. // 得到并打印新的网络参数
  1017. net_if_get(IF_INDEX,mac,ip,mask);
  1018. rt_printf("new mac: %02x:%02x:%02x:%02x:%02x:%02x\r\n",mac[0],mac[1],mac[2],mac[3],mac[4],mac[5]);
  1019. rt_printf("new ip:%d.%d.%d.%d\r\n",ip[0],ip[1],ip[2],ip[3]);
  1020. rt_printf("new mask:%d.%d.%d.%d\r\n",mask[0],mask[1],mask[2],mask[3]);
  1021. rt_printf("new gateway:%d.%d.%d.%d\r\n",gateway[0],gateway[1],gateway[2],gateway[3]);
  1022. rt_printf("net_if_test ok\r\n");
  1023. return 0;
  1024. }
  1025. int socket_test(void)
  1026. {
  1027. unsigned char buf[512];
  1028. SOCKET sock, new_fd;
  1029. struct sockaddr_in server_addr;
  1030. struct sockaddr_in client_addr;
  1031. int sin_size, portnumber;
  1032. char hello[] = "Hello! Are You Fine?\n";
  1033. portnumber = 1111;
  1034. /*服务器端开始建立socket描述符*/
  1035. if ((sock = rt_socket(AF_INET, SOCK_STREAM, 0)) == RT_SOCKET_ERR)
  1036. {
  1037. rt_printf ("Socket error:%d\n\a", sock);
  1038. }
  1039. /*服务器端填充sockaddr结构*/
  1040. memset (&server_addr, 0, sizeof (struct sockaddr_in));//将
  1041. server_addr.sin_family = AF_INET;
  1042. server_addr.sin_addr.s_addr = htonl (INADDR_ANY);
  1043. server_addr.sin_port = htons (portnumber);
  1044. /*捆绑sock描述符*/
  1045. if (rt_bind (sock, (struct sockaddr *)(&server_addr), sizeof (struct sockaddr)) == -1)
  1046. {
  1047. rt_printf ("Bind error:%d\n\a", sock);
  1048. }
  1049. /*监听sock描述符*/
  1050. if (rt_listen (sock, 5) == -1)
  1051. {
  1052. rt_printf ("Listen error:%d\n\a", sock);
  1053. }
  1054. while(1)
  1055. {
  1056. /*服务器阻塞,直到客户程序建立连接*/
  1057. sin_size = sizeof (struct sockaddr_in);
  1058. if ((new_fd = rt_accept (sock, (struct sockaddr*)(&client_addr), &sin_size)) == RT_SOCKET_ACCEPT_RST)
  1059. {
  1060. rt_printf ("Accept error:%d\n\a", sock);
  1061. }
  1062. rt_printf ("Server get connection from %d\n", new_fd);
  1063. if (rt_send(new_fd, hello, strlen(hello),0) == -1)
  1064. {
  1065. rt_printf ("Write Error:%d\n", new_fd);
  1066. }
  1067. while(1)
  1068. {
  1069. int flag;
  1070. if ((flag = rt_recv (new_fd, buf, 512, 0)) < 0)
  1071. {
  1072. rt_printf ("Reading data error!\n");
  1073. break;
  1074. }
  1075. if (flag == 0)
  1076. {
  1077. rt_printf ("Ending current connection!\n");
  1078. break;
  1079. }
  1080. else
  1081. {
  1082. rt_printf("-->%s\n", buf);
  1083. if (strstr (buf, "exit"))
  1084. {
  1085. rt_printf ("Ending current connection!\n");
  1086. break;
  1087. }
  1088. }
  1089. rt_socket_close (new_fd);
  1090. }
  1091. rt_socket_close (sock);
  1092. }
  1093. return 0;
  1094. }
  1095. /*------------------------------ 文件结束 -------------------------------------
  1096. */