sys.c 28 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148
  1. #include <sys/types.h>
  2. #include <sys/stat.h>
  3. #include <fcntl.h>
  4. #include <unistd.h>
  5. #include <string.h>
  6. #include <limits.h>
  7. #include "elog.h"
  8. #include "sys.h"
  9. #include "upgrade.h"
  10. #include "paras.h"
  11. #include "cfg.h"
  12. #include "file.h"
  13. #include "lock.h"
  14. #include "thread.h"
  15. #include "shell.h"
  16. #include "common.h"
  17. #include "sqlite_handle.h"
  18. //#include <sys/sysinfo.h>
  19. #include <sys/statvfs.h>
  20. #if 0
  21. #define LOGE log_e
  22. #define LOGW log_w
  23. #define LOGD log_d
  24. #else
  25. #define LOGD printf
  26. #define LOGW printf
  27. #define LOGE printf
  28. #endif
  29. //#define USE_BASH
  30. #define USE_FILE
  31. //#define USE_API
  32. #define DFLT_NIC "eth0"
  33. //#define FILE_NET "/root/interfaces"
  34. #define FILE_NET "/etc/network/interfaces"
  35. #define FILE_DNS "/etc/dnsmasq.conf"
  36. typedef struct {
  37. char *pbuf;
  38. int dlen;
  39. int blen;
  40. }node_t;
  41. static int set_net(NetworkInfo_t *para, int ipv);
  42. static int node_free(node_t *nd)
  43. {
  44. if(!nd) {
  45. return -1;
  46. }
  47. free(nd->pbuf);
  48. nd->pbuf = NULL;
  49. nd->dlen = 0;
  50. nd->blen = 0;
  51. return 0;
  52. }
  53. static int exe_cmd(char *cmd, node_t *nd)
  54. {
  55. int r=0;
  56. FILE *fp=NULL;
  57. char *pbuf=NULL;
  58. int onelen=1000;
  59. int dlen=0,blen=onelen;
  60. fp = popen(cmd, "r");
  61. if (fp == NULL) {
  62. return -1;
  63. }
  64. if(nd) {
  65. pbuf = malloc(onelen);
  66. while(pbuf) {
  67. r = fread(pbuf+dlen, sizeof(char), onelen, fp);
  68. if(r<=0) {
  69. LOGE("exe_cmd fread failed\n");
  70. break;
  71. }
  72. dlen += r;
  73. if(dlen+onelen>blen) {
  74. char* p = realloc(pbuf, dlen+onelen);
  75. if(!p) {
  76. LOGE("exe_cmd relloac %d failed\n", dlen+onelen);
  77. free(pbuf); pclose(fp);
  78. return -1;
  79. }
  80. pbuf = p;
  81. blen = dlen+onelen;
  82. }
  83. }
  84. nd->pbuf = pbuf;
  85. nd->dlen = dlen;
  86. nd->blen = blen;
  87. }
  88. pclose(fp);
  89. return 0;
  90. }
  91. ////////////////////////////////////////////////////////////////////
  92. int sys_reboot(void)
  93. {
  94. int r=0;
  95. r = system("reboot");
  96. if(r) {
  97. LOGE("sys_reboot failed!\n");
  98. return -1;
  99. }
  100. return r;
  101. }
  102. #include "upgrade.h"
  103. int sys_set_factory(void)
  104. {
  105. int r=0;
  106. #ifdef KEY_PARAS_IN_FILE
  107. r = paras_reset();
  108. #else
  109. char dir[100],src[200],dst[200];
  110. extern GlobalDeviceManager __globalDeviceManage;
  111. sqlite_deinit(__globalDeviceManage.db, &__globalDeviceManage.dbThreadlock);
  112. sys_get_path(dir, "");
  113. snprintf(src, sizeof(src), "%s/%s", dir, DATABASE_BAK_NAME);
  114. snprintf(dst, sizeof(dst), "%s/%s", dir, DATABASE_NAME);
  115. LOGD("___ file_copy from %s to %s\n", src, dst);
  116. r = file_copy(dst, src);
  117. if(r==0) {
  118. #if 0
  119. r = set_net(&DFLT_PARAS.dev._network_info, IP_V4);
  120. if(r) {
  121. LOGE("____ set_net failed\n");
  122. return -1;
  123. }
  124. #endif
  125. LOGD("____ exit smartPDU now ...\n");
  126. exit(0);
  127. }
  128. #endif
  129. return r;
  130. }
  131. int sys_get_path(char *path, char *name)
  132. {
  133. char tmp[500];
  134. pid_t pid = getpid();
  135. snprintf(tmp, sizeof(tmp), "/proc/%d/exe", pid);
  136. readlink(tmp, path, sizeof(tmp));
  137. char *p = strrchr(path, '/');
  138. if(p && strlen(path)>1) {
  139. *p = 0;
  140. if(p && strlen(name)>0) {
  141. strcat(path, "/");
  142. strcat(path, name);
  143. }
  144. }
  145. return 0;
  146. }
  147. ////////////////////////////////////////////////
  148. #include <net/if.h>
  149. #include <sys/socket.h>
  150. #include <net/route.h>
  151. #include <sys/ioctl.h>
  152. #include <netinet/in.h>
  153. #include <arpa/inet.h>
  154. #include <unistd.h>
  155. #include <net/if.h>
  156. #include <ifaddrs.h>
  157. #include <netinet/in.h>
  158. #include <linux/if.h>
  159. #include <linux/if_link.h>
  160. #include <linux/netlink.h>
  161. #include <linux/rtnetlink.h>
  162. void print_net(char*s, NetworkInfo_t *p)
  163. {
  164. LOGD("__%s__net.mode: %d\n", s, p->mode);
  165. LOGD("__%s__net.addr: %s\n", s, p->ip_address);
  166. LOGD("__%s__net.mask: %s\n", s, p->mask);
  167. LOGD("__%s__net.gateway: %s\n", s, p->gateway);
  168. LOGD("__%s__net.dns1: %s\n", s, p->dns1);
  169. LOGD("__%s__net.dns2: %s\n", s, p->dns2);
  170. }
  171. void print_modbus(char*s, ModbusInfo_t *p)
  172. {
  173. LOGD("__%s__mbus.mode: %d\n", s, p->product_modbus_type);
  174. LOGD("__%s__mbus.addr: %d\n", s, p->product_modbus_baud);
  175. LOGD("__%s__mbus.mask: %d\n", s, p->product_modbus_addr);
  176. LOGD("__%s__mbus.mask: %s\n", s, p->product_relay_ctrl_port);
  177. }
  178. static int set_ipaddr(int family, const char* iface, const char* addr, const char* netmask)
  179. {
  180. struct ifreq ifr;
  181. struct sockaddr_in *sin;
  182. struct sockaddr_in6 *sin6;
  183. int fd = socket(family, SOCK_DGRAM, 0);
  184. if (fd < 0) {
  185. LOGD("socket SOCK_DGRAM failed\n");
  186. return -1;
  187. }
  188. strcpy(ifr.ifr_name, iface);
  189. sin = (struct sockaddr_in*)&ifr.ifr_addr;
  190. sin6 = (struct sockaddr_in6*)&ifr.ifr_addr;
  191. if(family==AF_INET) {
  192. sin->sin_family = family;
  193. sin->sin_port = 0;
  194. inet_pton(family, addr, &sin->sin_addr);
  195. if (ioctl(fd, SIOCSIFADDR, &ifr) < 0) {
  196. LOGD("ioctl SIOCSIFADDR \n");
  197. close(fd);
  198. return -1;
  199. }
  200. sin->sin_family = family;
  201. inet_pton(family, netmask, &sin->sin_addr);
  202. if (ioctl(fd, SIOCSIFNETMASK, &ifr) < 0) {
  203. LOGD("ioctl SIOCSIFNETMASK \n");
  204. close(fd);
  205. return -1;
  206. }
  207. }
  208. else {
  209. sin6->sin6_family = family;
  210. sin6->sin6_port = 0;
  211. inet_pton(family, addr, &sin6->sin6_addr);
  212. if (ioctl(fd, SIOCSIFADDR, &ifr) < 0) {
  213. LOGD("ioctl SIOCSIFADDR \n");
  214. close(fd);
  215. return -1;
  216. }
  217. sin6->sin6_family = family;
  218. inet_pton(family, netmask, &sin6->sin6_addr);
  219. if (ioctl(fd, SIOCSIFNETMASK, &ifr) < 0) {
  220. LOGD("ioctl SIOCSIFNETMASK \n");
  221. close(fd);
  222. return -1;
  223. }
  224. }
  225. close(fd);
  226. return 0;
  227. }
  228. static int get_ipaddr(int family, char *iface, char *addr, char *netmask)
  229. {
  230. struct ifaddrs *ifap, *ifa;
  231. if (getifaddrs(&ifap) == -1) {
  232. LOGE("___getifaddrs() error\n");
  233. return -1;
  234. }
  235. for (ifa = ifap; ifa != NULL; ifa = ifa->ifa_next) {
  236. if (ifa->ifa_addr != NULL && ifa->ifa_addr->sa_family == family) {
  237. if(family==AF_INET) {
  238. struct sockaddr_in *sa,*msk;
  239. sa = (struct sockaddr_in *) ifa->ifa_addr;
  240. msk = (struct sockaddr_in *) ifa->ifa_netmask;
  241. if(addr) {
  242. inet_ntop(family, &sa->sin_addr, addr, INET6_ADDRSTRLEN);
  243. printf("___addr: %s\n", addr);
  244. }
  245. if(netmask) {
  246. inet_ntop(family, &msk->sin_addr, netmask, INET6_ADDRSTRLEN);
  247. //printf("___netmask: %s\n", netmask);
  248. }
  249. }
  250. else {
  251. struct sockaddr_in6 *sa,*msk;
  252. sa = (struct sockaddr_in6 *) ifa->ifa_addr;
  253. msk = (struct sockaddr_in6 *) ifa->ifa_netmask;
  254. if(addr) {
  255. inet_ntop(family, &sa->sin6_addr, addr, INET6_ADDRSTRLEN);
  256. //printf("___addr6: %s\n", addr);
  257. }
  258. if(netmask) {
  259. inet_ntop(family, &msk->sin6_addr, netmask, INET6_ADDRSTRLEN);
  260. //printf("___netmask6: %s\n", netmask);
  261. }
  262. }
  263. }
  264. }
  265. freeifaddrs(ifap);
  266. return 0;
  267. }
  268. static int set_gateway(int family, const char* gateway)
  269. {
  270. struct rtentry rt;
  271. int fd;
  272. fd = socket(family, SOCK_DGRAM, 0);
  273. if (fd < 0) {
  274. LOGE("socket SOCK_DGRAM failed\n");
  275. return -1;
  276. }
  277. memset(&rt, 0, sizeof(rt));
  278. if(family==AF_INET) {
  279. struct sockaddr_in* sin;
  280. sin = (struct sockaddr_in*)&rt.rt_gateway;
  281. sin->sin_family = family;
  282. sin->sin_port = 0;
  283. inet_pton(family, gateway, &sin->sin_addr);
  284. sin = (struct sockaddr_in*)&rt.rt_dst;
  285. sin->sin_family = family;
  286. sin->sin_addr.s_addr = 0;
  287. }
  288. else {
  289. struct sockaddr_in6* sin;
  290. sin = (struct sockaddr_in6*)&rt.rt_gateway;
  291. sin->sin6_family = family;
  292. sin->sin6_port = 0;
  293. inet_pton(family, gateway, &sin->sin6_addr);
  294. sin = (struct sockaddr_in6*)&rt.rt_dst;
  295. sin->sin6_family = family;
  296. //sin->sin6_addr.s_addr[0] = 0;
  297. }
  298. rt.rt_flags = RTF_GATEWAY;
  299. if (ioctl(fd, SIOCADDRT, &rt) < 0) {
  300. LOGE("ioctl SIOCADDRT failed\n");
  301. close(fd);
  302. return -1;
  303. }
  304. close(fd);
  305. return 0;
  306. }
  307. static int get_gateway(int family, const char* gateway)
  308. {
  309. #if 0
  310. int sockfd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
  311. if (sockfd == -1) {
  312. log_e("set_gateway, socket AF_NETLINK error\n");
  313. return -1;
  314. }
  315. struct sockaddr_nl sa;
  316. memset(&sa, 0, sizeof(sa));
  317. sa.nl_family = AF_NETLINK;
  318. if (bind(sockfd, (struct sockaddr *) &sa, sizeof(sa)) == -1) {
  319. log_e("set_gateway, bind error\n");
  320. return -1;
  321. }
  322. struct nlreq req;
  323. memset(&req, 0, sizeof(req));
  324. req.hdr.nlmsg_len = NLMSG_LENGTH(sizeof(struct rtmsg));
  325. req.hdr.nlmsg_type = RTM_GETROUTE;
  326. req.hdr.nlmsg_flags = NLM_F_REQUEST | NLM_F_DUMP;
  327. req.msg.rtm_family = family;
  328. req.msg.rtm_table = RT_TABLE_MAIN;
  329. req.msg.rtm_protocol = RTPROT_UNSPEC;
  330. req.msg.rtm_scope = RT_SCOPE_UNIVERSE;
  331. req.msg.rtm_type = RTN_UNICAST;
  332. struct iovec iov;
  333. memset(&iov, 0, sizeof(iov));
  334. iov.iov_base = &req;
  335. iov.iov_len = req.hdr.nlmsg_len;
  336. struct msghdr msg;
  337. memset(&msg, 0, sizeof(msg));
  338. msg.msg_name = &sa;
  339. msg.msg_namelen = sizeof(sa);
  340. msg.msg_iov = &iov;
  341. msg.msg_iovlen = 1;
  342. char buf[BUFSIZE];
  343. memset(buf, 0, sizeof(buf));
  344. struct nlmsghdr *hdr;
  345. int len;
  346. if (sendmsg(sockfd, &msg, 0) == -1) {
  347. log_e("set_gateway, sendmsg error\n");
  348. return -1;
  349. }
  350. while ((len = recv(sockfd, buf, sizeof(buf), 0)) > 0) {
  351. for (hdr = (struct nlmsghdr *) buf; NLMSG_OK(hdr, len); hdr = NLMSG_NEXT(hdr, len)) {
  352. if (hdr->nlmsg_type == NLMSG_DONE) {
  353. goto finish;
  354. }
  355. if (hdr->nlmsg_type == NLMSG_ERROR) {
  356. log_e("set_gateway, NLMSG_ERROR\n");
  357. return -1;
  358. }
  359. struct rtmsg *rt = (struct rtmsg *) NLMSG_DATA(hdr);
  360. if (rt->rtm_family != family || rt->rtm_table != RT_TABLE_MAIN || rt->rtm_type != RTN_UNICAST) {
  361. continue;
  362. }
  363. struct rtattr *attr;
  364. int attrlen;
  365. for (attr = (struct rtattr *) RTM_RTA(rt), attrlen = RTM_PAYLOAD(hdr); RTA_OK(attr, attrlen); attr = RTA_NEXT(attr, attrlen)) {
  366. if (attr->rta_type == RTA_GATEWAY) {
  367. char gw_addr[INET_ADDRSTRLEN];
  368. struct in_addr addr;
  369. memcpy(&addr, RTA_DATA(attr), sizeof(addr));
  370. if (inet_ntop(family, &addr, gateway, INET6_ADDRSTRLEN) == NULL) {
  371. log_e("set_gateway, inet_ntop error\n");
  372. continue;
  373. }
  374. printf("Gateway address: %s\n", gw_addr);
  375. log_i("set_gateway, Gateway address: %s\n");
  376. goto finish;
  377. }
  378. }
  379. }
  380. }
  381. finish:
  382. close(sockfd);
  383. #endif
  384. return 0;
  385. }
  386. static int set_dhcp(int on)
  387. {
  388. if(on) {
  389. system("killall udhcpc && udhcpc –i eth0 –s /usr/share/udhcpc/default.script &");
  390. }
  391. else {
  392. system("killall udhcpc");
  393. }
  394. return 0;
  395. }
  396. static int is_number(char *str)
  397. {
  398. while (*str) {
  399. if (!isdigit(*str)) {
  400. return 0;
  401. }
  402. str++;
  403. }
  404. return 1;
  405. }
  406. static int ip_check(char* addr)
  407. {
  408. int flag=IP_ERR;
  409. struct in_addr addr4;
  410. struct in6_addr addr6;
  411. if (inet_pton(AF_INET, addr, &addr4) == 1) {
  412. flag = IP_V4;
  413. } else if (inet_pton(AF_INET6, addr, &addr6) == 1) {
  414. flag = IP_V6;
  415. }
  416. if(flag==IP_ERR) {
  417. LOGE("____ wrong ip addr: %s\n", addr);
  418. }
  419. return flag;
  420. }
  421. static int is_ipv6_prefix(char *ss)
  422. {
  423. if(is_number(ss)) {
  424. int val = atoi(ss);
  425. if(val>1 && val<128) {
  426. return 1;
  427. }
  428. }
  429. return 0;
  430. }
  431. ///////////////////////////////////////////////
  432. typedef struct {
  433. char lo[1000];
  434. char e0[1000];
  435. char e1[1000];
  436. }interface_t;
  437. typedef uint32_t u32;
  438. #define XLEN(a,b) ((u32)(a)-(u32)(b))
  439. typedef struct {
  440. char *ptr;
  441. int len;
  442. }nic_t;
  443. static int get_nic(char *nbuf, int nlen, char *nic, char *fbuf, int flen)
  444. {
  445. int r,xlen;
  446. char tmp[20];
  447. char *p1,*p2;
  448. char *fend=fbuf+flen-1;
  449. sprintf(tmp, "auto %s", nic);
  450. p1 = strstr(fbuf, tmp);
  451. if(!p1) {
  452. return -1;
  453. }
  454. p2 = strstr(p1+1, "auto");
  455. if(!p2 || p2>fend) {
  456. p2 = fend;
  457. }
  458. xlen = XLEN(p2,p1);
  459. memcpy(nbuf, p1, xlen);
  460. nbuf[xlen] = 0;
  461. return 0;
  462. }
  463. static int get_ip(char *nbuf, char *token, char *ipaddr, int ipv)
  464. {
  465. int cnt=0;
  466. char *p,*p1,*p2;
  467. char *pbuf=NULL;
  468. if(ipv==IP_V4) {
  469. pbuf = strstr(nbuf, "inet");
  470. }
  471. else {
  472. pbuf = strstr(nbuf, "inet6");
  473. }
  474. if(pbuf==NULL) {
  475. LOGE("___get_ip, not found inet or inet6\n");
  476. return -1;
  477. }
  478. p = strstr(pbuf, token);
  479. if(p) {
  480. p += strlen(token);
  481. while(isspace(*p)) p++;
  482. p1 = p;
  483. while(!isspace(*p)) p++;
  484. p2 = p;
  485. for(p=p1; p<p2; p++) {
  486. if(isgraph(*p)) {
  487. ipaddr[cnt++] = *p;
  488. }
  489. }
  490. ipaddr[cnt] = 0;
  491. return 0;
  492. }
  493. return -1;
  494. }
  495. static int get_dns(char *nbuf, char *dns1, char *dns2)
  496. {
  497. int cnt;
  498. char *p1,*p2;
  499. char *token="server=";
  500. p1 = strstr(nbuf, token);
  501. if(p1==NULL) {
  502. dns1[0] = 0;
  503. return -1;
  504. }
  505. p1 += strlen(token); cnt=0;
  506. while(!isspace(*p1)) {
  507. dns1[cnt++] = *p1;
  508. p1++;
  509. }
  510. dns1[cnt] = 0;
  511. p2 = strstr(p1, token);
  512. if(p2==NULL) {
  513. dns2[0] = 0;
  514. return -1;
  515. }
  516. p2 += strlen(token); cnt=0;
  517. while(!isspace(*p2)) {
  518. dns2[cnt++] = *p2;
  519. p2++;
  520. }
  521. dns2[cnt] = 0;
  522. return 0;
  523. }
  524. static int set_dns(NetworkInfo_t *para)
  525. {
  526. int r;
  527. char tmp[100];
  528. char buf[500];
  529. LOGD("_____ set_dns\n");
  530. int chk1=ip_check(para->dns1);
  531. int chk2=ip_check(para->dns2);
  532. if( chk1==IP_ERR && chk2==IP_ERR) {
  533. LOGE("___ dns1 and dns2 are all wrong!\n");
  534. return -1;
  535. }
  536. memset(buf, 0, sizeof(buf));
  537. strcpy(buf, "interface=eth0\n");
  538. if(chk1) {
  539. sprintf(tmp, "server=%s\n", para->dns1);
  540. strcat(buf, tmp);
  541. }
  542. if(chk2) {
  543. sprintf(tmp, "server=%s\n", para->dns2);
  544. strcat(buf, tmp);
  545. }
  546. r = file_save(FILE_DNS, buf, strlen(buf))+1;
  547. if(r==0) {
  548. system("killall dnsmasq; /usr/sbin/dnsmasq &");
  549. }
  550. return r;
  551. }
  552. static int get_net(NetworkInfo_t *para, int ipv)
  553. {
  554. int r;
  555. char *p=NULL;
  556. interface_t intf;
  557. buf_t fbuf;
  558. r = file_load(FILE_NET, &fbuf);
  559. if(r) {
  560. return -1;
  561. }
  562. memset(&intf, 0, sizeof(intf));
  563. get_nic(intf.lo, sizeof(intf.lo), "lo", fbuf.buf, fbuf.dlen);
  564. get_nic(intf.e0, sizeof(intf.e0), "eth0", fbuf.buf, fbuf.dlen);
  565. get_nic(intf.e1, sizeof(intf.e1), "eth1", fbuf.buf, fbuf.dlen);
  566. p = strstr(intf.e0, (ipv==IP_V4)?"inet":"inet6");
  567. if(p==NULL) {
  568. LOGE("___get_ipv%d, %s not found!\n", (ipv==IP_V4)?4:6, (ipv==IP_V4)?"inet":"inet6");
  569. free(fbuf.buf); return -1;
  570. }
  571. if(strstr(p, "dhcp")) {
  572. para->mode = 0;
  573. }
  574. else if(strstr(p, "static")) {
  575. para->mode = 1;
  576. } else {
  577. LOGE("___get_net, dhcp/static not found!\n");
  578. free(fbuf.buf); return -1;
  579. }
  580. get_ip(p, "address", para->ip_address, ipv);
  581. get_ip(p, "netmask", para->mask, ipv);
  582. get_ip(p, "gateway", para->gateway, ipv);
  583. free(fbuf.buf);
  584. r = file_load(FILE_DNS, &fbuf);
  585. if(r) {
  586. return -1;
  587. }
  588. get_dns(fbuf.buf, para->dns1, para->dns2);
  589. return 0;
  590. }
  591. static int nw_check(NetworkInfo_t *para, int ipv)
  592. {
  593. if(ipv==IP_V4) {
  594. if((ip_check(para->ip_address)!=IP_V4) || (ip_check(para->mask)!=IP_V4) || (ip_check(para->gateway)!=IP_V4)) {
  595. return -1;
  596. }
  597. }
  598. else {
  599. if((ip_check(para->ip_address)!=IP_V6) || (!is_ipv6_prefix(para->mask)) || (ip_check(para->gateway)!=IP_V6)) {
  600. return -1;
  601. }
  602. }
  603. return 0;
  604. }
  605. static int set_net(NetworkInfo_t *para, int ipv)
  606. {
  607. int r,ipv2;
  608. char *p1,*p2,*p3,*p4;
  609. int flen=100*1000;
  610. NetworkInfo_t info4,info6,*info;
  611. get_net(&info4, IP_V4);
  612. get_net(&info6, IP_V6);
  613. info = ((ipv==IP_V4)?&info4:&info6);
  614. if(para->mode<2 && para->mode!=info->mode) {
  615. info->mode = para->mode;
  616. }
  617. ipv2 = ip_check(para->ip_address);
  618. if((ipv==ipv2) && strcmp(para->ip_address, info->ip_address)) {
  619. strcpy(info->ip_address, para->ip_address);
  620. }
  621. if(ipv==IP_V4) {
  622. ipv2 = ip_check(para->mask);
  623. }
  624. else {
  625. ipv2=is_ipv6_prefix(para->mask)?IP_V6:IP_ERR;
  626. }
  627. if((ipv==ipv2) && strcmp(para->mask, info->mask)) {
  628. strcpy(info->mask, para->mask);
  629. }
  630. ipv2 = ip_check(para->gateway);
  631. if((ipv==ipv2) && strcmp(para->gateway, info->gateway)) {
  632. strcpy(info->gateway, para->gateway);
  633. }
  634. char *fbuf=malloc(flen);
  635. if(!fbuf) {
  636. LOGE("___ malloc %d failed\n", flen);
  637. return -1;
  638. }
  639. #if 0
  640. file_load(FILE_NET, fbuf);
  641. get_nic(pNet->lo, sizeof(pNet->lo), "lo", fbuf, flen);
  642. get_nic(pNet->e1, sizeof(pNet->e1), "eth1", fbuf, flen);
  643. snprintf(pNet->e0, sizeof(pNet->e0), "auto eth0\niface eth0 inet %s\n address %s\n netmask %s\n gateway %s\n\n",
  644. info4.mode?"static":"dhcp", info4.ip_address, info4.mask, info4.gateway);
  645. strcpy(fbuf, pNet->lo);
  646. strcat(fbuf, pNet->e0);
  647. strcat(fbuf, pNet->e1);
  648. #else
  649. char *lo="auto lo\niface lo inet loopback\n\n";
  650. char *eth1="auto eth1\niface eth1 inet dhcp\npre-up /etc/network/nfs_check\nwait-delay 5\nhostname $(hostname)\n";
  651. char eth0[1000];
  652. int chk4=nw_check(&info4, IP_V4);
  653. int chk6=nw_check(&info6, IP_V6);
  654. char temp[400];
  655. strcpy(eth0, "auto eth0\n");
  656. if(chk4==0) {
  657. sprintf(temp, "iface eth0 inet %s\n address %s\n netmask %s\n gateway %s\n\n", (info4.mode==IP_STATIC)?"static":"dhcp", info4.ip_address, info4.mask, info4.gateway);
  658. strcat(eth0, temp);
  659. }
  660. if(chk6==0) {
  661. sprintf(temp, "iface eth0 inet6 %s\n address %s\n netmask %s\n gateway %s\n\n", (info6.mode==IP_STATIC)?"static":"dhcp", info6.ip_address, info6.mask, info6.gateway);
  662. strcat(eth0, temp);
  663. }
  664. sprintf(fbuf, "%s%s%s", lo, eth0, eth1);
  665. #endif
  666. //LOGD("___interfaces___:\n%s", fbuf);
  667. r = file_save(FILE_NET, fbuf, strlen(fbuf)); free(fbuf);
  668. return r;
  669. }
  670. //////////////////////////////////////////////////////////////////////////////
  671. int sys_set_net(NetworkInfo_t *para, int ipVersion)
  672. {
  673. int r,ipv;
  674. #ifdef USE_BASH
  675. char cmd[200];
  676. snprintf(cmd, sizeof(cmd), "ip addr flush dev %s", DFLT_NIC);
  677. r = system(cmd);
  678. if(r) {
  679. LOGE("set ipaddr failed\n");
  680. return -1;
  681. }
  682. snprintf(cmd, sizeof(cmd), "ip addr add %s/%s dev %s", para->ip_address, para->mask, DFLT_NIC);
  683. r = system(cmd);
  684. if(r) {
  685. LOGE("set netmask %s failed\n", para->mask);
  686. return -1;
  687. }
  688. snprintf(cmd, sizeof(cmd), "ip route add default via %s dev %s", para->gateway, DFLT_NIC);
  689. r = system(cmd);
  690. if(r) {
  691. LOGE("set gateway %s failed\n", para->gateway);
  692. return -1;
  693. }
  694. snprintf(cmd, sizeof(cmd), "ip link set dev %s up", DFLT_NIC);
  695. r = system(cmd);
  696. if(r) {
  697. LOGE("restart %s failed\n", DFLT_NIC);
  698. return -1;
  699. }
  700. #elif defined USE_FILE
  701. NetworkInfo_t para2;
  702. r = get_net(&para2, ipVersion);
  703. if(r==0) {
  704. if(memcmp(para, &para2, sizeof(NetworkInfo_t))==0) {
  705. LOGW("%s NetworkInfo is same, no need to set\n", DFLT_NIC);
  706. return -1;
  707. }
  708. set_dns(para);
  709. r = set_net(para, ipVersion);
  710. if(r==0) {
  711. //system("ip addr flush dev eth0");
  712. //sprintf(eth0, "ip link set dev eth0 %s/%d && ip link set eth0 up && ip link set eth0 down", info->ip_address, info->netmask);
  713. //system(eth0);
  714. char buff[256]={0};
  715. sprintf(buff,"%s %s %s %s %s","ifconfig eth0",para->ip_address,"netmask",para->mask,"up");
  716. system(buff);
  717. memset(buff,0,sizeof(buff));
  718. sprintf(buff,"%s %s","route add default gw ",para->gateway);
  719. system(buff);
  720. //system("ifconfig eth0 down; ifconfig eth0 up");
  721. exit(0);
  722. }
  723. }
  724. else {
  725. LOGE("get_net failed\n");
  726. }
  727. #elif defined USE_API
  728. if(para->mode==IP_DHCP) {
  729. r = set_dhcp(1);
  730. }
  731. else {
  732. set_dhcp(0);
  733. int family=(ipv==IP_V4)?AF_INET:AF_INET6;
  734. r = set_ipaddr(family, "eth0", para->ip_address, para->mask);
  735. if(r) {
  736. LOGE("set_ipaddr failed\n");
  737. return r;
  738. }
  739. r = set_gateway(family, para->gateway);
  740. if(r) {
  741. LOGE("set_gateway failed\n");
  742. return r;
  743. }
  744. r = set_dns(family, "eth0", para->dns1, para->dns2);
  745. if(r) {
  746. LOGE("set_dns failed\n");
  747. return r;
  748. }
  749. }
  750. #endif
  751. return r;
  752. }
  753. int sys_get_net(NetworkInfo_t *para, int ipVersion)
  754. {
  755. int r;
  756. int family=AF_INET;
  757. #ifdef USE_BASH
  758. #elif defined USE_FILE
  759. r = get_net(para, ipVersion);
  760. #elif defined USE_API
  761. r = get_ipaddr(family, DFLT_NIC, para->ip_address, para->mask);
  762. if(r) {
  763. LOGE("set_ipaddr failed\n");
  764. return r;
  765. }
  766. r = get_gateway(family, para->gateway);
  767. if(r) {
  768. LOGE("set_gateway failed\n");
  769. return r;
  770. }
  771. r = get_dns(family, DFLT_NIC, para->dns1, para->dns2);
  772. if(r) {
  773. LOGE("set_dns failed\n");
  774. return r;
  775. }
  776. #endif
  777. return r;
  778. }
  779. #define MAX_BUFFER_SIZE 4096
  780. static void* plug_thread(void *arg)
  781. {
  782. int nlSock;
  783. struct sockaddr_nl srcAddr;
  784. struct nlmsghdr *nlMsg;
  785. char buffer[MAX_BUFFER_SIZE];
  786. int plug_in=0,plug_out=0;
  787. printf("___plug_thread start\n");
  788. nlSock = socket(PF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
  789. if (nlSock < 0) {
  790. printf("Failed to create Netlink socket");
  791. goto quit;
  792. }
  793. memset(&srcAddr, 0, sizeof(srcAddr));
  794. srcAddr.nl_family = AF_NETLINK;
  795. srcAddr.nl_groups = RTMGRP_LINK | RTMGRP_NOTIFY;
  796. if (bind(nlSock, (struct sockaddr *)&srcAddr, sizeof(srcAddr)) < 0) {
  797. perror("Failed to bind Netlink socket");
  798. close(nlSock);
  799. goto quit;
  800. }
  801. int flags = fcntl(nlSock, F_GETFL, 0);
  802. fcntl(nlSock, F_SETFL, flags | O_NONBLOCK);
  803. while (1) {
  804. fd_set readSet;
  805. FD_ZERO(&readSet);
  806. FD_SET(nlSock, &readSet);
  807. // 设置超时时间为 1 秒
  808. struct timeval timeout;
  809. timeout.tv_sec = 1;
  810. timeout.tv_usec = 0;
  811. // 监视 Netlink Socket 的可读状态
  812. int result = select(nlSock + 1, &readSet, NULL, NULL, &timeout);
  813. if (result == -1) {
  814. perror("Failed to select");
  815. goto quit;
  816. } else if (result == 0) {
  817. // 超时,没有新数据到达
  818. //printf("Timeout\n");
  819. continue;
  820. }
  821. if (FD_ISSET(nlSock, &readSet)) {
  822. // 接收消息
  823. int bytesRead = recv(nlSock, buffer, MAX_BUFFER_SIZE, 0);
  824. if (bytesRead < 0) {
  825. perror("Failed to receive message from Netlink socket");
  826. close(nlSock);
  827. goto quit;
  828. }
  829. // 遍历接收到的消息,获取到Netlink的消息头。
  830. nlMsg = (struct nlmsghdr *)buffer;
  831. while (NLMSG_OK(nlMsg, bytesRead)) {
  832. if (nlMsg->nlmsg_type == NLMSG_DONE) {
  833. break;
  834. }
  835. if (nlMsg->nlmsg_type == RTM_NEWLINK || nlMsg->nlmsg_type == RTM_DELLINK) {
  836. //NLMSG_DATA()获取完整的ifinfomsg结构体的指针
  837. //根据消息头nlMsg获取到其后面的数据部分ifinfomsg结构体的指针
  838. struct ifinfomsg *ifInfo = (struct ifinfomsg *)NLMSG_DATA(nlMsg);
  839. if (ifInfo->ifi_family == AF_UNSPEC) {
  840. //获取路由信息的属性,以及消息负载的长度,通过这两个参数来判断本次的数据是否有效
  841. struct rtattr *attribute = IFLA_RTA(ifInfo);
  842. int attributeLen = IFLA_PAYLOAD(nlMsg);
  843. while (RTA_OK(attribute, attributeLen)) {
  844. char *ifaceName = (char *)RTA_DATA(attribute);
  845. if ((attribute->rta_type == IFLA_IFNAME) && (strcmp(ifaceName, "eth0")==0)) {
  846. //printf("____Interface: %s \n", ifaceName);
  847. if((ifInfo->ifi_flags & IFF_LOWER_UP)) {
  848. if(plug_in==0) {
  849. system("ifup eth0");
  850. printf("____eth0 plug in\n");
  851. plug_in = 1; plug_out = 0;
  852. }
  853. }
  854. else {
  855. if(plug_out==0) {
  856. system("ifdown eth0");
  857. printf("____eth0 plug out\n");
  858. plug_in = 0; plug_out = 1;
  859. }
  860. }
  861. }
  862. attribute = RTA_NEXT(attribute, attributeLen);
  863. }
  864. }
  865. }
  866. nlMsg = NLMSG_NEXT(nlMsg, bytesRead);
  867. }
  868. }
  869. }
  870. close(nlSock);
  871. quit:
  872. printf("___plug_thread exit\n");
  873. pthread_exit(NULL);
  874. }
  875. int sys_get_disk_ava_MB(char *path)
  876. {
  877. int r;
  878. struct statvfs st;
  879. r = statvfs(path, &st);
  880. if(r) {
  881. return -1;
  882. }
  883. int avaMB = (st.f_frsize*st.f_bavail)/(1024*1024);
  884. //int freeMB = (st.f_frsize*st.f_bfree)/(1024*1024);
  885. //printf("____ava: %dMB, free %dMB\n", avaMB, freeMB);
  886. LOGD("___ RomAvailable rom: %dMB\n", avaMB);
  887. return avaMB;
  888. }
  889. int sys_get_mem_ava_MB(void)
  890. {
  891. int r;
  892. long v1=0,v2=0;
  893. char buf[32];
  894. struct sysinfo info;
  895. #if 0
  896. r = sysinfo(&info);
  897. if(r) {
  898. return -1;
  899. }
  900. value = (info.freeram*info.mem_unit)/(1014*1024);
  901. #else
  902. FILE *fp = fopen("/proc/meminfo", "r");
  903. if(!fp) {
  904. return -1;
  905. }
  906. fscanf(fp, "MemTotal: %s kB\n", buf);
  907. fscanf(fp, "MemFree: %s kB\n", buf);
  908. v1 = atol(buf);
  909. fscanf(fp, "MemAvailable: %s kB\n", buf);
  910. v2 = atol(buf);
  911. fclose(fp);
  912. LOGD("___MemFree: %ldkB, MemAvailable: %ldkB\n", v1, v2);
  913. v2 /= 1024;
  914. #endif
  915. return v2;
  916. }
  917. #ifdef RAM_DEBUG
  918. #define RAM_THRD 50
  919. #define RAM_PERIOD 1
  920. #else
  921. #define RAM_THRD 20
  922. #define RAM_PERIOD 60
  923. #endif
  924. #ifdef ROM_DEBUG
  925. #define ROM_THRD 50
  926. #define ROM_PERIOD 1
  927. #else
  928. #define ROM_THRD 30
  929. #define ROM_PERIOD 1
  930. #endif
  931. static void mem_check(void)
  932. {
  933. int avaMB,flag=0;
  934. static uint32_t mem_cnt=0;
  935. mem_cnt++;
  936. if(mem_cnt%RAM_PERIOD==0) {
  937. avaMB = sys_get_mem_ava_MB();
  938. if(avaMB<RAM_THRD) {
  939. LOGE("___ free mem less than %dMB, reboot now ...\n", RAM_THRD);
  940. sys_reboot();
  941. }
  942. }
  943. if(mem_cnt%ROM_PERIOD==0) {
  944. avaMB = sys_get_disk_ava_MB("/mnt/UDISK/");
  945. if(avaMB<ROM_THRD) {
  946. flag = 1;
  947. LOGE("___ free 2222 less than MB, reboot now ...\n");
  948. }
  949. dev_sql_set_overwrite(flag);
  950. }
  951. }
  952. static void* polling_thread(void *arg)
  953. {
  954. thread_handle_t *h=(thread_handle_t*)arg;
  955. while(h->quit==0) {
  956. mem_check();
  957. sleep(1);
  958. }
  959. pthread_exit(NULL);
  960. }
  961. int sys_init(void)
  962. {
  963. //userShellInit();
  964. initLogger();
  965. lock_s_init();
  966. //thread_start(THREAD_ID_PLUG, plug_thread, NULL, 4*MB, 0);
  967. thread_start(THREAD_ID_POLLING, polling_thread, NULL, 4*MB, 0);
  968. return 0;
  969. }