From a0b6e513b1625adb3e05092c7d22482897a15e25 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Axel=20D=C3=B6rfler?= <axeld@pinc-software.de>
Date: Sun, 9 Mar 2008 15:23:58 +0000
Subject: [PATCH] Added a program that tests how getpeername() behaves.

git-svn-id: file:///srv/svn/repos/haiku/haiku/trunk@24324 a95241bf-73f2-0310-859d-f6bbb57e9c96
---
 src/tests/kits/net/Jamfile         |  2 +
 src/tests/kits/net/getpeername.cpp | 74 ++++++++++++++++++++++++++++++
 2 files changed, 76 insertions(+)
 create mode 100644 src/tests/kits/net/getpeername.cpp

diff --git a/src/tests/kits/net/Jamfile b/src/tests/kits/net/Jamfile
index 2e41d915be..2536cff692 100644
--- a/src/tests/kits/net/Jamfile
+++ b/src/tests/kits/net/Jamfile
@@ -9,6 +9,8 @@ SimpleTest udp_server : udp_server.c : $(TARGET_NETWORK_LIBS) ;
 SimpleTest tcp_server : tcp_server.c : $(TARGET_NETWORK_LIBS) ;
 SimpleTest tcp_client : tcp_client.c : $(TARGET_NETWORK_LIBS) ;
 
+SimpleTest getpeername : getpeername.cpp : $(TARGET_NETWORK_LIBS) ;
+
 SubInclude HAIKU_TOP src tests kits net DialUpPreflet ;
 SubInclude HAIKU_TOP src tests kits net multicast ;
 SubInclude HAIKU_TOP src tests kits net netperf ;
diff --git a/src/tests/kits/net/getpeername.cpp b/src/tests/kits/net/getpeername.cpp
new file mode 100644
index 0000000000..b3f477a760
--- /dev/null
+++ b/src/tests/kits/net/getpeername.cpp
@@ -0,0 +1,74 @@
+/*
+ * Copyright 2008, Axel Dörfler, axeld@pinc-software.de.
+ * Distributed under the terms of the MIT license.
+ */
+
+
+#include <errno.h>
+#include <netdb.h>
+#include <netinet/in.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/socket.h>
+#include <unistd.h>
+
+
+extern const char* __progname;
+
+
+int
+main(int argc, char** argv)
+{
+	if (argc < 2) {
+		printf("usage: %s <host> [port]\n"
+			"Connects to the host (default port 21, ftp), and calls\n"
+			"getpeername() to see if it works correctly with a short\n"
+			"buffer.\n", __progname);
+		return 0;
+	}
+
+	struct hostent* host = gethostbyname(argv[1]);
+	if (host == NULL) {
+		perror("gethostbyname");
+		return 1;
+	}
+
+	uint16_t port = 21;
+	if (argc > 2)
+		port = atol(argv[2]);
+
+	int socket = ::socket(AF_INET, SOCK_STREAM, 0);
+	if (socket < 0) {
+		perror("socket");
+		return 1;
+	}
+
+	sockaddr_in address;
+	memset(&address, 0, sizeof(sockaddr_in));
+
+	address.sin_family = AF_INET;
+	address.sin_port = htons(port);
+	address.sin_addr = *((struct in_addr*)host->h_addr);
+
+	socklen_t length = 14;
+	sockaddr buffer;
+	int result = getpeername(socket, &buffer, &length);
+	printf("getpeername() before connect(): %d (%s), length: %d\n", result,
+		strerror(errno), length);
+
+	if (connect(socket, (sockaddr*)&address, sizeof(sockaddr_in)) < 0) {
+		perror("connect");
+		return 1;
+	}
+
+	errno = 0;
+	length = 14;
+	result = getpeername(socket, &buffer, &length);
+	printf("getpeername() after connect(): %d (%s), length: %d\n", result,
+		strerror(errno), length);
+
+	close(socket);
+	return 0;
+}
+