lib/names-net.c

   1 /*
   2  *      The PCI Library -- Resolving ID's via DNS
   3  *
   4  *      Copyright (c) 2007--2008 Martin Mares <mj@ucw.cz>
   5  *
   6  *      Can be freely distributed and used under the terms of the GNU GPL.
   7  */
   8
   9 #include <string.h>
  10 #include <stdlib.h>
  11
  12 #include "internal.h"
  13 #include "names.h"
  14
  15 #ifdef PCI_USE_DNS
  16
  17 #include <netinet/in.h>
  18 #include <arpa/nameser.h>
  19 #include <resolv.h>
  20
  21 char
  22 *pci_id_net_lookup(struct pci_access *a, int cat, int id1, int id2, int id3, int id4)
  23 {
  24   char name[256], dnsname[256], txt[256];
  25   byte answer[4096];
  26   const byte *data;
  27   int res, i, j, dlen;
  28   ns_msg m;
  29   ns_rr rr;
  30
  31   switch (cat)
  32     {
  33     case ID_VENDOR:
  34       sprintf(name, "%04x", id1);
  35       break;
  36     case ID_DEVICE:
  37       sprintf(name, "%04x.%04x", id2, id1);
  38       break;
  39     case ID_SUBSYSTEM:
  40       sprintf(name, "%04x.%04x.%04x.%04x", id4, id3, id2, id1);
  41       break;
  42     case ID_GEN_SUBSYSTEM:
  43       sprintf(name, "%04x.%04x.s", id2, id1);
  44       break;
  45     case ID_CLASS:
  46       sprintf(name, "%02x.c", id1);
  47       break;
  48     case ID_SUBCLASS:
  49       sprintf(name, "%02x.%02x.c", id2, id1);
  50       break;
  51     case ID_PROGIF:
  52       sprintf(name, "%02x.%02x.%02x.c", id3, id2, id1);
  53       break;
  54     default:
  55       return NULL;
  56     }
  57   sprintf(dnsname, "%s.%s", name, a->id_domain);
  58
  59   a->debug("Resolving %s\n", dnsname);
  60   res_init();
  61   res = res_query(dnsname, ns_c_in, ns_t_txt, answer, sizeof(answer));
  62   if (res < 0)
  63     {
  64       a->debug("\tfailed, h_errno=%d\n", _res.res_h_errno);
  65       return NULL;
  66     }
  67   if (ns_initparse(answer, res, &m) < 0)
  68     {
  69       a->debug("\tinitparse failed\n");
  70       return NULL;
  71     }
  72   for (i=0; ns_parserr(&m, ns_s_an, i, &rr) >= 0; i++)
  73     {
  74       a->debug("\tanswer %d (class %d, type %d)\n", i, ns_rr_class(rr), ns_rr_type(rr));
  75       if (ns_rr_class(rr) != ns_c_in || ns_rr_type(rr) != ns_t_txt)
  76         continue;
  77       data = ns_rr_rdata(rr);
  78       dlen = ns_rr_rdlen(rr);
  79       j = 0;
  80       while (j < dlen && j+1+data[j] <= dlen)
  81         {
  82           memcpy(txt, &data[j+1], data[j]);
  83           txt[data[j]] = 0;
  84           j += 1+data[j];
  85           a->debug("\t\t%s\n", txt);
  86           if (txt[0] == 'i' && txt[1] == '=')
  87             return strdup(txt+2);
  88         }
  89     }
  90
  91   return NULL;
  92 }
  93
  94 #else
  95
  96 char *pci_id_net_lookup(struct pci_access *a UNUSED, int cat UNUSED, int id1 UNUSED, int id2 UNUSED, int id3 UNUSED, int id4 UNUSED)
  97 {
  98   return NULL;
  99 }
 100
 101 #endif
 102
 103 void
 104 pci_set_net_domain(struct pci_access *a, char *name, int to_be_freed)
 105 {
 106   if (a->free_id_domain)
 107     free(a->id_domain);
 108   a->id_domain = name;
 109   a->free_id_domain = to_be_freed;
 110 }