| 1 | /* | 
|---|
| 2 | Unix SMB/CIFS implementation. | 
|---|
| 3 |  | 
|---|
| 4 | a raw async NBT DGRAM library | 
|---|
| 5 |  | 
|---|
| 6 | Copyright (C) Andrew Tridgell 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 3 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, see <http://www.gnu.org/licenses/>. | 
|---|
| 20 | */ | 
|---|
| 21 |  | 
|---|
| 22 | #include "../libcli/netlogon.h" | 
|---|
| 23 |  | 
|---|
| 24 | /* | 
|---|
| 25 | a datagram name request | 
|---|
| 26 | */ | 
|---|
| 27 | struct nbt_dgram_request { | 
|---|
| 28 | struct nbt_dgram_request *next, *prev; | 
|---|
| 29 |  | 
|---|
| 30 | /* where to send the request */ | 
|---|
| 31 | struct socket_address *dest; | 
|---|
| 32 |  | 
|---|
| 33 | /* the encoded request */ | 
|---|
| 34 | DATA_BLOB encoded; | 
|---|
| 35 | }; | 
|---|
| 36 |  | 
|---|
| 37 | /* | 
|---|
| 38 | context structure for operations on dgram packets | 
|---|
| 39 | */ | 
|---|
| 40 | struct nbt_dgram_socket { | 
|---|
| 41 | struct socket_context *sock; | 
|---|
| 42 | struct tevent_context *event_ctx; | 
|---|
| 43 | struct smb_iconv_convenience *iconv_convenience; | 
|---|
| 44 |  | 
|---|
| 45 | /* the fd event */ | 
|---|
| 46 | struct tevent_fd *fde; | 
|---|
| 47 |  | 
|---|
| 48 | /* a queue of outgoing requests */ | 
|---|
| 49 | struct nbt_dgram_request *send_queue; | 
|---|
| 50 |  | 
|---|
| 51 | /* a list of mailslot handlers */ | 
|---|
| 52 | struct dgram_mailslot_handler *mailslot_handlers; | 
|---|
| 53 |  | 
|---|
| 54 | /* what to do with incoming request packets */ | 
|---|
| 55 | struct { | 
|---|
| 56 | void (*handler)(struct nbt_dgram_socket *, struct nbt_dgram_packet *, | 
|---|
| 57 | struct socket_address *src); | 
|---|
| 58 | void *private_data; | 
|---|
| 59 | } incoming; | 
|---|
| 60 | }; | 
|---|
| 61 |  | 
|---|
| 62 |  | 
|---|
| 63 | /* | 
|---|
| 64 | the mailslot code keeps a list of mailslot handlers. A mailslot | 
|---|
| 65 | handler is a function that receives incoming packets for a specific | 
|---|
| 66 | mailslot name. When a caller needs to send a mailslot and wants to | 
|---|
| 67 | get a reply then it needs to register itself as listening for | 
|---|
| 68 | incoming packets on the reply mailslot | 
|---|
| 69 | */ | 
|---|
| 70 |  | 
|---|
| 71 | typedef void (*dgram_mailslot_handler_t)(struct dgram_mailslot_handler *, | 
|---|
| 72 | struct nbt_dgram_packet *, | 
|---|
| 73 | struct socket_address *src); | 
|---|
| 74 |  | 
|---|
| 75 | struct dgram_mailslot_handler { | 
|---|
| 76 | struct dgram_mailslot_handler *next, *prev; | 
|---|
| 77 |  | 
|---|
| 78 | struct nbt_dgram_socket *dgmsock; | 
|---|
| 79 | const char *mailslot_name; | 
|---|
| 80 |  | 
|---|
| 81 | dgram_mailslot_handler_t handler; | 
|---|
| 82 | void *private_data; | 
|---|
| 83 | }; | 
|---|
| 84 |  | 
|---|
| 85 |  | 
|---|
| 86 | /* prototypes */ | 
|---|
| 87 | NTSTATUS nbt_dgram_send(struct nbt_dgram_socket *dgmsock, | 
|---|
| 88 | struct nbt_dgram_packet *packet, | 
|---|
| 89 | struct socket_address *dest); | 
|---|
| 90 | NTSTATUS dgram_set_incoming_handler(struct nbt_dgram_socket *dgmsock, | 
|---|
| 91 | void (*handler)(struct nbt_dgram_socket *, | 
|---|
| 92 | struct nbt_dgram_packet *, | 
|---|
| 93 | struct socket_address *), | 
|---|
| 94 | void *private_data); | 
|---|
| 95 | struct nbt_dgram_socket *nbt_dgram_socket_init(TALLOC_CTX *mem_ctx, | 
|---|
| 96 | struct tevent_context *event_ctx, | 
|---|
| 97 | struct smb_iconv_convenience *); | 
|---|
| 98 |  | 
|---|
| 99 | const char *dgram_mailslot_name(struct nbt_dgram_packet *packet); | 
|---|
| 100 | struct dgram_mailslot_handler *dgram_mailslot_find(struct nbt_dgram_socket *dgmsock, | 
|---|
| 101 | const char *mailslot_name); | 
|---|
| 102 | struct dgram_mailslot_handler *dgram_mailslot_listen(struct nbt_dgram_socket *dgmsock, | 
|---|
| 103 | const char *mailslot_name, | 
|---|
| 104 | dgram_mailslot_handler_t handler, | 
|---|
| 105 | void *private_data); | 
|---|
| 106 | struct dgram_mailslot_handler *dgram_mailslot_temp(struct nbt_dgram_socket *dgmsock, | 
|---|
| 107 | const char *mailslot_name, | 
|---|
| 108 | dgram_mailslot_handler_t handler, | 
|---|
| 109 | void *private_data); | 
|---|
| 110 | DATA_BLOB dgram_mailslot_data(struct nbt_dgram_packet *dgram); | 
|---|
| 111 |  | 
|---|
| 112 |  | 
|---|
| 113 | NTSTATUS dgram_mailslot_send(struct nbt_dgram_socket *dgmsock, | 
|---|
| 114 | enum dgram_msg_type msg_type, | 
|---|
| 115 | const char *mailslot_name, | 
|---|
| 116 | struct nbt_name *dest_name, | 
|---|
| 117 | struct socket_address *dest, | 
|---|
| 118 | struct nbt_name *src_name, | 
|---|
| 119 | DATA_BLOB *request); | 
|---|
| 120 |  | 
|---|
| 121 | NTSTATUS dgram_mailslot_netlogon_send(struct nbt_dgram_socket *dgmsock, | 
|---|
| 122 | struct nbt_name *dest_name, | 
|---|
| 123 | struct socket_address *dest, | 
|---|
| 124 | const char *mailslot_name, | 
|---|
| 125 | struct nbt_name *src_name, | 
|---|
| 126 | struct nbt_netlogon_packet *request); | 
|---|
| 127 | NTSTATUS dgram_mailslot_netlogon_reply(struct nbt_dgram_socket *dgmsock, | 
|---|
| 128 | struct nbt_dgram_packet *request, | 
|---|
| 129 | const char *my_netbios_name, | 
|---|
| 130 | const char *mailslot_name, | 
|---|
| 131 | struct nbt_netlogon_response *reply); | 
|---|
| 132 | NTSTATUS dgram_mailslot_netlogon_parse_request(struct dgram_mailslot_handler *dgmslot, | 
|---|
| 133 | TALLOC_CTX *mem_ctx, | 
|---|
| 134 | struct nbt_dgram_packet *dgram, | 
|---|
| 135 | struct nbt_netlogon_packet *netlogon); | 
|---|
| 136 |  | 
|---|
| 137 | NTSTATUS dgram_mailslot_netlogon_parse_response(struct dgram_mailslot_handler *dgmslot, | 
|---|
| 138 | TALLOC_CTX *mem_ctx, | 
|---|
| 139 | struct nbt_dgram_packet *dgram, | 
|---|
| 140 | struct nbt_netlogon_response *netlogon); | 
|---|
| 141 |  | 
|---|
| 142 | NTSTATUS dgram_mailslot_browse_send(struct nbt_dgram_socket *dgmsock, | 
|---|
| 143 | struct nbt_name *dest_name, | 
|---|
| 144 | struct socket_address *dest, | 
|---|
| 145 | struct nbt_name *src_name, | 
|---|
| 146 | struct nbt_browse_packet *request); | 
|---|
| 147 |  | 
|---|
| 148 | NTSTATUS dgram_mailslot_browse_reply(struct nbt_dgram_socket *dgmsock, | 
|---|
| 149 | struct nbt_dgram_packet *request, | 
|---|
| 150 | const char *mailslot_name, | 
|---|
| 151 | const char *my_netbios_name, | 
|---|
| 152 | struct nbt_browse_packet *reply); | 
|---|
| 153 |  | 
|---|
| 154 | NTSTATUS dgram_mailslot_browse_parse(struct dgram_mailslot_handler *dgmslot, | 
|---|
| 155 | TALLOC_CTX *mem_ctx, | 
|---|
| 156 | struct nbt_dgram_packet *dgram, | 
|---|
| 157 | struct nbt_browse_packet *pkt); | 
|---|