540f4e2c38036d57871140cf7c4eadacb8075143
[bbaumbach/samba-autobuild/.git] / source4 / nbt_server / wins / wins_dns_proxy.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    wins server dns proxy
5
6    Copyright (C) Stefan Metzmacher      2005
7       
8    This program is free software; you can redistribute it and/or modify
9    it under the terms of the GNU General Public License as published by
10    the Free Software Foundation; either version 2 of the License, or
11    (at your option) any later version.
12    
13    This program is distributed in the hope that it will be useful,
14    but WITHOUT ANY WARRANTY; without even the implied warranty of
15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16    GNU General Public License for more details.
17    
18    You should have received a copy of the GNU General Public License
19    along with this program; if not, write to the Free Software
20    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21 */
22
23 #include "includes.h"
24 #include "nbt_server/nbt_server.h"
25 #include "nbt_server/wins/winsdb.h"
26 #include "nbt_server/wins/winsserver.h"
27 #include "system/time.h"
28 #include "libcli/composite/composite.h"
29 #include "smbd/service_task.h"
30
31 struct wins_dns_proxy_state {
32         struct nbt_name_socket *nbtsock;
33         struct nbt_name_packet *packet;
34         struct socket_address *src;
35 };
36
37 static void nbtd_wins_dns_proxy_handler(struct composite_context *creq)
38 {
39         NTSTATUS status;
40         struct wins_dns_proxy_state *s = talloc_get_type(creq->async.private_data,
41                                                          struct wins_dns_proxy_state);
42         struct nbt_name *name = &s->packet->questions[0].name;
43         const char *address;
44         const char **addresses;
45         uint16_t nb_flags = 0; /* TODO: ... */
46
47         status = resolve_name_recv(creq, s->packet, &address);
48         if (!NT_STATUS_IS_OK(status)) {
49                 goto notfound;
50         }
51
52         addresses = str_list_add(NULL, address);
53         talloc_steal(s->packet, addresses);
54         if (!addresses) goto notfound;
55
56         nbtd_name_query_reply(s->nbtsock, s->packet, s->src, name, 
57                               0, nb_flags, addresses);
58         return;
59 notfound:
60         nbtd_negative_name_query_reply(s->nbtsock, s->packet, s->src);
61 }
62
63 /*
64   dns proxy query a name
65 */
66 void nbtd_wins_dns_proxy_query(struct nbt_name_socket *nbtsock, 
67                                struct nbt_name_packet *packet, 
68                                struct socket_address *src)
69 {
70         struct nbt_name *name = &packet->questions[0].name;
71         struct nbtd_interface *iface = talloc_get_type(nbtsock->incoming.private,
72                                                        struct nbtd_interface);
73         struct wins_dns_proxy_state *s;
74         struct composite_context *creq;
75         const char *methods[] = {
76                 "host",
77                 NULL
78         };
79
80         s = talloc(nbtsock, struct wins_dns_proxy_state);
81         if (!s) goto failed;
82         s->nbtsock      = nbtsock;
83         s->packet       = talloc_steal(s, packet);
84         s->src          = src;
85         if (!talloc_reference(s, src)) {
86                 goto failed;
87         }
88
89         creq = resolve_name_send(name, iface->nbtsrv->task->event_ctx, methods);
90         if (!creq) goto failed;
91
92         creq->async.fn          = nbtd_wins_dns_proxy_handler;
93         creq->async.private_data= s;
94         return;
95 failed:
96         nbtd_negative_name_query_reply(nbtsock, packet, src);
97 }