LCOV - code coverage report
Current view: top level - source3/winbindd - winbindd_show_sequence.c (source / functions) Hit Total Coverage
Test: coverage report for master 2b515b7d Lines: 54 74 73.0 %
Date: 2024-02-28 12:06:22 Functions: 4 4 100.0 %

          Line data    Source code
       1             : /*
       2             :    Unix SMB/CIFS implementation.
       3             :    async implementation of WINBINDD_SHOW_SEQUENCE
       4             :    Copyright (C) Volker Lendecke 2009
       5             : 
       6             :    This program is free software; you can redistribute it and/or modify
       7             :    it under the terms of the GNU General Public License as published by
       8             :    the Free Software Foundation; either version 3 of the License, or
       9             :    (at your option) any later version.
      10             : 
      11             :    This program is distributed in the hope that it will be useful,
      12             :    but WITHOUT ANY WARRANTY; without even the implied warranty of
      13             :    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
      14             :    GNU General Public License for more details.
      15             : 
      16             :    You should have received a copy of the GNU General Public License
      17             :    along with this program.  If not, see <http://www.gnu.org/licenses/>.
      18             : */
      19             : 
      20             : #include "includes.h"
      21             : #include "winbindd.h"
      22             : 
      23             : struct winbindd_show_sequence_state {
      24             :         bool one_domain;
      25             :         /* One domain */
      26             :         uint32_t seqnum;
      27             : 
      28             :         /* All domains */
      29             :         int num_domains;
      30             :         NTSTATUS *statuses;
      31             :         struct winbindd_domain **domains;
      32             :         uint32_t *seqnums;
      33             : };
      34             : 
      35             : static void winbindd_show_sequence_done_one(struct tevent_req *subreq);
      36             : static void winbindd_show_sequence_done_all(struct tevent_req *subreq);
      37             : 
      38          20 : struct tevent_req *winbindd_show_sequence_send(TALLOC_CTX *mem_ctx,
      39             :                                                struct tevent_context *ev,
      40             :                                                struct winbindd_cli_state *cli,
      41             :                                                struct winbindd_request *request)
      42             : {
      43           0 :         struct tevent_req *req, *subreq;
      44           0 :         struct winbindd_show_sequence_state *state;
      45             : 
      46          20 :         req = tevent_req_create(mem_ctx, &state,
      47             :                                 struct winbindd_show_sequence_state);
      48          20 :         if (req == NULL) {
      49           0 :                 return NULL;
      50             :         }
      51          20 :         state->one_domain = false;
      52          20 :         state->domains = NULL;
      53          20 :         state->statuses = NULL;
      54          20 :         state->seqnums = NULL;
      55             : 
      56             :         /* Ensure null termination */
      57          20 :         request->domain_name[sizeof(request->domain_name)-1]='\0';
      58             : 
      59          20 :         DEBUG(3, ("show_sequence %s\n", request->domain_name));
      60             : 
      61          20 :         if (request->domain_name[0] != '\0') {
      62           0 :                 struct winbindd_domain *domain;
      63             : 
      64          16 :                 state->one_domain = true;
      65             : 
      66          16 :                 domain = find_domain_from_name_noinit(
      67          16 :                         request->domain_name);
      68          16 :                 if (domain == NULL) {
      69           0 :                         tevent_req_nterror(req, NT_STATUS_NO_SUCH_DOMAIN);
      70           0 :                         return tevent_req_post(req, ev);
      71             :                 }
      72             : 
      73          16 :                 subreq = wb_seqnum_send(state, ev, domain);
      74          16 :                 if (tevent_req_nomem(subreq, req)) {
      75           0 :                         return tevent_req_post(req, ev);
      76             :                 }
      77          16 :                 tevent_req_set_callback(
      78             :                         subreq, winbindd_show_sequence_done_one, req);
      79          16 :                 return req;
      80             :         }
      81             : 
      82           4 :         subreq = wb_seqnums_send(state, ev);
      83           4 :         if (tevent_req_nomem(subreq, req)) {
      84           0 :                 return tevent_req_post(req, ev);
      85             :         }
      86           4 :         tevent_req_set_callback(subreq, winbindd_show_sequence_done_all, req);
      87           4 :         return req;
      88             : }
      89             : 
      90          16 : static void winbindd_show_sequence_done_one(struct tevent_req *subreq)
      91             : {
      92          16 :         struct tevent_req *req = tevent_req_callback_data(
      93             :                 subreq, struct tevent_req);
      94          16 :         struct winbindd_show_sequence_state *state = tevent_req_data(
      95             :                 req, struct winbindd_show_sequence_state);
      96           0 :         NTSTATUS status;
      97             : 
      98          16 :         status = wb_seqnum_recv(subreq, &state->seqnum);
      99          16 :         TALLOC_FREE(subreq);
     100          16 :         if (tevent_req_nterror(req, status)) {
     101           0 :                 return;
     102             :         }
     103          16 :         tevent_req_done(req);
     104             : }
     105             : 
     106           4 : static void winbindd_show_sequence_done_all(struct tevent_req *subreq)
     107             : {
     108           4 :         struct tevent_req *req = tevent_req_callback_data(
     109             :                 subreq, struct tevent_req);
     110           4 :         struct winbindd_show_sequence_state *state = tevent_req_data(
     111             :                 req, struct winbindd_show_sequence_state);
     112           0 :         NTSTATUS status;
     113             : 
     114           4 :         status = wb_seqnums_recv(subreq, state, &state->num_domains,
     115             :                                  &state->domains, &state->statuses,
     116             :                                  &state->seqnums);
     117           4 :         TALLOC_FREE(subreq);
     118           4 :         if (tevent_req_nterror(req, status)) {
     119           0 :                 return;
     120             :         }
     121           4 :         tevent_req_done(req);
     122             : }
     123             : 
     124          20 : NTSTATUS winbindd_show_sequence_recv(struct tevent_req *req,
     125             :                                      struct winbindd_response *response)
     126             : {
     127          20 :         struct winbindd_show_sequence_state *state = tevent_req_data(
     128             :                 req, struct winbindd_show_sequence_state);
     129           0 :         NTSTATUS status;
     130           0 :         char *extra_data;
     131           0 :         int i;
     132             : 
     133          20 :         if (tevent_req_is_nterror(req, &status)) {
     134           0 :                 return status;
     135             :         }
     136             : 
     137          20 :         if (state->one_domain) {
     138          16 :                 response->data.sequence_number = state->seqnum;
     139          16 :                 return NT_STATUS_OK;
     140             :         }
     141             : 
     142           4 :         extra_data = talloc_strdup(response, "");
     143           4 :         if (extra_data == NULL) {
     144           0 :                 return NT_STATUS_NO_MEMORY;
     145             :         }
     146             : 
     147          20 :         for (i=0; i<state->num_domains; i++) {
     148          16 :                 if (!NT_STATUS_IS_OK(state->statuses[i])
     149          16 :                     || (state->seqnums[i] == DOM_SEQUENCE_NONE)) {
     150           0 :                         extra_data = talloc_asprintf_append_buffer(
     151             :                                 extra_data, "%s : DISCONNECTED\n",
     152           0 :                                 state->domains[i]->name);
     153             :                 } else {
     154          16 :                         extra_data = talloc_asprintf_append_buffer(
     155             :                                 extra_data, "%s : %d\n",
     156          16 :                                 state->domains[i]->name,
     157          16 :                                 (int)state->seqnums[i]);
     158             :                 }
     159          16 :                 if (extra_data == NULL) {
     160           0 :                         return NT_STATUS_NO_MEMORY;
     161             :                 }
     162             :         }
     163             : 
     164           4 :         response->extra_data.data = extra_data;
     165           4 :         response->length += talloc_get_size(extra_data);
     166           4 :         return NT_STATUS_OK;
     167             : }

Generated by: LCOV version 1.14