* Port locate_host() to IPv6

[citadel.git] / webcit / locate_host.c

diff --git a/webcit/locate_host.c b/webcit/locate_host.c
index 923a68165ad003c1e600ace51e6c9e2b29090220..165081005e6bdd67e1b0ffa3f252316402852a29 100644 (file)
--- a/webcit/locate_host.c
+++ b/webcit/locate_host.c
@@ -1,42 +1,58 @@
 /*
  * $Id$
  *
- * Examine a socket and determine the name/address of the originating host.
+ * Given a socket, supply the name of the host at the other end.
+ *
+ * Copyright (c) 1996-2010 by the citadel.org team
+ *
+ * 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 3 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  */
 
-
-#include <ctype.h>
-#include <stdlib.h>
-#include <unistd.h>
-#include <stdio.h>
-#include <fcntl.h>
-#include <signal.h>
-#include <sys/types.h>
-#include <sys/wait.h>
-#include <sys/socket.h>
-#include <sys/time.h>
-#include <limits.h>
-#include <netinet/in.h>
-#include <netdb.h>
-#include <string.h>
-#include <pwd.h>
-#include <errno.h>
-#include <stdarg.h>
-#include <pthread.h>
-#include <signal.h>
 #include "webcit.h"
 
-void locate_host(char *tbuf, int client_socket)
+
+#ifdef CTDL_IPV6
+
+void locate_host(StrBuf *tbuf, int client_socket)
+{
+       struct sockaddr_in6 clientaddr;
+       unsigned int addrlen = sizeof(clientaddr);
+       char str[256];
+
+       getpeername(client_socket, (struct sockaddr *)&clientaddr, &addrlen);
+       if(inet_ntop(AF_INET6, &clientaddr.sin6_addr, str, sizeof(str))) {
+               StrBufAppendBufPlain(tbuf, str, -1, 0);
+       }
+       else {
+               StrBufAppendBufPlain(tbuf, HKEY("<unknown>"), 0);
+       }
+}
+
+#else /* CTDL_IPV6 */
+
+void locate_host(StrBuf *tbuf, int client_socket)
 {
        struct sockaddr_in cs;
        struct hostent *ch;
-       int len;
+       socklen_t len;
        char *i;
        int a1, a2, a3, a4;
 
        len = sizeof(cs);
        if (getpeername(client_socket, (struct sockaddr *) &cs, &len) < 0) {
-               strcpy(tbuf, "<unknown>");
+               StrBufAppendBufPlain(tbuf, HKEY("<unknown>"), 0);
                return;
        }
        if ((ch = gethostbyaddr((char *) &cs.sin_addr, sizeof(cs.sin_addr),
@@ -46,8 +62,10 @@ void locate_host(char *tbuf, int client_socket)
                a2 = ((*i++) & 0xff);
                a3 = ((*i++) & 0xff);
                a4 = ((*i++) & 0xff);
-               sprintf(tbuf, "%d.%d.%d.%d", a1, a2, a3, a4);
+               StrBufPrintf(tbuf, "%d.%d.%d.%d", a1, a2, a3, a4);
                return;
        }
-       safestrncpy(tbuf, ch->h_name, 64);
+       StrBufAppendBufPlain(tbuf, ch->h_name, -1, 0);
 }
+
+#endif /* CTDL_IPV6 */