| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211 |
- #include <sys/types.h>
- #include <sys/stat.h>
- #include <sys/sysinfo.h>
- #include <sys/statvfs.h>
- #include <fcntl.h>
- #include <unistd.h>
- #include <string.h>
- #include <limits.h>
- #include "elog.h"
- #include "sys.h"
- #include "upgrade.h"
- #include "paras.h"
- #include "cfg.h"
- #include "file.h"
- #include "lock.h"
- #include "thread.h"
- #include "shell.h"
- #include "common.h"
- #include "sqlite_handle.h"
- #if 0
- #define LOGE log_e
- #define LOGW log_w
- #define LOGD log_d
- #else
- #define LOGD printf
- #define LOGW printf
- #define LOGE printf
- #endif
- //#define USE_BASH
- #define USE_FILE
- //#define USE_API
- #define DFLT_NIC "eth0"
- //#define FILE_NET "/root/interfaces"
- #define FILE_NET "/etc/network/interfaces"
- #define FILE_DNS "/etc/dnsmasq.conf"
- typedef struct {
- char *pbuf;
- int dlen;
- int blen;
- }node_t;
- static int set_net(NetworkInfo_t *para, int ipv);
- static int node_free(node_t *nd)
- {
- if(!nd) {
- return -1;
- }
- free(nd->pbuf);
- nd->pbuf = NULL;
- nd->dlen = 0;
- nd->blen = 0;
- return 0;
- }
- static int exe_cmd(char *cmd, node_t *nd)
- {
- int r=0;
- FILE *fp=NULL;
- char *pbuf=NULL;
- int onelen=1000;
- int dlen=0,blen=onelen;
-
- fp = popen(cmd, "r");
- if (fp == NULL) {
- return -1;
- }
- if(nd) {
- pbuf = malloc(onelen);
- while(pbuf) {
- r = fread(pbuf+dlen, sizeof(char), onelen, fp);
- if(r<=0) {
- LOGE("exe_cmd fread failed\n");
- break;
- }
- dlen += r;
- if(dlen+onelen>blen) {
- char* p = realloc(pbuf, dlen+onelen);
- if(!p) {
- LOGE("exe_cmd relloac %d failed\n", dlen+onelen);
- free(pbuf); pclose(fp);
- return -1;
- }
- pbuf = p;
- blen = dlen+onelen;
- }
- }
-
- nd->pbuf = pbuf;
- nd->dlen = dlen;
- nd->blen = blen;
- }
- pclose(fp);
-
- return 0;
- }
- ////////////////////////////////////////////////////////////////////
- int sys_reboot(void)
- {
- int r=0;
- r = system("reboot");
- if(r) {
- LOGE("sys_reboot failed!\n");
- return -1;
- }
- return r;
- }
- #include "upgrade.h"
- int sys_set_factory(void)
- {
- int r=0;
- #ifdef KEY_PARAS_IN_FILE
- r = paras_reset();
- #else
- char dir[100],src[200],dst[200];
- extern GlobalDeviceManager __globalDeviceManage;
- sqlite_deinit(__globalDeviceManage.db, &__globalDeviceManage.dbThreadlock);
- sys_get_path(dir, "");
- snprintf(src, sizeof(src), "%s/%s", dir, DATABASE_BAK_NAME);
- snprintf(dst, sizeof(dst), "%s/%s", dir, DATABASE_NAME);
- LOGD("___ file_copy from %s to %s\n", src, dst);
- r = file_copy(dst, src);
- if(r==0) {
- #if 0
- r = set_net(&DFLT_PARAS.dev._network_info, IP_V4);
- if(r) {
- LOGE("____ set_net failed\n");
- return -1;
- }
- #endif
- LOGD("____ exit smartPDU now ...\n");
- exit(0);
- }
- #endif
- return r;
- }
- int sys_get_path(char *path, char *name)
- {
- char tmp[500];
- pid_t pid = getpid();
- snprintf(tmp, sizeof(tmp), "/proc/%d/exe", pid);
- readlink(tmp, path, sizeof(tmp));
- char *p = strrchr(path, '/');
- if(p && strlen(path)>1) {
- *p = 0;
- if(p && strlen(name)>0) {
- strcat(path, "/");
- strcat(path, name);
- }
- }
-
- return 0;
- }
- ////////////////////////////////////////////////
- #include <net/if.h>
- #include <sys/socket.h>
- #include <net/route.h>
- #include <sys/ioctl.h>
- #include <netinet/in.h>
- #include <arpa/inet.h>
- #include <unistd.h>
- #include <net/if.h>
- #include <ifaddrs.h>
- #include <netinet/in.h>
- #include <linux/if.h>
- #include <linux/if_link.h>
- #include <linux/netlink.h>
- #include <linux/rtnetlink.h>
- void print_net(char*s, NetworkInfo_t *p)
- {
- LOGD("__%s__net.mode: %d\n", s, p->mode);
- LOGD("__%s__net.addr: %s\n", s, p->ip_address);
- LOGD("__%s__net.mask: %s\n", s, p->mask);
- LOGD("__%s__net.gateway: %s\n", s, p->gateway);
- LOGD("__%s__net.dns1: %s\n", s, p->dns1);
- LOGD("__%s__net.dns2: %s\n", s, p->dns2);
- }
- void print_modbus(char*s, ModbusInfo_t *p)
- {
- LOGD("__%s__mbus.mode: %d\n", s, p->product_modbus_type);
- LOGD("__%s__mbus.addr: %d\n", s, p->product_modbus_baud);
- LOGD("__%s__mbus.mask: %d\n", s, p->product_modbus_addr);
- LOGD("__%s__mbus.mask: %s\n", s, p->product_relay_ctrl_port);
- }
- static int set_ipaddr(int family, const char* iface, const char* addr, const char* netmask)
- {
- struct ifreq ifr;
- struct sockaddr_in *sin;
- struct sockaddr_in6 *sin6;
-
- int fd = socket(family, SOCK_DGRAM, 0);
- if (fd < 0) {
- LOGD("socket SOCK_DGRAM failed\n");
- return -1;
- }
-
- strcpy(ifr.ifr_name, iface);
- sin = (struct sockaddr_in*)&ifr.ifr_addr;
- sin6 = (struct sockaddr_in6*)&ifr.ifr_addr;
- if(family==AF_INET) {
- sin->sin_family = family;
- sin->sin_port = 0;
- inet_pton(family, addr, &sin->sin_addr);
-
- if (ioctl(fd, SIOCSIFADDR, &ifr) < 0) {
- LOGD("ioctl SIOCSIFADDR \n");
- close(fd);
- return -1;
- }
-
- sin->sin_family = family;
- inet_pton(family, netmask, &sin->sin_addr);
- if (ioctl(fd, SIOCSIFNETMASK, &ifr) < 0) {
- LOGD("ioctl SIOCSIFNETMASK \n");
- close(fd);
- return -1;
- }
- }
- else {
- sin6->sin6_family = family;
- sin6->sin6_port = 0;
- inet_pton(family, addr, &sin6->sin6_addr);
-
- if (ioctl(fd, SIOCSIFADDR, &ifr) < 0) {
- LOGD("ioctl SIOCSIFADDR \n");
- close(fd);
- return -1;
- }
-
- sin6->sin6_family = family;
- inet_pton(family, netmask, &sin6->sin6_addr);
- if (ioctl(fd, SIOCSIFNETMASK, &ifr) < 0) {
- LOGD("ioctl SIOCSIFNETMASK \n");
- close(fd);
- return -1;
- }
- }
- close(fd);
-
- return 0;
- }
- static int get_ipaddr(int family, char *iface, char *addr, char *netmask)
- {
- struct ifaddrs *ifap, *ifa;
- if (getifaddrs(&ifap) == -1) {
- LOGE("___getifaddrs() error\n");
- return -1;
- }
- for (ifa = ifap; ifa != NULL; ifa = ifa->ifa_next) {
- if (ifa->ifa_addr != NULL && ifa->ifa_addr->sa_family == family) {
- if(family==AF_INET) {
- struct sockaddr_in *sa,*msk;
- sa = (struct sockaddr_in *) ifa->ifa_addr;
- msk = (struct sockaddr_in *) ifa->ifa_netmask;
- if(addr) {
- inet_ntop(family, &sa->sin_addr, addr, INET6_ADDRSTRLEN);
- printf("___addr: %s\n", addr);
- }
- if(netmask) {
- inet_ntop(family, &msk->sin_addr, netmask, INET6_ADDRSTRLEN);
- //printf("___netmask: %s\n", netmask);
- }
- }
- else {
- struct sockaddr_in6 *sa,*msk;
- sa = (struct sockaddr_in6 *) ifa->ifa_addr;
- msk = (struct sockaddr_in6 *) ifa->ifa_netmask;
-
- if(addr) {
- inet_ntop(family, &sa->sin6_addr, addr, INET6_ADDRSTRLEN);
- //printf("___addr6: %s\n", addr);
- }
- if(netmask) {
- inet_ntop(family, &msk->sin6_addr, netmask, INET6_ADDRSTRLEN);
- //printf("___netmask6: %s\n", netmask);
- }
- }
- }
- }
- freeifaddrs(ifap);
-
- return 0;
- }
- static int set_gateway(int family, const char* gateway)
- {
- struct rtentry rt;
-
- int fd;
-
- fd = socket(family, SOCK_DGRAM, 0);
- if (fd < 0) {
- LOGE("socket SOCK_DGRAM failed\n");
- return -1;
- }
-
- memset(&rt, 0, sizeof(rt));
- if(family==AF_INET) {
- struct sockaddr_in* sin;
- sin = (struct sockaddr_in*)&rt.rt_gateway;
- sin->sin_family = family;
- sin->sin_port = 0;
- inet_pton(family, gateway, &sin->sin_addr);
-
- sin = (struct sockaddr_in*)&rt.rt_dst;
- sin->sin_family = family;
- sin->sin_addr.s_addr = 0;
- }
- else {
- struct sockaddr_in6* sin;
- sin = (struct sockaddr_in6*)&rt.rt_gateway;
- sin->sin6_family = family;
- sin->sin6_port = 0;
- inet_pton(family, gateway, &sin->sin6_addr);
-
- sin = (struct sockaddr_in6*)&rt.rt_dst;
- sin->sin6_family = family;
- //sin->sin6_addr.s_addr[0] = 0;
- }
-
- rt.rt_flags = RTF_GATEWAY;
- if (ioctl(fd, SIOCADDRT, &rt) < 0) {
- LOGE("ioctl SIOCADDRT failed\n");
- close(fd);
- return -1;
- }
- close(fd);
-
- return 0;
- }
- static int get_gateway(int family, const char* gateway)
- {
- #if 0
- int sockfd = socket(AF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
- if (sockfd == -1) {
- log_e("set_gateway, socket AF_NETLINK error\n");
- return -1;
- }
- struct sockaddr_nl sa;
- memset(&sa, 0, sizeof(sa));
- sa.nl_family = AF_NETLINK;
- if (bind(sockfd, (struct sockaddr *) &sa, sizeof(sa)) == -1) {
- log_e("set_gateway, bind error\n");
- return -1;
- }
-
- struct nlreq req;
- memset(&req, 0, sizeof(req));
- req.hdr.nlmsg_len = NLMSG_LENGTH(sizeof(struct rtmsg));
- req.hdr.nlmsg_type = RTM_GETROUTE;
- req.hdr.nlmsg_flags = NLM_F_REQUEST | NLM_F_DUMP;
- req.msg.rtm_family = family;
- req.msg.rtm_table = RT_TABLE_MAIN;
- req.msg.rtm_protocol = RTPROT_UNSPEC;
- req.msg.rtm_scope = RT_SCOPE_UNIVERSE;
- req.msg.rtm_type = RTN_UNICAST;
- struct iovec iov;
- memset(&iov, 0, sizeof(iov));
- iov.iov_base = &req;
- iov.iov_len = req.hdr.nlmsg_len;
- struct msghdr msg;
- memset(&msg, 0, sizeof(msg));
- msg.msg_name = &sa;
- msg.msg_namelen = sizeof(sa);
- msg.msg_iov = &iov;
- msg.msg_iovlen = 1;
-
- char buf[BUFSIZE];
- memset(buf, 0, sizeof(buf));
- struct nlmsghdr *hdr;
- int len;
- if (sendmsg(sockfd, &msg, 0) == -1) {
- log_e("set_gateway, sendmsg error\n");
- return -1;
- }
- while ((len = recv(sockfd, buf, sizeof(buf), 0)) > 0) {
- for (hdr = (struct nlmsghdr *) buf; NLMSG_OK(hdr, len); hdr = NLMSG_NEXT(hdr, len)) {
- if (hdr->nlmsg_type == NLMSG_DONE) {
- goto finish;
- }
- if (hdr->nlmsg_type == NLMSG_ERROR) {
- log_e("set_gateway, NLMSG_ERROR\n");
- return -1;
- }
- struct rtmsg *rt = (struct rtmsg *) NLMSG_DATA(hdr);
- if (rt->rtm_family != family || rt->rtm_table != RT_TABLE_MAIN || rt->rtm_type != RTN_UNICAST) {
- continue;
- }
- struct rtattr *attr;
- int attrlen;
- for (attr = (struct rtattr *) RTM_RTA(rt), attrlen = RTM_PAYLOAD(hdr); RTA_OK(attr, attrlen); attr = RTA_NEXT(attr, attrlen)) {
- if (attr->rta_type == RTA_GATEWAY) {
- char gw_addr[INET_ADDRSTRLEN];
- struct in_addr addr;
- memcpy(&addr, RTA_DATA(attr), sizeof(addr));
- if (inet_ntop(family, &addr, gateway, INET6_ADDRSTRLEN) == NULL) {
- log_e("set_gateway, inet_ntop error\n");
- continue;
- }
- printf("Gateway address: %s\n", gw_addr);
- log_i("set_gateway, Gateway address: %s\n");
- goto finish;
- }
- }
- }
- }
- finish:
- close(sockfd);
- #endif
- return 0;
- }
- static int set_dhcp(int on)
- {
- if(on) {
- system("killall udhcpc && udhcpc –i eth0 –s /usr/share/udhcpc/default.script &");
- }
- else {
- system("killall udhcpc");
- }
- return 0;
- }
- static int is_number(char *str)
- {
- while (*str) {
- if (!isdigit(*str)) {
- return 0;
- }
- str++;
- }
- return 1;
- }
- static int ip_check(char* addr)
- {
- int flag=IP_ERR;
- struct in_addr addr4;
- struct in6_addr addr6;
- if (inet_pton(AF_INET, addr, &addr4) == 1) {
- flag = IP_V4;
- } else if (inet_pton(AF_INET6, addr, &addr6) == 1) {
- flag = IP_V6;
- }
- if(flag==IP_ERR) {
- LOGE("____ wrong ip addr: %s\n", addr);
- }
-
- return flag;
- }
- static int is_ipv6_prefix(char *ss)
- {
- if(is_number(ss)) {
- int val = atoi(ss);
- if(val>1 && val<128) {
- return 1;
- }
- }
- return 0;
- }
- ///////////////////////////////////////////////
- typedef struct {
- char lo[1000];
- char e0[1000];
- char e1[1000];
- }interface_t;
- typedef uint32_t u32;
- #define XLEN(a,b) ((u32)(a)-(u32)(b))
- typedef struct {
- char *ptr;
- int len;
- }nic_t;
- static int get_nic(char *nbuf, int nlen, char *nic, char *fbuf, int flen)
- {
- int r,xlen;
- char tmp[20];
- char *p1,*p2;
- char *fend=fbuf+flen-1;
- sprintf(tmp, "auto %s", nic);
- p1 = strstr(fbuf, tmp);
- if(!p1) {
- return -1;
- }
- p2 = strstr(p1+1, "auto");
- if(!p2 || p2>fend) {
- p2 = fend;
- }
- xlen = XLEN(p2,p1);
- memcpy(nbuf, p1, xlen);
- nbuf[xlen] = 0;
- return 0;
- }
- static int get_ip(char *nbuf, char *token, char *ipaddr, int ipv)
- {
- int cnt=0;
- char *p,*p1,*p2;
- char *pbuf=NULL;
- if(ipv==IP_V4) {
- pbuf = strstr(nbuf, "inet");
- }
- else {
- pbuf = strstr(nbuf, "inet6");
- }
- if(pbuf==NULL) {
- LOGE("___get_ip, not found inet or inet6\n");
- return -1;
- }
- p = strstr(pbuf, token);
- if(p) {
- p += strlen(token);
- while(isspace(*p)) p++;
- p1 = p;
- while(!isspace(*p)) p++;
- p2 = p;
- for(p=p1; p<p2; p++) {
- if(isgraph(*p)) {
- ipaddr[cnt++] = *p;
- }
- }
- ipaddr[cnt] = 0;
- return 0;
- }
- return -1;
- }
- static int get_dns(char *nbuf, char *dns1, char *dns2)
- {
- int cnt;
- char *p1,*p2;
- char *token="server=";
- p1 = strstr(nbuf, token);
- if(p1==NULL) {
- dns1[0] = 0;
- return -1;
- }
- p1 += strlen(token); cnt=0;
- while(!isspace(*p1)) {
- dns1[cnt++] = *p1;
- p1++;
- }
- dns1[cnt] = 0;
- p2 = strstr(p1, token);
- if(p2==NULL) {
- dns2[0] = 0;
- return -1;
- }
- p2 += strlen(token); cnt=0;
- while(!isspace(*p2)) {
- dns2[cnt++] = *p2;
- p2++;
- }
- dns2[cnt] = 0;
- return 0;
- }
- static int set_dns(NetworkInfo_t *para)
- {
- int r;
- char tmp[100];
- char buf[500];
-
- LOGD("_____ set_dns\n");
- int chk1=ip_check(para->dns1);
- int chk2=ip_check(para->dns2);
- if( chk1==IP_ERR && chk2==IP_ERR) {
- LOGE("___ dns1 and dns2 are all wrong!\n");
- return -1;
- }
- memset(buf, 0, sizeof(buf));
- strcpy(buf, "interface=eth0\n");
- if(chk1) {
- sprintf(tmp, "server=%s\n", para->dns1);
- strcat(buf, tmp);
- }
- if(chk2) {
- sprintf(tmp, "server=%s\n", para->dns2);
- strcat(buf, tmp);
- }
- r = file_save(FILE_DNS, buf, strlen(buf))+1;
- if(r==0) {
- system("killall dnsmasq; /usr/sbin/dnsmasq &");
- }
- return r;
- }
- static int get_net(NetworkInfo_t *para, int ipv)
- {
- int r;
- char *p=NULL;
- interface_t intf;
- buf_t fbuf;
- r = file_load(FILE_NET, &fbuf);
- if(r) {
- return -1;
- }
-
- memset(&intf, 0, sizeof(intf));
- get_nic(intf.lo, sizeof(intf.lo), "lo", fbuf.buf, fbuf.dlen);
- get_nic(intf.e0, sizeof(intf.e0), "eth0", fbuf.buf, fbuf.dlen);
- get_nic(intf.e1, sizeof(intf.e1), "eth1", fbuf.buf, fbuf.dlen);
- p = strstr(intf.e0, (ipv==IP_V4)?"inet":"inet6");
- if(p==NULL) {
- LOGE("___get_ipv%d, %s not found!\n", (ipv==IP_V4)?4:6, (ipv==IP_V4)?"inet":"inet6");
- free(fbuf.buf); return -1;
- }
-
- if(strstr(p, "dhcp")) {
- para->mode = 0;
- }
- else if(strstr(p, "static")) {
- para->mode = 1;
- } else {
- LOGE("___get_net, dhcp/static not found!\n");
- free(fbuf.buf); return -1;
- }
- get_ip(p, "address", para->ip_address, ipv);
- get_ip(p, "netmask", para->mask, ipv);
- get_ip(p, "gateway", para->gateway, ipv);
- free(fbuf.buf);
- r = file_load(FILE_DNS, &fbuf);
- if(r) {
- return -1;
- }
- get_dns(fbuf.buf, para->dns1, para->dns2);
- return 0;
- }
- static int nw_check(NetworkInfo_t *para, int ipv)
- {
- if(ipv==IP_V4) {
- if((ip_check(para->ip_address)!=IP_V4) || (ip_check(para->mask)!=IP_V4) || (ip_check(para->gateway)!=IP_V4)) {
- return -1;
- }
- }
- else {
- if((ip_check(para->ip_address)!=IP_V6) || (!is_ipv6_prefix(para->mask)) || (ip_check(para->gateway)!=IP_V6)) {
- return -1;
- }
- }
- return 0;
- }
- static int set_net(NetworkInfo_t *para, int ipv)
- {
- int r,ipv2;
- char *p1,*p2,*p3,*p4;
- int flen=100*1000;
- NetworkInfo_t info4,info6,*info;
- get_net(&info4, IP_V4);
- get_net(&info6, IP_V6);
- info = ((ipv==IP_V4)?&info4:&info6);
-
- if(para->mode<2 && para->mode!=info->mode) {
- info->mode = para->mode;
- }
-
- ipv2 = ip_check(para->ip_address);
- if((ipv==ipv2) && strcmp(para->ip_address, info->ip_address)) {
- strcpy(info->ip_address, para->ip_address);
- }
-
- if(ipv==IP_V4) {
- ipv2 = ip_check(para->mask);
- }
- else {
- ipv2=is_ipv6_prefix(para->mask)?IP_V6:IP_ERR;
- }
- if((ipv==ipv2) && strcmp(para->mask, info->mask)) {
- strcpy(info->mask, para->mask);
- }
- ipv2 = ip_check(para->gateway);
- if((ipv==ipv2) && strcmp(para->gateway, info->gateway)) {
- strcpy(info->gateway, para->gateway);
- }
-
- char *fbuf=malloc(flen);
- if(!fbuf) {
- LOGE("___ malloc %d failed\n", flen);
- return -1;
- }
- #if 0
- file_load(FILE_NET, fbuf);
- get_nic(pNet->lo, sizeof(pNet->lo), "lo", fbuf, flen);
- get_nic(pNet->e1, sizeof(pNet->e1), "eth1", fbuf, flen);
- snprintf(pNet->e0, sizeof(pNet->e0), "auto eth0\niface eth0 inet %s\n address %s\n netmask %s\n gateway %s\n\n",
- info4.mode?"static":"dhcp", info4.ip_address, info4.mask, info4.gateway);
- strcpy(fbuf, pNet->lo);
- strcat(fbuf, pNet->e0);
- strcat(fbuf, pNet->e1);
- #else
- char *lo="auto lo\niface lo inet loopback\n\n";
- char *eth1="auto eth1\niface eth1 inet dhcp\npre-up /etc/network/nfs_check\nwait-delay 5\nhostname $(hostname)\n";
- char eth0[1000];
- int chk4=nw_check(&info4, IP_V4);
- int chk6=nw_check(&info6, IP_V6);
- char temp[400];
- strcpy(eth0, "auto eth0\n");
- if(chk4==0) {
- 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);
- strcat(eth0, temp);
- }
- if(chk6==0) {
- 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);
- strcat(eth0, temp);
- }
- sprintf(fbuf, "%s%s%s", lo, eth0, eth1);
- #endif
- //LOGD("___interfaces___:\n%s", fbuf);
- r = file_save(FILE_NET, fbuf, strlen(fbuf)); free(fbuf);
- return r;
- }
- //////////////////////////////////////////////////////////////////////////////
- int sys_set_net(NetworkInfo_t *para, int ipVersion)
- {
- int r,ipv;
- #ifdef USE_BASH
- char cmd[200];
- snprintf(cmd, sizeof(cmd), "ip addr flush dev %s", DFLT_NIC);
- r = system(cmd);
- if(r) {
- LOGE("set ipaddr failed\n");
- return -1;
- }
- snprintf(cmd, sizeof(cmd), "ip addr add %s/%s dev %s", para->ip_address, para->mask, DFLT_NIC);
- r = system(cmd);
- if(r) {
- LOGE("set netmask %s failed\n", para->mask);
- return -1;
- }
- snprintf(cmd, sizeof(cmd), "ip route add default via %s dev %s", para->gateway, DFLT_NIC);
- r = system(cmd);
- if(r) {
- LOGE("set gateway %s failed\n", para->gateway);
- return -1;
- }
- snprintf(cmd, sizeof(cmd), "ip link set dev %s up", DFLT_NIC);
- r = system(cmd);
- if(r) {
- LOGE("restart %s failed\n", DFLT_NIC);
- return -1;
- }
- #elif defined USE_FILE
- NetworkInfo_t para2;
- r = get_net(¶2, ipVersion);
- if(r==0) {
- if(memcmp(para, ¶2, sizeof(NetworkInfo_t))==0) {
- LOGW("%s NetworkInfo is same, no need to set\n", DFLT_NIC);
- return -1;
- }
- set_dns(para);
- r = set_net(para, ipVersion);
- if(r==0) {
- //system("ip addr flush dev eth0");
- //sprintf(eth0, "ip link set dev eth0 %s/%d && ip link set eth0 up && ip link set eth0 down", info->ip_address, info->netmask);
- //system(eth0);
- system("ifdown eth0; ifup eth0");
- exit(0);
- }
- }
- else {
- LOGE("get_net failed\n");
- }
- #elif defined USE_API
- if(para->mode==IP_DHCP) {
- r = set_dhcp(1);
- }
- else {
- set_dhcp(0);
- int family=(ipv==IP_V4)?AF_INET:AF_INET6;
- r = set_ipaddr(family, "eth0", para->ip_address, para->mask);
- if(r) {
- LOGE("set_ipaddr failed\n");
- return r;
- }
-
- r = set_gateway(family, para->gateway);
- if(r) {
- LOGE("set_gateway failed\n");
- return r;
- }
- r = set_dns(family, "eth0", para->dns1, para->dns2);
- if(r) {
- LOGE("set_dns failed\n");
- return r;
- }
- }
- #endif
-
- return r;
- }
- int sys_get_net(NetworkInfo_t *para, int ipVersion)
- {
- int r;
- int family=AF_INET;
- #ifdef USE_BASH
-
- #elif defined USE_FILE
- r = get_net(para, ipVersion);
- #elif defined USE_API
- r = get_ipaddr(family, DFLT_NIC, para->ip_address, para->mask);
- if(r) {
- LOGE("set_ipaddr failed\n");
- return r;
- }
-
- r = get_gateway(family, para->gateway);
- if(r) {
- LOGE("set_gateway failed\n");
- return r;
- }
- r = get_dns(family, DFLT_NIC, para->dns1, para->dns2);
- if(r) {
- LOGE("set_dns failed\n");
- return r;
- }
- #endif
- return r;
- }
- int sys_is_running(char *prog)
- {
- FILE *fp;
- int pid=0;
- char buf[20]={0};
- char command[200];
- snprintf(command, sizeof(command), "ps -ef | grep -v grep | grep -w -c %s", prog);
- fp = popen(command, "r");
- if (fp == NULL) {
- LOGE("execute %s failed: %s", command, strerror(errno));
- return 0;
- }
- if (fgets(buf, sizeof(buf), fp)) {
- pid = atoi(buf);
- }
- pclose(fp);
-
- return (pid)?1:0;
- }
- int sys_get_disk_ava_MB(char *path)
- {
- int r;
- struct statvfs st;
- r = statvfs(path, &st);
- if(r) {
- return -1;
- }
- int avaMB = (st.f_frsize*st.f_bavail)/(1024*1024);
- //int freeMB = (st.f_frsize*st.f_bfree)/(1024*1024);
- //printf("____ava: %dMB, free %dMB\n", avaMB, freeMB);
- return avaMB;
- }
- int sys_get_mem_ava_MB(void)
- {
- int r,value=0;
- char buff[32];
- struct sysinfo info;
- #if 0
- r = sysinfo(&info);
- if(r) {
- return -1;
- }
- value = (info.freeram*info.mem_unit)/(1014*1024);
- #else
- FILE *fp = fopen("/proc/meminfo", "r");
- if(!fp) {
- return -1;
- }
- fscanf(fp, "MemTotal: %s kB\n", buff);
- fscanf(fp, "MemFree: %s kB\n", buff);
- fscanf(fp, "MemAvailable: %s kB\n", buff);
- value = atol(buff)/1024;
- fclose(fp);
- #endif
- return value;
- }
- int sys_get_mac(char *mac, int maclen)
- {
- int sock;
- struct ifreq ifr;
- if(!mac || maclen<6) {
- return -1;
- }
- sock = socket(AF_INET, SOCK_DGRAM, 0);
- if (sock == -1){
- return -1;
- }
- strcpy(ifr.ifr_name, "eth0");
- if (ioctl(sock, SIOCGIFHWADDR, &ifr) < 0){
- close(sock); return -1;
- }
- close(sock);
- uint8_t *p = (uint8_t*)ifr.ifr_hwaddr.sa_data;
- snprintf(mac, maclen, "%02x%02x%02x%02x%02x%02x", p[0], p[1], p[2], p[3], p[4], p[5]);
- return 0;
- }
- int sys_get_chip_id(int *id)
- {
- int sock;
- struct ifreq ifr;
- if(!id) {
- return -1;
- }
- sock = socket(AF_INET, SOCK_DGRAM, 0);
- if (sock == -1){
- return -1;
- }
- strcpy(ifr.ifr_name, "eth0");
- if (ioctl(sock, SIOCGIFHWADDR, &ifr) < 0){
- close(sock); return -1;
- }
- close(sock);
- *id = *((int*)ifr.ifr_hwaddr.sa_data);
- return 0;
- }
- ///////////////////////////////////////////////////////////////////////
- #define MAX_BUFFER_SIZE 4096
- static void* plug_thread(void *arg)
- {
- int nlSock;
- struct sockaddr_nl srcAddr;
- struct nlmsghdr *nlMsg;
- char buffer[MAX_BUFFER_SIZE];
- int plug_in=0,plug_out=0;
- thread_handle_t *h=(thread_handle_t*)arg;
- printf("___plug_thread start\n");
- nlSock = socket(PF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
- if (nlSock < 0) {
- printf("Failed to create Netlink socket");
- goto quit;
- }
- memset(&srcAddr, 0, sizeof(srcAddr));
- srcAddr.nl_family = AF_NETLINK;
- srcAddr.nl_groups = RTMGRP_LINK | RTMGRP_NOTIFY;
- if (bind(nlSock, (struct sockaddr *)&srcAddr, sizeof(srcAddr)) < 0) {
- perror("Failed to bind Netlink socket");
- close(nlSock);
- goto quit;
- }
- int flags = fcntl(nlSock, F_GETFL, 0);
- fcntl(nlSock, F_SETFL, flags | O_NONBLOCK);
- while (h->quit==0) {
- fd_set readSet;
- FD_ZERO(&readSet);
- FD_SET(nlSock, &readSet);
- // 设置超时时间为 1 秒
- struct timeval timeout;
- timeout.tv_sec = 1;
- timeout.tv_usec = 0;
- // 监视 Netlink Socket 的可读状态
- int result = select(nlSock + 1, &readSet, NULL, NULL, &timeout);
- if (result == -1) {
- perror("Failed to select");
- goto quit;
- } else if (result == 0) {
- // 超时,没有新数据到达
- //printf("Timeout\n");
- continue;
- }
- if (FD_ISSET(nlSock, &readSet)) {
- // 接收消息
- int bytesRead = recv(nlSock, buffer, MAX_BUFFER_SIZE, 0);
- if (bytesRead < 0) {
- perror("Failed to receive message from Netlink socket");
- close(nlSock);
- goto quit;
- }
- // 遍历接收到的消息,获取到Netlink的消息头。
- nlMsg = (struct nlmsghdr *)buffer;
- while (NLMSG_OK(nlMsg, bytesRead)) {
- if (nlMsg->nlmsg_type == NLMSG_DONE) {
- break;
- }
- if (nlMsg->nlmsg_type == RTM_NEWLINK || nlMsg->nlmsg_type == RTM_DELLINK) {
- //NLMSG_DATA()获取完整的ifinfomsg结构体的指针
- //根据消息头nlMsg获取到其后面的数据部分ifinfomsg结构体的指针
- struct ifinfomsg *ifInfo = (struct ifinfomsg *)NLMSG_DATA(nlMsg);
- if (ifInfo->ifi_family == AF_UNSPEC) {
- //获取路由信息的属性,以及消息负载的长度,通过这两个参数来判断本次的数据是否有效
- struct rtattr *attribute = IFLA_RTA(ifInfo);
- int attributeLen = IFLA_PAYLOAD(nlMsg);
-
- while (RTA_OK(attribute, attributeLen)) {
- char *ifaceName = (char *)RTA_DATA(attribute);
- if ((attribute->rta_type == IFLA_IFNAME) && (strcmp(ifaceName, "eth0")==0)) {
-
- //printf("____Interface: %s \n", ifaceName);
- if((ifInfo->ifi_flags & IFF_LOWER_UP)) {
- if(plug_in==0) {
- system("ifup eth0");
- printf("____eth0 plug in\n");
- plug_in = 1; plug_out = 0;
- }
- }
- else {
- if(plug_out==0) {
- system("ifdown eth0");
- printf("____eth0 plug out\n");
- plug_in = 0; plug_out = 1;
- }
- }
- }
- attribute = RTA_NEXT(attribute, attributeLen);
- }
- }
- }
- nlMsg = NLMSG_NEXT(nlMsg, bytesRead);
- }
- }
- }
- close(nlSock);
- quit:
- printf("___plug_thread exit\n");
- pthread_exit(NULL);
- }
- ///////////////////////////////////////////////////////////
- //#define RAM_DEBUG
- //#define ROM_DEBUG
- #ifdef RAM_DEBUG
- #define RAM_THRD 177
- #define RAM_PERIOD 1
- #else
- #define RAM_THRD 50
- #define RAM_PERIOD 60
- #endif
- #ifdef ROM_DEBUG
- #define ROM_THRD 500
- #define ROM_PERIOD 1
- #else
- #define ROM_THRD 50
- #define ROM_PERIOD 1
- #endif
- static void mem_check(void)
- {
- int avaMB,flag=0;
- static uint32_t mem_cnt=0;
- mem_cnt++;
- if(mem_cnt%RAM_PERIOD==0) {
- avaMB = sys_get_mem_ava_MB();
- printf("___ available mem: %dMB, threshold: %dMB\n", avaMB, RAM_THRD);
- if(avaMB<RAM_THRD) {
- printf("___ free mem less than %dMB, reboot now ...\n", RAM_THRD);
- sys_reboot();
- }
- }
- if(mem_cnt%ROM_PERIOD==0) {
- avaMB = sys_get_disk_ava_MB("/");
- printf("___ available rom: %dMB, threshold: %dMB\n", avaMB, ROM_THRD);
- if(avaMB<ROM_THRD) {
- flag = 1;
- }
- dev_sql_set_overwrite(flag);
- }
- }
- static void* polling_thread(void *arg)
- {
- thread_handle_t *h=(thread_handle_t*)arg;
- while(h->quit==0) {
- mem_check();
- sleep(1);
- }
- pthread_exit(NULL);
- }
- int sys_init(void)
- {
- //userShellInit();
- initLogger();
- lock_s_init();
-
- thread_start(THREAD_ID_POLLING, polling_thread, NULL, 4*MB, 0);
- return 0;
- }
|