sbin/rconfig/client.c

   1 /*
   2  * RCONFIG/CLIENT.C
   3  *
   4  * $DragonFly: src/sbin/rconfig/client.c,v 1.1 2004/06/18 02:46:46 dillon Exp $
   5  */
   6
   7 #include "defs.h"
   8
   9 #define LONG_ALIGN(n)   (((n) + (sizeof(long) - 1)) & ~(sizeof(long) - 1))
  10
  11 static void load_client_broadcast_tags(tag_t tag, const char *tagName);
  12
  13 void
  14 doClient(void)
  15 {
  16     int done = 0;
  17     tag_t tag;
  18
  19     /*
  20      * The server list is in the form host[:tag]
  21      */
  22     for (tag = AddrBase; tag && !done; tag = tag->next) {
  23         struct sockaddr_in sain;
  24         struct sockaddr_in rsin;
  25         char *tagName;
  26         char *host = NULL;
  27         char *res = NULL;
  28         char *buf = NULL;
  29         int len;
  30         int ufd = -1;
  31         FILE *fi = NULL;
  32         FILE *fo = NULL;
  33         int rc;
  34
  35         bzero(&sain, sizeof(sain));
  36         if (tag->name == NULL) {
  37             load_client_broadcast_tags(tag, "auto");
  38             continue;
  39         }
  40         if (tag->name[0] == ':') {
  41             load_client_broadcast_tags(tag, tag->name + 1);
  42             continue;
  43         }
  44         host = strdup(tag->name);
  45         if ((tagName = strchr(host, ':')) != NULL)
  46             *tagName++ = 0;
  47         else
  48             tagName = "auto";
  49         tagName = strdup(tagName);
  50         if (inet_aton(host, &sain.sin_addr) == 0) {
  51             struct hostent *hp;
  52             if ((hp = gethostbyname2(host, AF_INET)) == NULL) {
  53                 fprintf(stderr, "Unable to resolve %s\n", host);
  54                 exit(1);
  55             }
  56             bcopy(hp->h_addr_list[0], &sain.sin_addr, hp->h_length);
  57             free(host);
  58             host = strdup(hp->h_name);
  59             endhostent();
  60         }
  61         sain.sin_port = htons(257);
  62         sain.sin_len = sizeof(sain);
  63         sain.sin_family = AF_INET;
  64
  65         /*
  66          * Do a couple of UDP transactions to locate the tag on the server.
  67          */
  68         printf("%s:%s - ", host, tagName);
  69         fflush(stdout);
  70         rc = udp_transact(&sain, &rsin, &ufd, &res, &len, "TAG %s\r\n", tagName);
  71         if (rc != 101 || res == NULL) {
  72             printf("NO LUCK %s\n", (res ? res : ""));
  73         } else {
  74             printf("%s -", res);
  75             fflush(stdout);
  76             rc = tcp_transact(&rsin, &fi, &fo, &buf, &len, "TAG %s\r\n", tagName);
  77             if (rc == 201 && buf) {
  78                 int ffd;
  79                 char *path;
  80
  81                 asprintf(&path, "%s/%s.sh", WorkDir, tagName);
  82                 ffd = open(path, O_CREAT|O_TRUNC|O_RDWR, 0755);
  83                 if (ffd >= 0 && write(ffd, buf, len) == len) {
  84                     printf("running %s [%d] in", path, len);
  85                     close(ffd);
  86                     ffd = -1;
  87                     for (rc = 5; rc > 0; --rc) {
  88                         printf(" %d", rc);
  89                         fflush(stdout);
  90                         sleep(1);
  91                     }
  92                     printf(" 0\n");
  93                     fflush(stdout);
  94                     rc = system(path);
  95                     if (rc)
  96                         printf("rconfig script exit code %d\n", rc);
  97                     done = 1;
  98                 } else {
  99                     if (ffd >= 0) {
 100                         remove(path);
 101                         close(ffd);
 102                         ffd = -1;
 103                     }
 104                     printf(" unable to create %s [%d] - DOWNLOAD FAILED\n",
 105                             path, len);
 106                 }
 107             } else {
 108                 printf(" DOWNLOAD FAILED\n");
 109             }
 110         }
 111         if (ufd >= 0) {
 112             close(ufd);
 113             ufd = -1;
 114         }
 115         if (fi != NULL) {
 116             fclose(fi);
 117             fi = NULL;
 118         }
 119         if (fo != NULL) {
 120             fclose(fo);
 121             fo = NULL;
 122         }
 123         if (buf)
 124             free(buf);
 125         if (res)
 126             free(res);
 127         free(host);
 128         free(tagName);
 129     }
 130 }
 131
 132 static
 133 void
 134 load_client_broadcast_tags(tag_t tag, const char *tagName)
 135 {
 136     struct sockaddr_dl *sdl;
 137     struct if_msghdr *ifm;
 138     int mib[6];
 139     char *buf;
 140     int bytes;
 141     int i;
 142
 143     mib[0] = CTL_NET;
 144     mib[1] = PF_ROUTE;
 145     mib[2] = 0;
 146     mib[3] = AF_INET;
 147     mib[4] = NET_RT_IFLIST;
 148     mib[5] = 0;
 149
 150     if (sysctl(mib, 6, NULL, &bytes, NULL, 0) < 0) {
 151         printf("no interfaces!\n");
 152         exit(1);
 153     }
 154     buf = malloc(bytes);
 155     if (sysctl(mib, 6, buf, &bytes, NULL, 0) < 0) {
 156         printf("no interfaces!\n");
 157         exit(1);
 158     }
 159     ifm = (void *)buf;
 160     sdl = NULL;
 161     while ((char *)ifm < buf + bytes && ifm->ifm_msglen) {
 162         switch(ifm->ifm_type) {
 163         case RTM_IFINFO:
 164             if (ifm->ifm_flags & IFF_UP) {
 165                 sdl = (void *)(ifm + 1);
 166             } else {
 167                 sdl = NULL;
 168             }
 169             break;
 170         case RTM_NEWADDR:
 171             if (sdl) {
 172                 struct sockaddr_in *sain;
 173                 struct ifa_msghdr *ifam;
 174                 char *scan;
 175                 char *name;
 176                 tag_t ntag;
 177
 178                 ifam = (void *)ifm;
 179                 scan = (char *)(ifam + 1);
 180                 for (i = 0; i < RTAX_MAX; ++i) {
 181                     if ((1 << i) & ifam->ifam_addrs) {
 182                         sain = (void *)scan;
 183                         if (i == RTAX_BRD) {
 184                             asprintf(&name, "%s:%s",
 185                                     inet_ntoa(sain->sin_addr), tagName);
 186                             ntag = calloc(sizeof(struct tag), 1);
 187                             ntag->name = name;
 188                             ntag->flags = 0;
 189                             ntag->next = tag->next;
 190                             tag->next = ntag;
 191                             tag = ntag;
 192                             if (VerboseOpt)
 193                                 printf("add: %s (%s)\n", sdl->sdl_data, tag->name);
 194                         }
 195                         scan = scan + LONG_ALIGN(sain->sin_len);
 196                     }
 197                 }
 198             }
 199             break;
 200         }
 201         ifm = (void *)((char *)ifm + ifm->ifm_msglen);
 202     }
 203 }
 204