1 files changed, 88 insertions, 0 deletions

diff --git a/src/mt/lookup.cc b/src/mt/lookup.cc
new file mode 100644
index 0000000..c0e72c8
--- /dev/null
+++ b/src/mt/lookup.cc
@@ -0,0 +1,88 @@
+/*************************************************************************
+ *
+ * HTCd - Copyright (C) 1998-2006 Henrik Rydberg
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+ */
+
+#include <mt/lookup.h>
+#include <mt/mvec.h>
+#include <mt/lock.h>
+#include <arpa/inet.h>
+#include <netinet/in.h>
+#include <sys/socket.h>
+
+const time_t VALIDSEC=120;
+static msvec<IPnode> cache;
+static mutex_t mutex;
+
+int Lookup(IPnode& node,const in_addr& ia)
+{
+  MLOCK(mutex);
+  IPnode* p=cache.begin();
+  time_t now=time(0);
+  while(p<cache.end()) {
+    if(now>p->valid) cache.erase(p);
+    else if(memcmp(&p->addr,&ia,sizeof(ia))==0) {p->valid=now+VALIDSEC;node=*p;return 1;}
+    else p++;
+  }
+  hostent hent;
+  sref_t buf;
+  int herr;
+  hostent* hp=0;
+  gethostbyaddr_r((const char*)&ia,sizeof(ia),AF_INET,
+                  &hent,
+                  buf,sizeof(sref_t),
+                  &hp,
+                  &herr);
+  if(!hp) return 0;
+  memcpy(&node.addr,hp->h_addr,sizeof(hp->h_length));
+  node.ip=inet_ntoa(node.addr);
+  node.name=hp->h_name;
+  node.valid=now+VALIDSEC;
+  cache.push_back(node);
+  return 1;
+}
+
+int Lookup(IPnode& node,const sref& s)
+{
+  MLOCK(mutex);
+  IPnode* p=cache.begin();
+  time_t now=time(0);
+  while(p<cache.end()) {
+    if(now>p->valid) cache.erase(p);
+    else if(p->ip==s||p->name==s) { p->valid=now+VALIDSEC; node=*p; return 1; }
+    else p++;
+  }
+  hostent hent;
+  sref_t sbuf,buf;
+  int herr;
+  hostent* hp=0;
+  for(int i=0;s[i];i++) if(s[i]!='.'&&(s[i]<'0'||s[i]>'9')) {
+    gethostbyname_r(s.copyto(sbuf),&hent,buf,sizeof(sref_t),&hp,&herr);
+    break;
+  }
+  if(!hp) {
+    unsigned long a=inet_addr(s.copyto(buf));
+    gethostbyaddr_r((char*)&a,sizeof(a),AF_INET,&hent,buf,sizeof(sref_t),&hp,&herr);
+  }
+  if(!hp) return 0;
+  memcpy(&node.addr,hp->h_addr,sizeof(hp->h_length));
+  node.ip=inet_ntoa(node.addr);
+  node.name=hp->h_name;
+  node.valid=now+VALIDSEC;
+  cache.push_back(node);
+  return 1;
+}