Linux 程序中获取网关
2014-04-30 17:31
267 查看
#include <asm/types.h> #include <netinet/ether.h> #include <netinet/in.h> #include <net/if.h> #include <stdio.h> #include <sys/socket.h> #include <sys/ioctl.h> #include <linux/netlink.h> #include <linux/rtnetlink.h> #include <sys/types.h> #define BUFSIZE 8192 struct route_info{ u_int dstAddr; u_int srcAddr; u_int gateWay; char ifName[IF_NAMESIZE]; }; int readNlSock(int sockFd, char *bufPtr, int seqNum, int pId) { struct nlmsghdr *nlHdr; int readLen = 0, msgLen = 0; do{ /* Recieve response from the kernel */ if((readLen = recv(sockFd, bufPtr, BUFSIZE - msgLen, 0)) < 0) { perror("SOCK READ: "); return -1; } nlHdr = (struct nlmsghdr *)bufPtr; /* Check if the header is valid */ if((NLMSG_OK(nlHdr, readLen) == 0) || (nlHdr->nlmsg_type == NLMSG_ERROR)) { perror("Error in recieved packet"); return -1; } /* Check if the its the last message */ if(nlHdr->nlmsg_type == NLMSG_DONE) { break; } else { /* Else move the pointer to buffer appropriately */ bufPtr += readLen; msgLen += readLen; } /* Check if its a multi part message */ if((nlHdr->nlmsg_flags & NLM_F_MULTI) == 0) { /* return if its not */ break; } } while((nlHdr->nlmsg_seq != seqNum) || (nlHdr->nlmsg_pid != pId)); return msgLen; } /* For printing the routes. */ void printRoute(struct route_info *rtInfo) { char tempBuf[512]; /* Print Destination address */ if(rtInfo->dstAddr != 0) strcpy(tempBuf, (char *)inet_ntoa(rtInfo->dstAddr)); else sprintf(tempBuf,"*.*.*.*\t"); fprintf(stdout,"%s\t", tempBuf); /* Print Gateway address */ if(rtInfo->gateWay != 0) strcpy(tempBuf, (char *)inet_ntoa(rtInfo->gateWay)); else sprintf(tempBuf,"*.*.*.*\t"); fprintf(stdout,"%s\t", tempBuf); /* Print Interface Name*/ fprintf(stdout,"%s\t", rtInfo->ifName); /* Print Source address */ if(rtInfo->srcAddr != 0) strcpy(tempBuf, (char *)inet_ntoa(rtInfo->srcAddr)); else sprintf(tempBuf,"*.*.*.*\t"); fprintf(stdout,"%s\n", tempBuf); } /* For parsing the route info returned */ void parseRoutes(struct nlmsghdr *nlHdr, struct route_info *rtInfo,char *gateway) { struct rtmsg *rtMsg; struct rtattr *rtAttr; int rtLen; char *tempBuf = NULL; tempBuf = (char *)malloc(100); rtMsg = (struct rtmsg *)NLMSG_DATA(nlHdr); /* If the route is not for AF_INET or does not belong to main routing table then return. */ if((rtMsg->rtm_family != AF_INET) || (rtMsg->rtm_table != RT_TABLE_MAIN)) return; /* get the rtattr field */ rtAttr = (struct rtattr *)RTM_RTA(rtMsg); rtLen = RTM_PAYLOAD(nlHdr); for(;RTA_OK(rtAttr,rtLen);rtAttr = RTA_NEXT(rtAttr,rtLen)) { switch(rtAttr->rta_type) { case RTA_OIF: if_indextoname(*(int *)RTA_DATA(rtAttr), rtInfo->ifName); break; case RTA_GATEWAY: rtInfo->gateWay = *(u_int *)RTA_DATA(rtAttr); break; case RTA_PREFSRC: rtInfo->srcAddr = *(u_int *)RTA_DATA(rtAttr); break; case RTA_DST: rtInfo->dstAddr = *(u_int *)RTA_DATA(rtAttr); break; } } if (strstr((char *)inet_ntoa(rtInfo->dstAddr), "0.0.0.0")) sprintf(gateway, (char *)inet_ntoa(rtInfo->gateWay)); //printRoute(rtInfo); free(tempBuf); return; } int get_gateway(char *gateway) { struct nlmsghdr *nlMsg; struct rtmsg *rtMsg; struct route_info *rtInfo; char msgBuf[BUFSIZE]; int sock, len, msgSeq = 0; char buff[1024]; /* Create Socket */ if((sock = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_ROUTE)) <0) perror("Socket Creation: "); /* Initialize the buffer */ memset(msgBuf, 0, BUFSIZE); /* point the header and the msg structure pointers into the buffer */ nlMsg = (struct nlmsghdr *)msgBuf; rtMsg = (struct rtmsg *)NLMSG_DATA(nlMsg); /* Fill in the nlmsg header*/ nlMsg->nlmsg_len = NLMSG_LENGTH(sizeof(struct rtmsg)); // Length of message. nlMsg->nlmsg_type = RTM_GETROUTE; // Get the routes from kernel routing table . nlMsg->nlmsg_flags = NLM_F_DUMP | NLM_F_REQUEST; // The message is a request for dump. nlMsg->nlmsg_seq = msgSeq++; // Sequence of the message packet. nlMsg->nlmsg_pid = getpid(); // PID of process sending the request. /* Send the request */ if(send(sock, nlMsg, nlMsg->nlmsg_len, 0) <0){ printf("Write To Socket Failed...\n"); return -1; } /* Read the response */ if((len = readNlSock(sock, msgBuf, msgSeq, getpid()))< 0) { printf("Read From Socket Failed...\n"); return -1; } /* Parse and print the response */ rtInfo = (struct route_info *)malloc(sizeof(struct route_info)); for(;NLMSG_OK(nlMsg,len);nlMsg = NLMSG_NEXT(nlMsg,len)) { memset(rtInfo, 0, sizeof(struct route_info)); parseRoutes(nlMsg, rtInfo,gateway); } free(rtInfo); close(sock); return 0; } int main() { char gateway[255]={0}; get_gateway(gateway); printf("Gateway:%s\n", gateway); }
注:存在问题!
相关文章推荐
- Linux获取IP_网关_DNS
- Linux C++ 打日志程序和获取上一级目录的程序
- Linux C 程序执行 shell 命令并获取返回结果的方法
- linux c程序获取cpu使用率及内存使用情况
- Linux获取程序编译参数
- linux下获取程序所在目录绝对路径
- linux 系统获取网络ip, mask, gateway, dns信息小程序
- linux c程序中获取shell脚本输出
- 【Linux C代码分享二】UDP Socket示例程序:使用fgets函数获取屏幕输入
- linux下获取程序当前目录绝对路径
- Linux下获取系统的IP,子网掩码,网关,MAC和配置文件的修改
- Linux下获取IP、MAC、网关、掩码的shell脚本
- windows和linux下获取当前程序路径以及cpu数
- linux C程序中获取shell脚本输出(如获取system命令输出)
- Linux中获取当前程序的绝对路径
- linux关于readlink函数获取运行路径的小程序
- linux编程获取网关地址
- Linux 获取并分析程序崩溃时的调用堆栈
- python获取linux系统信息、性能阀值、短信网关发送的例子
- Linux 程序获取文件系统挂载信息