嵌入式 hi3518平台增加路由代码

1. <span style="font-family:Courier New;">
2. /********************************** (C) COPYRIGHT *******************************
3. * File Name          : add_route.c
4. * Author             : skdkjzz
5. * Date               : 2014/08/09
6. * Description        : 设置静态ip，添加路由表
7. *********************************************************************************/
8. #include <error.h>
9. #include <stdio.h>
10. #include <string.h>
11. #include <net/if.h>
12. #include <stdlib.h>
13. #include <arpa/inet.h>
14. #include <asm/types.h>            /* glibc 2 conflicts with linux/types.h */
15. #include <net/route.h>
16. #include <sys/errno.h>
17. #include <sys/ioctl.h>
18. #include <sys/types.h>
19. #include <sys/socket.h>
20. #include <netinet/in.h>
21. #include <net/if_arp.h>
22. #include <netinet/if_ether.h>
23. #include <fcntl.h>
24. #include <stdio.h>
25. #include <stdlib.h>
26. #include <errno.h>
27. #include <string.h>
28. #include <unistd.h>
29. #include <signal.h>
30. #include <dirent.h>
31. #include <arpa/inet.h>
32. #include <pthread.h>
33. #include <sys/vfs.h>
34. #include <sys/time.h>
35. #include <sys/wait.h>
36. #include <sys/ioctl.h>
37. #include <sys/types.h>
38. #include <sys/socket.h>
39. #include <net/if.h>
40. #include <netinet/in.h>
41. #include <net/route.h>
42. #include <net/if_arp.h>
43. #define SET_SA_FAMILY(addr, family)                        \
44. memset ((char *) &(addr), '\0', sizeof(addr));        \
45. addr.sa_family = (family);
46. #define SIN_ADDR(x)        (((struct sockaddr_in *) (&(x)))->sin_addr.s_addr)
47. int SetIfAddr(char *Ifname, char *Ipaddr, char *mask)
48. {
49. int fd;
50. //int rc;
51. struct ifreq ifr;
52. struct sockaddr_in *sin;
53. struct rtentry  rt;
54. fd = socket(AF_INET, SOCK_DGRAM, 0);
55. if(fd < 0)
56. {
57. perror("socket  error");
58. return -2;
59. }
60. memset(&ifr,0,sizeof(ifr));
61. strcpy(ifr.ifr_name,Ifname);
62. sin = (struct sockaddr_in*)&ifr.ifr_addr;
63. sin->sin_family = AF_INET;
64. //IP地址
65. if(inet_aton(Ipaddr,&(sin->sin_addr)) < 0)
66. {
67. perror("inet_aton   error");
68. return -3;
69. }
70. if(ioctl(fd,SIOCSIFADDR,&ifr) < 0)
71. {
72. perror("ioctl   SIOCSIFADDR   error");
73. return -4;
74. }
75. //子网掩码
76. if(inet_aton(mask,&(sin->sin_addr)) < 0)
77. {
78. perror("inet_pton   error");
79. return -5;
80. }
81. if(ioctl(fd, SIOCSIFNETMASK, &ifr) < 0)
82. {
83. perror("ioctl");
84. return -6;
85. }
86. return 0;
87. close(fd);
88. }
89. /*
90. * funtion:set the file lock
91. * params:fd is file Num,type is F_WRLCk,F_RDLCK or F_UNLCK
92. -----------------------
93. * time:2012-08-22 22:17
94. * version:1.0
95. */
96. int joseph_lock_set(int fd, int type)
97. {
98. struct flock  lock_file;
99. lock_file.l_whence = SEEK_SET;
100. lock_file.l_start = 0;
101. lock_file.l_len = 0;
102. lock_file.l_type = type;
103. lock_file.l_pid = -1;
104. fcntl(fd, F_GETLK, &lock_file);
105. if (lock_file.l_type != F_UNLCK)
106. {
107. if (lock_file.l_type == F_RDLCK)
108. {
109. }
110. else if (lock_file.l_type == F_WRLCK)
111. {
112. }
113. }
114. lock_file.l_type = type;
115. if ((fcntl(fd, F_SETLKW, &lock_file)) < 0)
116. {
117. return 1;
118. }
119. switch(lock_file.l_type)
120. {
121. case F_RDLCK:
122. {
123. }
124. break;
125. case F_WRLCK:
126. {
127. }
128. break;
129. case F_UNLCK:
130. {
131. return 1;
132. }
133. break;
134. default:
135. break;
136. }/* end of switch  */
137. return 0;
138. }
139. #define JOSEPH_ETH0_IFCG "/etc/sysconfig/network-scripts/ifcfg-eth0"
140. int joseph_read_user_file(char *param_name)
141. {
142. int fd;
143. int param_value;
144. char buf_name[64];
145. char buf[128];
146. FILE *fp;
147. strcpy(buf_name,param_name);
148. fp = fopen(JOSEPH_ETH0_IFCG, "r");
149. if(fp == NULL)
150. {
151. return -1;
152. }
153. fd=fileno(fp);
154. joseph_lock_set(fd, F_RDLCK);//set the write lock
155. while(fgets(buf, sizeof(buf), fp)){
156. if(strstr(buf,buf_name)){
157. printf("%s:[%d] %s = %s",__FUNCTION__,__LINE__,param_name,(buf+strlen(buf_name)+1));
158. }
159. }
160. joseph_lock_set(fd, F_UNLCK);// unlock the file
161. fclose(fp);
162. return 0;
163. }
164. void joseph_alter_ipnc_param(char *joseph_param_name, char *joseph_param_values)
165. {
166. char buf[1024];
167. char conf_path[64] = JOSEPH_ETH0_IFCG;
168. if((joseph_param_name == NULL) ||( joseph_param_values == NULL))
169. {
170. return;
171. }
172. if((strlen(joseph_param_name) == 0) ||( strlen(joseph_param_values) == 0))
173. {
174. return;
175. }
176. sprintf(buf,"sed -i 's/%s=.*/%s=%s/g' %s",\
177. joseph_param_name,joseph_param_name,joseph_param_values,conf_path);
178. system(buf);
179. return;
180. }
181. int  main ()
182. {
183. #if 1
184. static int sock_fd = -1;        /* socket for doing interface ioctls */
185. struct rtentry rt;
186. FILE *db_fd = (FILE *) 0;
187. char ifname[32];
188. u_int32_t dstaddr, gateway;
189. int j,k;
190. char *p;
191. FILE *fp;
192. char cmd[30];
193. char cmd1[5][20];
194. memset(cmd, 0, sizeof(cmd));
195. memset(cmd1, 0, sizeof(cmd1));
196. if((fp=fopen("/etc/sysconfig/network-scripts/ifcfg-eth0","r"))==NULL){
197. printf("Read ifcfg-eth0 error!\n");
198. return -1;
199. }
200. for(j=0;j<5;j++){
201. if((p=fgets(cmd,1024,fp))==NULL){
202. printf("read data error!\n");
203. return -2;
204. }
205. while(*p!='=')p++;
206. p++;
207. k=0;
208. while(*p!='\n')
209. {
210. cmd1[j][k]=*p;
211. k++;
212. p++;
213. }
214. memset(cmd, 0, sizeof(cmd));
215. }
216. if(SetIfAddr(cmd1[0],cmd1[1],cmd1[2])!=0)
217. {
218. printf("config ip failed!\n");
219. return -7;
220. }
221. strcpy(ifname, cmd1[0]);
222. dstaddr = inet_addr(cmd1[3]);
223. gateway = inet_addr(cmd1[4]);
224. /* Get an internet socket for doing socket ioctls. */
225. sock_fd = socket(AF_INET, SOCK_DGRAM, 0);
226. /*open for debug*/
227. db_fd = fopen("/tmp/addrt_db.txt","a+");
228. fprintf(db_fd, "ifname=%s ouraddr=%x gateway=%x\n", ifname, dstaddr, gateway);
229. memset (&rt, 0, sizeof (rt));
230. /*set Destination addr*/
231. SET_SA_FAMILY (rt.rt_dst, AF_INET);
232. SIN_ADDR(rt.rt_dst) = dstaddr;
233. /*set gw addr*/
234. SET_SA_FAMILY (rt.rt_gateway, AF_INET);
235. SIN_ADDR(rt.rt_gateway) = gateway;
236. fprintf(db_fd,"mygateway=%x\n", SIN_ADDR(rt.rt_gateway));
237. /*set genmask addr*/
238. SET_SA_FAMILY (rt.rt_genmask, AF_INET);
239. SIN_ADDR(rt.rt_genmask) = 0L;
240. rt.rt_dev = ifname;
241. rt.rt_flags = RTF_GATEWAY;
242. if (ioctl(sock_fd, SIOCADDRT, &rt) < 0)
243. {
244. fprintf(db_fd,"route add err num=%m\n",errno);
245. printf("Err\n");
246. return 0;
247. }
248. fprintf(db_fd,"route add success route=%x\n", SIN_ADDR(rt.rt_gateway));
249. fclose(db_fd);
250. #else
251. joseph_read_user_file("DEVICE");
252. joseph_read_user_file("IPADDR");
253. joseph_read_user_file("GATEWAY");
254. joseph_read_user_file("DNS");
255. joseph_read_user_file("BROADCAST");
256. joseph_read_user_file("BOOTPROTO");
257. joseph_read_user_file("ONBOOT");
258. joseph_alter_ipnc_param("IPADDR","10.10.1.250");
259. joseph_read_user_file("IPADDR");
260. #endif
261. return 0;
262. }
263. </span>