summaryrefslogtreecommitdiff
path: root/source3/torture/test_addrchange.c
blob: 9ccca1c6c5194ca04ec1d78e85b49954e22c318f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
/*
   Unix SMB/CIFS implementation.
   test for the addrchange functionality
   Copyright (C) Volker Lendecke 2011

   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, see <http://www.gnu.org/licenses/>.
*/

#include "includes.h"
#include "lib/addrchange.h"
#include "lib/util/tevent_ntstatus.h"
#include "proto.h"

extern int torture_numops;

bool run_addrchange(int dummy)
{
	struct addrchange_context *ctx;
	struct tevent_context *ev;
	NTSTATUS status;
	int i;

	ev = samba_tevent_context_init(talloc_tos());
	if (ev == NULL) {
		d_fprintf(stderr, "tevent_context_init failed\n");
		return false;
	}

	status = addrchange_context_create(talloc_tos(), &ctx);
	if (!NT_STATUS_IS_OK(status)) {
		d_fprintf(stderr, "addrchange_context_create failed: %s\n",
			  nt_errstr(status));
		return false;
	}

	for (i=0; i<torture_numops; i++) {
		enum addrchange_type type;
		struct sockaddr_storage addr;
		struct tevent_req *req;
		const char *typestr;
		char addrstr[INET6_ADDRSTRLEN];

		req = addrchange_send(talloc_tos(), ev, ctx);
		if (req == NULL) {
			d_fprintf(stderr, "addrchange_send failed\n");
			return false;
		}

		if (!tevent_req_poll_ntstatus(req, ev, &status)) {
			d_fprintf(stderr, "tevent_req_poll_ntstatus failed: "
				  "%s\n", nt_errstr(status));
			TALLOC_FREE(req);
			return false;
		}

		status = addrchange_recv(req, &type, &addr);
		TALLOC_FREE(req);
		if (!NT_STATUS_IS_OK(status)) {
			d_fprintf(stderr, "addrchange_recv failed: %s\n",
				  nt_errstr(status));
			return false;
		}

		switch(type) {
		case ADDRCHANGE_ADD:
			typestr = "add";
			break;
		case ADDRCHANGE_DEL:
			typestr = "del";
			break;
		default:
			typestr = talloc_asprintf(talloc_tos(), "unknown %d",
						  (int)type);
			break;
		}

		printf("Got %s %s\n", typestr,
		       print_sockaddr(addrstr, sizeof(addrstr), &addr));
	}
	TALLOC_FREE(ctx);
	TALLOC_FREE(ev);
	return true;
}