- <span style="font-family:Courier New;">
- /********************************** (C) COPYRIGHT *******************************
- * File Name : add_route.c
- * Author : skdkjzz
- * Date : 2014/08/09
- * Description : 设置静态ip,添加路由表
- *********************************************************************************/
- #include <error.h>
- #include <stdio.h>
- #include <string.h>
- #include <net/if.h>
- #include <stdlib.h>
- #include <arpa/inet.h>
- #include <asm/types.h> /* glibc 2 conflicts with linux/types.h */
- #include <net/route.h>
- #include <sys/errno.h>
- #include <sys/ioctl.h>
- #include <sys/types.h>
- #include <sys/socket.h>
- #include <netinet/in.h>
- #include <net/if_arp.h>
- #include <netinet/if_ether.h>
- #include <fcntl.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <errno.h>
- #include <string.h>
- #include <unistd.h>
- #include <signal.h>
- #include <dirent.h>
- #include <arpa/inet.h>
- #include <pthread.h>
- #include <sys/vfs.h>
- #include <sys/time.h>
- #include <sys/wait.h>
- #include <sys/ioctl.h>
- #include <sys/types.h>
- #include <sys/socket.h>
- #include <net/if.h>
- #include <netinet/in.h>
- #include <net/route.h>
- #include <net/if_arp.h>
- #define SET_SA_FAMILY(addr, family) \
- memset ((char *) &(addr), '\0', sizeof(addr)); \
- addr.sa_family = (family);
- #define SIN_ADDR(x) (((struct sockaddr_in *) (&(x)))->sin_addr.s_addr)
- int SetIfAddr(char *Ifname, char *Ipaddr, char *mask)
- {
- int fd;
- //int rc;
- struct ifreq ifr;
- struct sockaddr_in *sin;
- struct rtentry rt;
- fd = socket(AF_INET, SOCK_DGRAM, 0);
- if(fd < 0)
- {
- perror("socket error");
- return -2;
- }
- memset(&ifr,0,sizeof(ifr));
- strcpy(ifr.ifr_name,Ifname);
- sin = (struct sockaddr_in*)&ifr.ifr_addr;
- sin->sin_family = AF_INET;
- //IP地址
- if(inet_aton(Ipaddr,&(sin->sin_addr)) < 0)
- {
- perror("inet_aton error");
- return -3;
- }
- if(ioctl(fd,SIOCSIFADDR,&ifr) < 0)
- {
- perror("ioctl SIOCSIFADDR error");
- return -4;
- }
- //子网掩码
- if(inet_aton(mask,&(sin->sin_addr)) < 0)
- {
- perror("inet_pton error");
- return -5;
- }
- if(ioctl(fd, SIOCSIFNETMASK, &ifr) < 0)
- {
- perror("ioctl");
- return -6;
- }
- return 0;
- close(fd);
- }
- /*
- * funtion:set the file lock
- * params:fd is file Num,type is F_WRLCk,F_RDLCK or F_UNLCK
- -----------------------
- * time:2012-08-22 22:17
- * version:1.0
- */
- int joseph_lock_set(int fd, int type)
- {
- struct flock lock_file;
- lock_file.l_whence = SEEK_SET;
- lock_file.l_start = 0;
- lock_file.l_len = 0;
- lock_file.l_type = type;
- lock_file.l_pid = -1;
- fcntl(fd, F_GETLK, &lock_file);
- if (lock_file.l_type != F_UNLCK)
- {
- if (lock_file.l_type == F_RDLCK)
- {
- }
- else if (lock_file.l_type == F_WRLCK)
- {
- }
- }
- lock_file.l_type = type;
- if ((fcntl(fd, F_SETLKW, &lock_file)) < 0)
- {
- return 1;
- }
- switch(lock_file.l_type)
- {
- case F_RDLCK:
- {
- }
- break;
- case F_WRLCK:
- {
- }
- break;
- case F_UNLCK:
- {
- return 1;
- }
- break;
- default:
- break;
- }/* end of switch */
- return 0;
- }
- #define JOSEPH_ETH0_IFCG "/etc/sysconfig/network-scripts/ifcfg-eth0"
- int joseph_read_user_file(char *param_name)
- {
- int fd;
- int param_value;
- char buf_name[64];
- char buf[128];
- FILE *fp;
- strcpy(buf_name,param_name);
- fp = fopen(JOSEPH_ETH0_IFCG, "r");
- if(fp == NULL)
- {
- return -1;
- }
- fd=fileno(fp);
- joseph_lock_set(fd, F_RDLCK);//set the write lock
- while(fgets(buf, sizeof(buf), fp)){
- if(strstr(buf,buf_name)){
- printf("%s:[%d] %s = %s",__FUNCTION__,__LINE__,param_name,(buf+strlen(buf_name)+1));
- }
- }
- joseph_lock_set(fd, F_UNLCK);// unlock the file
- fclose(fp);
- return 0;
- }
- void joseph_alter_ipnc_param(char *joseph_param_name, char *joseph_param_values)
- {
- char buf[1024];
- char conf_path[64] = JOSEPH_ETH0_IFCG;
- if((joseph_param_name == NULL) ||( joseph_param_values == NULL))
- {
- return;
- }
- if((strlen(joseph_param_name) == 0) ||( strlen(joseph_param_values) == 0))
- {
- return;
- }
- sprintf(buf,"sed -i 's/%s=.*/%s=%s/g' %s",\
- joseph_param_name,joseph_param_name,joseph_param_values,conf_path);
- system(buf);
- return;
- }
- int main ()
- {
- #if 1
- static int sock_fd = -1; /* socket for doing interface ioctls */
- struct rtentry rt;
- FILE *db_fd = (FILE *) 0;
- char ifname[32];
- u_int32_t dstaddr, gateway;
- int j,k;
- char *p;
- FILE *fp;
- char cmd[30];
- char cmd1[5][20];
- memset(cmd, 0, sizeof(cmd));
- memset(cmd1, 0, sizeof(cmd1));
- if((fp=fopen("/etc/sysconfig/network-scripts/ifcfg-eth0","r"))==NULL){
- printf("Read ifcfg-eth0 error!\n");
- return -1;
- }
- for(j=0;j<5;j++){
- if((p=fgets(cmd,1024,fp))==NULL){
- printf("read data error!\n");
- return -2;
- }
- while(*p!='=')p++;
- p++;
- k=0;
- while(*p!='\n')
- {
- cmd1[j][k]=*p;
- k++;
- p++;
- }
- memset(cmd, 0, sizeof(cmd));
- }
- if(SetIfAddr(cmd1[0],cmd1[1],cmd1[2])!=0)
- {
- printf("config ip failed!\n");
- return -7;
- }
- strcpy(ifname, cmd1[0]);
- dstaddr = inet_addr(cmd1[3]);
- gateway = inet_addr(cmd1[4]);
- /* Get an internet socket for doing socket ioctls. */
- sock_fd = socket(AF_INET, SOCK_DGRAM, 0);
- /*open for debug*/
- db_fd = fopen("/tmp/addrt_db.txt","a+");
- fprintf(db_fd, "ifname=%s ouraddr=%x gateway=%x\n", ifname, dstaddr, gateway);
- memset (&rt, 0, sizeof (rt));
- /*set Destination addr*/
- SET_SA_FAMILY (rt.rt_dst, AF_INET);
- SIN_ADDR(rt.rt_dst) = dstaddr;
- /*set gw addr*/
- SET_SA_FAMILY (rt.rt_gateway, AF_INET);
- SIN_ADDR(rt.rt_gateway) = gateway;
- fprintf(db_fd,"mygateway=%x\n", SIN_ADDR(rt.rt_gateway));
- /*set genmask addr*/
- SET_SA_FAMILY (rt.rt_genmask, AF_INET);
- SIN_ADDR(rt.rt_genmask) = 0L;
- rt.rt_dev = ifname;
- rt.rt_flags = RTF_GATEWAY;
- if (ioctl(sock_fd, SIOCADDRT, &rt) < 0)
- {
- fprintf(db_fd,"route add err num=%m\n",errno);
- printf("Err\n");
- return 0;
- }
- fprintf(db_fd,"route add success route=%x\n", SIN_ADDR(rt.rt_gateway));
- fclose(db_fd);
- #else
- joseph_read_user_file("DEVICE");
- joseph_read_user_file("IPADDR");
- joseph_read_user_file("GATEWAY");
- joseph_read_user_file("DNS");
- joseph_read_user_file("BROADCAST");
- joseph_read_user_file("BOOTPROTO");
- joseph_read_user_file("ONBOOT");
- joseph_alter_ipnc_param("IPADDR","10.10.1.250");
- joseph_read_user_file("IPADDR");
- #endif
- return 0;
- }
- </span>