LCOV - code coverage report
Current view: top level - source4/libcli/dgram - netlogon.c (source / functions) Hit Total Coverage
Test: coverage report for abartlet/fix-coverage dd10fb34 Lines: 28 39 71.8 %
Date: 2021-09-23 10:06:22 Functions: 4 4 100.0 %

          Line data    Source code
       1             : /* 
       2             :    Unix SMB/CIFS implementation.
       3             : 
       4             :    handling for netlogon dgram requests
       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 "includes.h"
      23             : #include "libcli/dgram/libdgram.h"
      24             : #include "lib/socket/socket.h"
      25             : #include "libcli/resolve/resolve.h"
      26             : #include "librpc/gen_ndr/ndr_nbt.h"
      27             : 
      28             : /* 
      29             :    send a netlogon mailslot request 
      30             : */
      31          42 : NTSTATUS dgram_mailslot_netlogon_send(struct nbt_dgram_socket *dgmsock,
      32             :                                       struct nbt_name *dest_name,
      33             :                                       struct socket_address *dest,
      34             :                                       const char *mailslot,
      35             :                                       struct nbt_name *src_name,
      36             :                                       struct nbt_netlogon_packet *request)
      37             : {
      38             :         NTSTATUS status;
      39             :         enum ndr_err_code ndr_err;
      40             :         DATA_BLOB blob;
      41          42 :         TALLOC_CTX *tmp_ctx = talloc_new(dgmsock);
      42             : 
      43          42 :         ndr_err = ndr_push_struct_blob(&blob, tmp_ctx, request,
      44             :                                       (ndr_push_flags_fn_t)ndr_push_nbt_netlogon_packet);
      45          42 :         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
      46           0 :                 talloc_free(tmp_ctx);
      47           0 :                 return ndr_map_error2ntstatus(ndr_err);
      48             :         }
      49             : 
      50             : 
      51          42 :         status = dgram_mailslot_send(dgmsock, DGRAM_DIRECT_UNIQUE, 
      52             :                                      mailslot,
      53             :                                      dest_name, dest, 
      54             :                                      src_name, &blob);
      55          42 :         talloc_free(tmp_ctx);
      56          42 :         return status;
      57             : }
      58             : 
      59             : 
      60             : /* 
      61             :    send a netlogon mailslot reply
      62             : */
      63          30 : NTSTATUS dgram_mailslot_netlogon_reply(struct nbt_dgram_socket *dgmsock,
      64             :                                        struct nbt_dgram_packet *request,
      65             :                                        const char *my_netbios_name,
      66             :                                        const char *mailslot_name,
      67             :                                        struct nbt_netlogon_response *reply)
      68             : {
      69             :         NTSTATUS status;
      70             :         DATA_BLOB blob;
      71          30 :         TALLOC_CTX *tmp_ctx = talloc_new(dgmsock);
      72             :         struct nbt_name myname;
      73             :         struct socket_address *dest;
      74             : 
      75          30 :         status = push_nbt_netlogon_response(&blob, tmp_ctx, reply);
      76          30 :         if (!NT_STATUS_IS_OK(status)) {
      77           0 :                 return status;
      78             :         }
      79             : 
      80          30 :         make_nbt_name_client(&myname, my_netbios_name);
      81             : 
      82          30 :         dest = socket_address_from_strings(tmp_ctx, dgmsock->sock->backend_name, 
      83          30 :                                            request->src_addr, request->src_port);
      84          30 :         if (!dest) {
      85           0 :                 talloc_free(tmp_ctx);
      86           0 :                 return NT_STATUS_NO_MEMORY;
      87             :         }
      88             : 
      89          30 :         status = dgram_mailslot_send(dgmsock, DGRAM_DIRECT_UNIQUE, 
      90             :                                      mailslot_name,
      91             :                                      &request->data.msg.source_name,
      92             :                                      dest,
      93             :                                      &myname, &blob);
      94          30 :         talloc_free(tmp_ctx);
      95          30 :         return status;
      96             : }
      97             : 
      98             : 
      99             : /*
     100             :   parse a netlogon request. The packet must be a valid mailslot packet
     101             : */
     102          30 : NTSTATUS dgram_mailslot_netlogon_parse_request(TALLOC_CTX *mem_ctx,
     103             :                                                struct nbt_dgram_packet *dgram,
     104             :                                                struct nbt_netlogon_packet *netlogon)
     105             : {
     106          30 :         DATA_BLOB data = dgram_mailslot_data(dgram);
     107             :         enum ndr_err_code ndr_err;
     108             : 
     109          30 :         ndr_err = ndr_pull_struct_blob(&data, mem_ctx, netlogon,
     110             :            (ndr_pull_flags_fn_t)ndr_pull_nbt_netlogon_packet);
     111          30 :         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
     112           0 :                 NTSTATUS status = ndr_map_error2ntstatus(ndr_err);
     113           0 :                 DEBUG(0,("Failed to parse netlogon packet of length %d: %s\n",
     114             :                          (int)data.length, nt_errstr(status)));
     115           0 :                 if (DEBUGLVL(10)) {
     116           0 :                         file_save("netlogon.dat", data.data, data.length);
     117             :                 }
     118           0 :                 return status;
     119             :         }
     120          30 :         return NT_STATUS_OK;
     121             : }
     122             : 
     123             : /*
     124             :   parse a netlogon response. The packet must be a valid mailslot packet
     125             : */
     126          42 : NTSTATUS dgram_mailslot_netlogon_parse_response(TALLOC_CTX *mem_ctx,
     127             :                                                 struct nbt_dgram_packet *dgram,
     128             :                                                 struct nbt_netlogon_response *netlogon)
     129             : {
     130             :         NTSTATUS status;
     131          42 :         DATA_BLOB data = dgram_mailslot_data(dgram);
     132             :         
     133          42 :         status = pull_nbt_netlogon_response(&data, mem_ctx, netlogon);
     134          42 :         if (!NT_STATUS_IS_OK(status)) {
     135           0 :                 return status;
     136             :         }
     137             :         
     138          42 :         return NT_STATUS_OK;
     139             : }
     140             : 

Generated by: LCOV version 1.13