WINGs/host.c

   1 /*
   2  *  WINGs WMHost function library
   3  *
   4  *  Copyright (c) 1999 Dan Pascu
   5  *
   6  *  This program is free software; you can redistribute it and/or modify
   7  *  it under the terms of the GNU General Public License as published by
   8  *  the Free Software Foundation; either version 2 of the License, or
   9  *  (at your option) any later version.
  10  *
  11  *  This program is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  *  GNU General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU General Public License
  17  *  along with this program; if not, write to the Free Software
  18  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  19  */
  20
  21
  22
  23 #include <unistd.h>
  24 #include <string.h>
  25 #include <netdb.h>
  26 #include <sys/socket.h>
  27 #include <netinet/in.h>
  28 #include <arpa/inet.h>
  29
  30 #include "WUtil.h"
  31
  32
  33 typedef struct W_Host {
  34     char   *name;
  35
  36     WMBag  *names;
  37     WMBag  *addresses;
  38     int    nameCount;
  39     int    addressCount;
  40
  41     int    refCount;
  42 } W_Host;
  43
  44
  45 static WMHashTable *hostCache = NULL;
  46 static Bool hostCacheEnabled = True;
  47
  48
  49 /* Max hostname length (RFC  1123) */
  50 #define W_MAXHOSTNAMELEN        255
  51
  52 WMHost*
  53 WMGetCurrentHost()
  54 {
  55     char name[W_MAXHOSTNAMELEN+1];
  56
  57     if (gethostname(name, W_MAXHOSTNAMELEN) < 0) {
  58         wsyserror("Cannot get current host name");
  59         return NULL;
  60     }
  61
  62     name[W_MAXHOSTNAMELEN] = '\0';
  63
  64     return WMGetHostWithName(name);
  65 }
  66
  67
  68 WMHost*
  69 WMGetHostWithName(char* name)
  70 {
  71     struct hostent *host;
  72     struct in_addr in;
  73     WMHost *hPtr;
  74     int i;
  75
  76     if (name == NULL) {
  77         wwarning("NULL host name in 'WMGetHostWithName()'");
  78         return NULL;
  79     }
  80
  81     if (!hostCache)
  82         hostCache = WMCreateHashTable(WMStringPointerHashCallbacks);
  83
  84     hPtr = WMHashGet(hostCache, name);
  85     if (hPtr) {
  86         WMRetainHost(hPtr);
  87         return hPtr;
  88     }
  89
  90     host = gethostbyname(name);
  91     if (host == NULL) {
  92         wwarning("Invalid host name/address '%s'", name);
  93         return NULL;
  94     }
  95
  96     hPtr = (WMHost*)wmalloc(sizeof(WMHost));
  97     memset(hPtr, 0, sizeof(WMHost));
  98
  99     hPtr->names = WMCreateBag(1);
 100     hPtr->addresses = WMCreateBag(1);
 101
 102     WMPutInBag(hPtr->names, wstrdup(host->h_name));
 103
 104     for (i=0; host->h_aliases[i]!=NULL; i++) {
 105         WMPutInBag(hPtr->names, wstrdup(host->h_aliases[i]));
 106     }
 107
 108     hPtr->nameCount = WMGetBagItemCount(hPtr->names);
 109
 110     for (i=0; host->h_addr_list[i]!=NULL; i++) {
 111         memcpy((void*)&in.s_addr, (const void*)host->h_addr_list[i],
 112                host->h_length);
 113         WMPutInBag(hPtr->addresses, wstrdup(inet_ntoa(in)));
 114     }
 115
 116     hPtr->addressCount = WMGetBagItemCount(hPtr->addresses);
 117
 118     hPtr->refCount = 1;
 119
 120     if (hostCacheEnabled) {
 121         hPtr->name = wstrdup(name);
 122         wassertr(WMHashInsert(hostCache, hPtr->name, hPtr)==NULL);
 123         hPtr->refCount++;
 124     }
 125
 126     return hPtr;
 127 }
 128
 129
 130 WMHost*
 131 WMGetHostWithAddress(char* address)
 132 {
 133     if (address == NULL) {
 134         wwarning("NULL address in 'WMGetHostWithAddress()'");
 135         return NULL;
 136     }
 137
 138     return WMGetHostWithName(address);
 139 #if 0
 140     hostaddr.s_addr = inet_addr((char*)[address cString]);
 141     if (hostaddr.s_addr == -1) {
 142         return NULL;
 143     }
 144
 145     h = gethostbyaddr((char*)&hostaddr, sizeof(hostaddr), AF_INET);
 146 #endif
 147 }
 148
 149
 150 WMHost*
 151 WMRetainHost(WMHost *hPtr)
 152 {
 153     hPtr->refCount++;
 154     return hPtr;
 155 }
 156
 157
 158 void
 159 WMReleaseHost(WMHost *hPtr)
 160 {
 161     int i;
 162
 163     hPtr->refCount--;
 164
 165     if (hPtr->refCount > 0)
 166         return;
 167
 168     for (i=0; i<WMGetBagItemCount(hPtr->names); i++)
 169         wfree(WMGetFromBag(hPtr->names, i));
 170     for (i=0; i<WMGetBagItemCount(hPtr->addresses); i++)
 171         wfree(WMGetFromBag(hPtr->addresses, i));
 172
 173     WMFreeBag(hPtr->names);
 174     WMFreeBag(hPtr->addresses);
 175
 176     if (hPtr->name) {
 177         WMHashRemove(hostCache, hPtr->name);
 178         wfree(hPtr->name);
 179     }
 180
 181     wfree(hPtr);
 182 }
 183
 184
 185 void
 186 WMSetHostCacheEnabled(Bool flag)
 187 {
 188     hostCacheEnabled = flag;
 189 }
 190
 191
 192 Bool
 193 WMIsHostCacheEnabled()
 194 {
 195     return hostCacheEnabled;
 196 }
 197
 198
 199 void
 200 WMFlushHostCache()
 201 {
 202     if (hostCache && WMCountHashTable(hostCache)>0) {
 203         WMBag *hostBag = WMCreateBag(WMCountHashTable(hostCache));
 204         WMHashEnumerator enumer = WMEnumerateHashTable(hostCache);
 205         WMHost *hPtr;
 206         int i;
 207
 208         while ((hPtr = WMNextHashEnumeratorItem(&enumer))) {
 209             /* we can't release the host here, because we can't change the
 210              * hash while using the enumerator functions. */
 211             WMPutInBag(hostBag, hPtr);
 212         }
 213         for (i=0; i<WMGetBagItemCount(hostBag); i++)
 214             WMReleaseHost(WMGetFromBag(hostBag, i));
 215         WMFreeBag(hostBag);
 216         WMResetHashTable(hostCache);
 217     }
 218 }
 219
 220
 221 Bool
 222 WMIsHostEqualToHost(WMHost* hPtr, WMHost* aPtr)
 223 {
 224     int i;
 225
 226     wassertrv(hPtr!=NULL && aPtr!=NULL, False);
 227
 228     if (hPtr == aPtr)
 229         return True;
 230
 231     for (i=0; i<aPtr->addressCount; i++) {
 232         if (WMGetFirstInBag(hPtr->addresses, WMGetFromBag(aPtr->addresses, i)))
 233             return True;
 234     }
 235
 236     return False;
 237 }
 238
 239
 240 char*
 241 WMGetHostName(WMHost *hPtr)
 242 {
 243     return WMGetFromBag(hPtr->names, 0);
 244 }
 245
 246
 247 WMBag*
 248 WMGetHostNames(WMHost *hPtr)
 249 {
 250     return hPtr->names;
 251 }
 252
 253
 254 char*
 255 WMGetHostAddress(WMHost *hPtr)
 256 {
 257     return (hPtr->addressCount > 0 ? WMGetFromBag(hPtr->addresses, 0) : NULL);
 258 }
 259
 260
 261 WMBag*
 262 WMGetHostAddresses(WMHost *hPtr)
 263 {
 264     return hPtr->addresses;
 265 }
 266
 267
 268
 269