Diff for /embedtools/src/get1steth.c between versions 1.1 and 1.2

version 1.1, 2009/12/18 14:12:24 version 1.2, 2011/06/08 12:45:41
Line 0 Line 1
   /*************************************************************************
    * (C) 2010 AITNET - Sofia/Bulgaria - <office@aitbg.com>
    *  by Michael Pounov <misho@aitbg.com>
    *
    * $Author$
    * $Id$
    *
    *************************************************************************/
   #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;
   }

Removed from v.1.1  
changed lines
  Added in v.1.2


FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>