rt_socket.c 26 KB

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