--- embedtools/src/get1steth.c 2009/12/28 13:25:04 1.1.2.4 +++ embedtools/src/get1steth.c 2011/06/08 09:27:55 1.1.2.10 @@ -1,3 +1,11 @@ +/************************************************************************* + * (C) 2010 AITNET - Sofia/Bulgaria - + * by Michael Pounov + * + * $Author: misho $ + * $Id: get1steth.c,v 1.1.2.10 2011/06/08 09:27:55 misho Exp $ + * + *************************************************************************/ #include "global.h" #include "get1steth.h" @@ -17,6 +25,35 @@ static void Usage() "\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) @@ -28,6 +65,7 @@ int main(int argc, char **argv) struct ifreq ifr; struct vlanreq vlr; struct ifaliasreq ifra; + struct ifmediareq ifmr; int s; while ((ch = getopt(argc, argv, "hvg")) != -1) @@ -46,25 +84,53 @@ int main(int argc, char **argv) argc -= optind; argv += optind; - openlog("get1steth", LOG_CONS | LOG_PERROR, LOG_CONSOLE | LOG_USER); + 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) { - 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; + 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); @@ -72,6 +138,15 @@ int main(int argc, char **argv) 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); @@ -80,9 +155,10 @@ int main(int argc, char **argv) strlcpy(ifr.ifr_name, MGMT_IFACE, IFNAMSIZ); ifr.ifr_data = (void *) &vlr; - s = socket(PF_INET, SOCK_STREAM, 0); + s = socket(PF_INET, SOCK_DGRAM, 0); if (-1 == s) { - syslog(LOG_ERR, "Error:: socket(LOCAL) #%d - %s\n", errno, strerror(errno)); + syslog(LOG_ERR, "Error:: socket(INET) #%d - %s\n", errno, strerror(errno)); + closelog(); return 2; } if (ioctl(s, SIOCIFCREATE2, &ifr) == -1 && errno != EEXIST) {