sys.c 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218
  1. #include <sys/types.h>
  2. #include <sys/stat.h>
  3. #include <sys/sysinfo.h>
  4. #include <sys/statvfs.h>
  5. #include <fcntl.h>
  6. #include <unistd.h>
  7. #include <string.h>
  8. #include <limits.h>
  9. #include "elog.h"
  10. #include "sys.h"
  11. #include "upgrade.h"
  12. #include "paras.h"
  13. #include "cfg.h"
  14. #include "file.h"
  15. #include "lock.h"
  16. #include "thread.h"
  17. #include "shell.h"
  18. #include "common.h"
  19. #include "sqlite_handle.h"
  20. #if 1
  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. system("ifdown eth0; ifup eth0");
  715. exit(0);
  716. }
  717. }
  718. else {
  719. LOGE("get_net failed\n");
  720. }
  721. #elif defined USE_API
  722. if(para->mode==IP_DHCP) {
  723. r = set_dhcp(1);
  724. }
  725. else {
  726. set_dhcp(0);
  727. int family=(ipv==IP_V4)?AF_INET:AF_INET6;
  728. r = set_ipaddr(family, "eth0", para->ip_address, para->mask);
  729. if(r) {
  730. LOGE("set_ipaddr failed\n");
  731. return r;
  732. }
  733. r = set_gateway(family, para->gateway);
  734. if(r) {
  735. LOGE("set_gateway failed\n");
  736. return r;
  737. }
  738. r = set_dns(family, "eth0", para->dns1, para->dns2);
  739. if(r) {
  740. LOGE("set_dns failed\n");
  741. return r;
  742. }
  743. }
  744. #endif
  745. return r;
  746. }
  747. int sys_get_net(NetworkInfo_t *para, int ipVersion)
  748. {
  749. int r;
  750. int family=AF_INET;
  751. #ifdef USE_BASH
  752. #elif defined USE_FILE
  753. r = get_net(para, ipVersion);
  754. #elif defined USE_API
  755. r = get_ipaddr(family, DFLT_NIC, para->ip_address, para->mask);
  756. if(r) {
  757. LOGE("set_ipaddr failed\n");
  758. return r;
  759. }
  760. r = get_gateway(family, para->gateway);
  761. if(r) {
  762. LOGE("set_gateway failed\n");
  763. return r;
  764. }
  765. r = get_dns(family, DFLT_NIC, para->dns1, para->dns2);
  766. if(r) {
  767. LOGE("set_dns failed\n");
  768. return r;
  769. }
  770. #endif
  771. return r;
  772. }
  773. int sys_is_running(char *prog)
  774. {
  775. FILE *fp;
  776. int pid=0;
  777. char buf[20]={0};
  778. char command[200];
  779. snprintf(command, sizeof(command), "ps -ef | grep -v grep | grep -w -c %s", prog);
  780. fp = popen(command, "r");
  781. if (fp == NULL) {
  782. LOGE("execute %s failed: %s", command, strerror(errno));
  783. return 0;
  784. }
  785. if (fgets(buf, sizeof(buf), fp)) {
  786. pid = atoi(buf);
  787. }
  788. pclose(fp);
  789. return (pid)?1:0;
  790. }
  791. int sys_get_disk_ava_MB(char *path)
  792. {
  793. int r;
  794. struct statvfs st;
  795. r = statvfs(path, &st);
  796. if(r) {
  797. return -1;
  798. }
  799. int avaMB = (st.f_frsize*st.f_bavail)/(1024*1024);
  800. //int freeMB = (st.f_frsize*st.f_bfree)/(1024*1024);
  801. //printf("____ava: %dMB, free %dMB\n", avaMB, freeMB);
  802. return avaMB;
  803. }
  804. int sys_get_mem_ava_MB(void)
  805. {
  806. int r;
  807. long v1=0,v2;
  808. char buf[32];
  809. struct sysinfo info;
  810. #if 0
  811. r = sysinfo(&info);
  812. if(r) {
  813. return -1;
  814. }
  815. value = (info.freeram*info.mem_unit)/(1014*1024);
  816. #else
  817. FILE *fp = fopen("/proc/meminfo", "r");
  818. if(!fp) {
  819. return -1;
  820. }
  821. fscanf(fp, "MemTotal: %s kB\n", buf);
  822. fscanf(fp, "MemFree: %s kB\n", buf);
  823. v1 = atol(buf);
  824. fscanf(fp, "MemAvailable: %s kB\n", buf);
  825. v2 = atol(buf);
  826. fclose(fp);
  827. LOGD("___MemFree: %ldkB, MemAvailable: %ldkB\n", v1, v2);
  828. v2 /= 1024;
  829. #endif
  830. return v2;
  831. }
  832. int sys_get_mac(char *mac, int maclen)
  833. {
  834. int sock;
  835. struct ifreq ifr;
  836. if(!mac || maclen<6) {
  837. return -1;
  838. }
  839. sock = socket(AF_INET, SOCK_DGRAM, 0);
  840. if (sock == -1){
  841. return -1;
  842. }
  843. strcpy(ifr.ifr_name, "eth0");
  844. if (ioctl(sock, SIOCGIFHWADDR, &ifr) < 0){
  845. close(sock); return -1;
  846. }
  847. close(sock);
  848. uint8_t *p = (uint8_t*)ifr.ifr_hwaddr.sa_data;
  849. snprintf(mac, maclen, "%02x%02x%02x%02x%02x%02x", p[0], p[1], p[2], p[3], p[4], p[5]);
  850. return 0;
  851. }
  852. int sys_get_chip_id(int *id)
  853. {
  854. int sock;
  855. struct ifreq ifr;
  856. if(!id) {
  857. return -1;
  858. }
  859. sock = socket(AF_INET, SOCK_DGRAM, 0);
  860. if (sock == -1){
  861. return -1;
  862. }
  863. strcpy(ifr.ifr_name, "eth0");
  864. if (ioctl(sock, SIOCGIFHWADDR, &ifr) < 0){
  865. close(sock); return -1;
  866. }
  867. close(sock);
  868. *id = *((int*)ifr.ifr_hwaddr.sa_data);
  869. return 0;
  870. }
  871. ///////////////////////////////////////////////////////////////////////
  872. #define MAX_BUFFER_SIZE 4096
  873. static void* plug_thread(void *arg)
  874. {
  875. int nlSock;
  876. struct sockaddr_nl srcAddr;
  877. struct nlmsghdr *nlMsg;
  878. char buffer[MAX_BUFFER_SIZE];
  879. int plug_in=0,plug_out=0;
  880. thread_handle_t *h=(thread_handle_t*)arg;
  881. printf("___plug_thread start\n");
  882. nlSock = socket(PF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
  883. if (nlSock < 0) {
  884. printf("Failed to create Netlink socket");
  885. goto quit;
  886. }
  887. memset(&srcAddr, 0, sizeof(srcAddr));
  888. srcAddr.nl_family = AF_NETLINK;
  889. srcAddr.nl_groups = RTMGRP_LINK | RTMGRP_NOTIFY;
  890. if (bind(nlSock, (struct sockaddr *)&srcAddr, sizeof(srcAddr)) < 0) {
  891. perror("Failed to bind Netlink socket");
  892. close(nlSock);
  893. goto quit;
  894. }
  895. int flags = fcntl(nlSock, F_GETFL, 0);
  896. fcntl(nlSock, F_SETFL, flags | O_NONBLOCK);
  897. while (h->quit==0) {
  898. fd_set readSet;
  899. FD_ZERO(&readSet);
  900. FD_SET(nlSock, &readSet);
  901. // 设置超时时间为 1 秒
  902. struct timeval timeout;
  903. timeout.tv_sec = 1;
  904. timeout.tv_usec = 0;
  905. // 监视 Netlink Socket 的可读状态
  906. int result = select(nlSock + 1, &readSet, NULL, NULL, &timeout);
  907. if (result == -1) {
  908. perror("Failed to select");
  909. goto quit;
  910. } else if (result == 0) {
  911. // 超时,没有新数据到达
  912. //printf("Timeout\n");
  913. continue;
  914. }
  915. if (FD_ISSET(nlSock, &readSet)) {
  916. // 接收消息
  917. int bytesRead = recv(nlSock, buffer, MAX_BUFFER_SIZE, 0);
  918. if (bytesRead < 0) {
  919. perror("Failed to receive message from Netlink socket");
  920. close(nlSock);
  921. goto quit;
  922. }
  923. // 遍历接收到的消息,获取到Netlink的消息头。
  924. nlMsg = (struct nlmsghdr *)buffer;
  925. while (NLMSG_OK(nlMsg, bytesRead)) {
  926. if (nlMsg->nlmsg_type == NLMSG_DONE) {
  927. break;
  928. }
  929. if (nlMsg->nlmsg_type == RTM_NEWLINK || nlMsg->nlmsg_type == RTM_DELLINK) {
  930. //NLMSG_DATA()获取完整的ifinfomsg结构体的指针
  931. //根据消息头nlMsg获取到其后面的数据部分ifinfomsg结构体的指针
  932. struct ifinfomsg *ifInfo = (struct ifinfomsg *)NLMSG_DATA(nlMsg);
  933. if (ifInfo->ifi_family == AF_UNSPEC) {
  934. //获取路由信息的属性,以及消息负载的长度,通过这两个参数来判断本次的数据是否有效
  935. struct rtattr *attribute = IFLA_RTA(ifInfo);
  936. int attributeLen = IFLA_PAYLOAD(nlMsg);
  937. while (RTA_OK(attribute, attributeLen)) {
  938. char *ifaceName = (char *)RTA_DATA(attribute);
  939. if ((attribute->rta_type == IFLA_IFNAME) && (strcmp(ifaceName, "eth0")==0)) {
  940. //printf("____Interface: %s \n", ifaceName);
  941. if((ifInfo->ifi_flags & IFF_LOWER_UP)) {
  942. if(plug_in==0) {
  943. system("ifup eth0");
  944. printf("____eth0 plug in\n");
  945. plug_in = 1; plug_out = 0;
  946. }
  947. }
  948. else {
  949. if(plug_out==0) {
  950. system("ifdown eth0");
  951. printf("____eth0 plug out\n");
  952. plug_in = 0; plug_out = 1;
  953. }
  954. }
  955. }
  956. attribute = RTA_NEXT(attribute, attributeLen);
  957. }
  958. }
  959. }
  960. nlMsg = NLMSG_NEXT(nlMsg, bytesRead);
  961. }
  962. }
  963. }
  964. close(nlSock);
  965. quit:
  966. printf("___plug_thread exit\n");
  967. pthread_exit(NULL);
  968. }
  969. ///////////////////////////////////////////////////////////
  970. //#define RAM_DEBUG
  971. //#define ROM_DEBUG
  972. #ifdef RAM_DEBUG
  973. #define RAM_THRD 50
  974. #define RAM_PERIOD 1
  975. #else
  976. #define RAM_THRD 50
  977. #define RAM_PERIOD 60
  978. #endif
  979. #ifdef ROM_DEBUG
  980. #define ROM_THRD 50
  981. #define ROM_PERIOD 1
  982. #else
  983. #define ROM_THRD 50
  984. #define ROM_PERIOD 1
  985. #endif
  986. static void mem_check(void)
  987. {
  988. int avaMB,flag=0;
  989. static uint32_t mem_cnt=0;
  990. mem_cnt++;
  991. if(mem_cnt%RAM_PERIOD==0) {
  992. avaMB = sys_get_mem_ava_MB();
  993. if(avaMB<RAM_THRD) {
  994. LOGE("___ free mem less than %dMB, reboot now ...\n", RAM_THRD);
  995. sys_reboot();
  996. }
  997. }
  998. if(mem_cnt%ROM_PERIOD==0) {
  999. avaMB = sys_get_disk_ava_MB("/");
  1000. LOGD("___ available rom: %dMB, threshold: %dMB\n", avaMB, ROM_THRD);
  1001. if(avaMB<ROM_THRD) {
  1002. flag = 1;
  1003. }
  1004. dev_sql_set_overwrite(flag);
  1005. }
  1006. }
  1007. static void* polling_thread(void *arg)
  1008. {
  1009. thread_handle_t *h=(thread_handle_t*)arg;
  1010. while(h->quit==0) {
  1011. mem_check();
  1012. sleep(1);
  1013. }
  1014. pthread_exit(NULL);
  1015. }
  1016. int sys_init(void)
  1017. {
  1018. //userShellInit();
  1019. initLogger();
  1020. lock_s_init();
  1021. thread_start(THREAD_ID_POLLING, polling_thread, NULL, 4*MB, 0);
  1022. return 0;
  1023. }