--- embedtools/src/get1steth.c 2009/12/18 14:12:24 1.1 +++ embedtools/src/get1steth.c 2011/06/08 12:45:41 1.2 @@ -0,0 +1,211 @@ +/************************************************************************* + * (C) 2010 AITNET - Sofia/Bulgaria - + * by Michael Pounov + * + * $Author: misho $ + * $Id: get1steth.c,v 1.2 2011/06/08 12:45:41 misho Exp $ + * + *************************************************************************/ +#include "global.h" +#include "get1steth.h" + + +char szIface[MAX_STR]; +int Verbose; +extern char compiled[], compiledby[], compilehost[]; + + +static void Usage() +{ + printf("-= GET_FIRST_ETHERNET =- Get First Ethernet Interface tool\n" + "=== %s === %s@%s ===\n\n" + "Syntax: get1steth [option] [custom_first_interface]\n\n" + "\t-v\t\tVerbose (more -v more verbosity)\n" + "\t-g\t\tOnly get first interface, print and exit ...\n" + "\n", compiled, compiledby, compilehost); +} + +#ifdef HAVE_KLDNEXT +static int kldLoad() +{ + struct module_stat mstat; + register int i, j; + u_char flg = 0; + + memset(&mstat, 0, sizeof mstat); + mstat.version = sizeof mstat; + for (i = kldnext(0); i > 0; i = kldnext(i)) + for (j = kldfirstmod(i); j > 0; j = modfnext(j)) { + if (modstat(j, &mstat) == -1) + continue; + + if (!strncmp(MODVLAN, mstat.name, sizeof MODVLAN)) { + flg = 1; + break; + } + } + if (flg) + return 0; + + if (kldload(MODVLAN) == -1) + return -1; + + return 1; +} +#endif + +// ------------------------------- + +int main(int argc, char **argv) +{ + char ch, GetOnly = 0; + struct ifaddrs *ifa, *ifp; + struct sockaddr_dl *sdl; + struct sockaddr_in *sin; + struct ifreq ifr; + struct vlanreq vlr; + struct ifaliasreq ifra; + struct ifmediareq ifmr; + int s; + + while ((ch = getopt(argc, argv, "hvg")) != -1) + switch (ch) { + case 'g': + GetOnly = 1; + break; + case 'v': + Verbose++; + break; + case 'h': + default: + Usage(); + return 1; + } + argc -= optind; + argv += optind; + + openlog("get1steth", LOG_CONS | LOG_PERROR, LOG_USER); + + if (argc) { + strlcpy(szIface, *argv, MAX_STR); + VERB(1) syslog(LOG_NOTICE, "Info:: Get CUSTOM first interface %s\n", szIface); + } else { + s = socket(PF_INET, SOCK_DGRAM, 0); + if (-1 == s) { + syslog(LOG_ERR, "Error:: socket(INET) #%d - %s\n", errno, strerror(errno)); + closelog(); + return 1; + } + + getifaddrs(&ifa); + for (ifp = ifa; ifp; ifp = ifp->ifa_next) { + if (PF_LINK == ifp->ifa_addr->sa_family && + IFT_ETHER == ((struct sockaddr_dl*) ifp->ifa_addr)->sdl_type) { + memset(&ifmr, 0, sizeof ifmr); + strlcpy(ifmr.ifm_name, ifp->ifa_name, IFNAMSIZ); + if (ioctl(s, SIOCGIFMEDIA, &ifmr) == -1) { + syslog(LOG_ERR, "Error:: media interface=%s ioctl(SIOCGIFMEDIA) #%d - %s\n", + ifmr.ifm_name, errno, strerror(errno)); + close(s); + closelog(); + return 1; + } + + if (IFM_ETHER == IFM_TYPE(ifmr.ifm_current)) { + strlcpy(szIface, ifp->ifa_name, MAX_STR); + sdl = (struct sockaddr_dl*) ifp->ifa_addr; + VERB(2) syslog(LOG_NOTICE, "Info:: Get first interface=%s MAC=%s\n", szIface, + ether_ntoa((struct ether_addr*) LLADDR(sdl))); + break; + } + } + } + freeifaddrs(ifa); + + close(s); + } + if (!*szIface) { + syslog(LOG_NOTICE, "Info:: Ethernet interface not found!!!\n"); + + closelog(); + return 1; + } + + if (GetOnly) { + printf("%s\n", szIface); + + closelog(); + return 0; + } + +#ifdef HAVE_KLDNEXT + s = kldLoad(); + if (s == -1) { + syslog(LOG_ERR, "Error:: kldload(if_vlan) Can`t operate with vlans ...\n"); + return 1; + } else + VERB(3) syslog(LOG_NOTICE, "VLAN module ... %s\n", s ? "Loaded" : "Already loaded"); +#endif + + // create vlan + memset(&ifr, 0, sizeof ifr); + strlcpy(vlr.vlr_parent, szIface, IFNAMSIZ); + vlr.vlr_tag = MGMT_VTAG; + + strlcpy(ifr.ifr_name, MGMT_IFACE, IFNAMSIZ); + ifr.ifr_data = (void *) &vlr; + + s = socket(PF_INET, SOCK_DGRAM, 0); + if (-1 == s) { + syslog(LOG_ERR, "Error:: socket(INET) #%d - %s\n", errno, strerror(errno)); + closelog(); + return 2; + } +#ifdef SIOCIFCREATE2 + if (ioctl(s, SIOCIFCREATE2, &ifr) == -1 && errno != EEXIST) { +#else + if (ioctl(s, SIOCIFCREATE, &ifr) == -1 && errno != EEXIST) { +#endif + syslog(LOG_ERR, "Error:: Create interface=%s ioctl(SIOCIFCREATE2) #%d - %s\n", + MGMT_IFACE, errno, strerror(errno)); + close(s); + return 2; + } else + VERB(2) syslog(LOG_NOTICE, "Info:: Created interface=%s\n", MGMT_IFACE); + // rename iface + ifr.ifr_data = MGMT_NAME; +#ifdef SIOCSIFNAME + if (errno != EEXIST && ioctl(s, SIOCSIFNAME, &ifr) == -1) { +#else + if (errno != EEXIST && ioctl(s, SIOCGIFDESCR, &ifr) == -1) { +#endif + syslog(LOG_ERR, "Error:: Managment interface=%s ioctl(SIOCSIFNAME) #%d - %s\n", + MGMT_NAME, errno, strerror(errno)); + close(s); + return 2; + } else + VERB(2) syslog(LOG_NOTICE, "Info:: Managment interface=%s\n", MGMT_NAME); + + // assign address & up + memset(&ifra, 0, sizeof ifra); + strlcpy(ifra.ifra_name, MGMT_NAME, IFNAMSIZ); + sin = (struct sockaddr_in*) &ifra.ifra_addr; + sin->sin_len = sizeof ifra.ifra_addr; + sin->sin_family = AF_INET; + sin->sin_addr.s_addr = inet_addr(MGMT_ADDR); + sin = (struct sockaddr_in*) &ifra.ifra_mask; + sin->sin_len = sizeof ifra.ifra_mask; + sin->sin_family = AF_INET; + sin->sin_addr.s_addr = inet_addr(MGMT_MASK); + if (ioctl(s, SIOCAIFADDR, &ifra) == -1) { + syslog(LOG_ERR, "Error:: IP %s ioctl(SIOCAIFADDR) #%d - %s\n", + MGMT_ADDR, errno, strerror(errno)); + close(s); + return 2; + } else + VERB(2) syslog(LOG_NOTICE, "Info:: IP %s\n", MGMT_ADDR); + + close(s); + closelog(); + return 0; +}