1/*
2 * CDDL HEADER START
3 *
4 * The contents of this file are subject to the terms
5 * of the Common Development and Distribution License
6 * (the "License").  You may not use this file except
7 * in compliance with the License.
8 *
9 * You can obtain a copy of the license at
10 * src/OPENSOLARIS.LICENSE
11 * or http://www.opensolaris.org/os/licensing.
12 * See the License for the specific language governing
13 * permissions and limitations under the License.
14 *
15 * When distributing Covered Code, include this CDDL
16 * HEADER in each file and include the License file at
17 * usr/src/OPENSOLARIS.LICENSE.  If applicable,
18 * add the following below this CDDL HEADER, with the
19 * fields enclosed by brackets "[]" replaced with your
20 * own identifying information: Portions Copyright [yyyy]
21 * [name of copyright owner]
22 *
23 * CDDL HEADER END
24 */
25
26/*
27 * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
28 * Use is subject to license terms.
29 */
30
31/*
32 * getpeername test
33 */
34
35
36#include <sys/types.h>
37#include <sys/socket.h>
38#include <netinet/in.h>
39#include <netinet/tcp.h>
40#include <arpa/inet.h>
41#include <netdb.h>
42#include <string.h>
43#include <unistd.h>
44#include <stdlib.h>
45#include <stdio.h>
46#include <fcntl.h>
47#include <errno.h>
48
49#include "libmicro.h"
50
51#define	FIRSTPORT		12345
52
53static int			sock = -1;
54
55int
56benchmark_init()
57{
58	(void) sprintf(lm_usage, "notes: measures getpeername()\n");
59	lm_tsdsize = 0;
60
61	return (0);
62}
63
64int
65benchmark_initrun()
66{
67	int			j;
68	int			opt = 1;
69	int			result;
70	socklen_t		size;
71	struct hostent	*host;
72	struct sockaddr_in	adds;
73	int			sock2, sock3;
74
75	sock2 = socket(AF_INET, SOCK_STREAM, 0);
76	if (sock2 == -1) {
77		perror("socket");
78		exit(1);
79	}
80
81	if (setsockopt(sock2, SOL_SOCKET, SO_REUSEADDR,
82	    &opt, sizeof (int)) == -1) {
83		perror("setsockopt");
84		exit(1);
85	}
86
87	if ((host = gethostbyname("localhost")) == NULL) {
88		perror("gethostbyname");
89		exit(1);
90	}
91
92	j = FIRSTPORT;
93	for (;;) {
94		(void) memset(&adds, 0, sizeof (struct sockaddr_in));
95		adds.sin_family = AF_INET;
96		adds.sin_port = htons(j++);
97		(void) memcpy(&adds.sin_addr.s_addr, host->h_addr_list[0],
98		    sizeof (struct in_addr));
99
100		if (bind(sock2, (struct sockaddr *)&adds,
101		    sizeof (struct sockaddr_in)) == 0) {
102			break;
103		}
104
105		if (errno != EADDRINUSE) {
106			perror("bind");
107			exit(1);
108		}
109	}
110
111	if (listen(sock2, 5) == -1) {
112		perror("listen");
113		exit(1);
114	}
115
116	sock3 = socket(AF_INET, SOCK_STREAM, 0);
117	if (sock3 == -1) {
118		perror("socket");
119		exit(1);
120	}
121
122	if (fcntl(sock3, F_SETFL, O_NDELAY) == -1) {
123		perror("fcntl");
124		exit(1);
125	}
126
127	result = connect(sock3, (struct sockaddr *)&adds,
128	    sizeof (struct sockaddr_in));
129	if ((result == -1) && (errno != EINPROGRESS)) {
130		perror("connect");
131		exit(1);
132	}
133
134	size = sizeof (struct sockaddr);
135	sock = accept(sock2, (struct sockaddr *)&adds, &size);
136	if (sock == -1) {
137		perror("accept");
138		exit(1);
139	}
140
141	return (0);
142}
143
144/*ARGSUSED*/
145int
146benchmark(void *tsd, result_t *res)
147{
148	int			i;
149	struct sockaddr_in	adds;
150	socklen_t		size;
151
152	for (i = 0; i < lm_optB; i++) {
153		size = sizeof (struct sockaddr_in);
154		if (getpeername(sock, (struct sockaddr *)&adds, &size) == -1) {
155			perror("getpeername");
156			exit(1);
157			res->re_errors++;
158		}
159	}
160	res->re_count = i;
161
162	return (0);
163}
164