Line data    Source code

       1             : /*
       2             :    Unix SMB/CIFS implementation.
       3             : 
       4             :    async seqnums, update the seqnums in winbindd_cache.c
       5             : 
       6             :    Copyright (C) Volker Lendecke 2009
       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 "winbindd.h"
      24             : #include "librpc/gen_ndr/ndr_winbind_c.h"
      25             : 
      26             : struct wb_seqnums_state {
      27             :         int num_domains;
      28             :         int num_received;
      29             : 
      30             :         struct tevent_req **subreqs;
      31             :         struct winbindd_domain **domains;
      32             :         NTSTATUS *statuses;
      33             :         uint32_t *seqnums;
      34             : };
      35             : 
      36             : static void wb_seqnums_done(struct tevent_req *subreq);
      37             : 
      38           4 : struct tevent_req *wb_seqnums_send(TALLOC_CTX *mem_ctx,
      39             :                                    struct tevent_context *ev)
      40             : {
      41           0 :         struct tevent_req *req;
      42           0 :         struct wb_seqnums_state *state;
      43           0 :         struct winbindd_domain *domain;
      44           0 :         int i;
      45             : 
      46           4 :         req = tevent_req_create(mem_ctx, &state, struct wb_seqnums_state);
      47           4 :         if (req == NULL) {
      48           0 :                 return NULL;
      49             :         }
      50           4 :         state->num_received = 0;
      51           4 :         state->num_domains = 0;
      52             : 
      53          20 :         for (domain = domain_list(); domain != NULL; domain = domain->next) {
      54          16 :                 state->num_domains += 1;
      55             :         }
      56             : 
      57           4 :         state->subreqs = talloc_array(state, struct tevent_req *,
      58             :                                       state->num_domains);
      59           4 :         state->domains = talloc_zero_array(state, struct winbindd_domain *,
      60             :                                            state->num_domains);
      61           4 :         state->statuses = talloc_array(state, NTSTATUS, state->num_domains);
      62           4 :         state->seqnums = talloc_array(state, uint32_t, state->num_domains);
      63             : 
      64           4 :         if ((state->subreqs == NULL) || (state->domains == NULL) ||
      65           4 :             (state->statuses == NULL) || (state->seqnums == NULL)) {
      66           0 :                 tevent_req_nterror(req, NT_STATUS_NO_MEMORY);
      67           0 :                 return tevent_req_post(req, ev);
      68             :         }
      69             : 
      70           4 :         i = 0;
      71             : 
      72          20 :         for (domain = domain_list(); domain != NULL; domain = domain->next) {
      73          16 :                 state->domains[i] = domain;
      74          16 :                 state->subreqs[i] = wb_seqnum_send(state->subreqs, ev, domain);
      75          16 :                 if (tevent_req_nomem(state->subreqs[i], req)) {
      76             :                         /* Don't even start all the other requests */
      77           0 :                         TALLOC_FREE(state->subreqs);
      78           0 :                         return tevent_req_post(req, ev);
      79             :                 }
      80          16 :                 tevent_req_set_callback(state->subreqs[i], wb_seqnums_done,
      81             :                                         req);
      82          16 :                 i += 1;
      83             :         }
      84           4 :         return req;
      85             : }
      86             : 
      87          16 : static void wb_seqnums_done(struct tevent_req *subreq)
      88             : {
      89          16 :         struct tevent_req *req = tevent_req_callback_data(
      90             :                 subreq, struct tevent_req);
      91          16 :         struct wb_seqnums_state *state = tevent_req_data(
      92             :                 req, struct wb_seqnums_state);
      93           0 :         NTSTATUS status;
      94           0 :         uint32_t seqnum;
      95           0 :         int i;
      96             : 
      97          16 :         status = wb_seqnum_recv(subreq, &seqnum);
      98             : 
      99          42 :         for (i=0; i<state->num_domains; i++) {
     100          42 :                 if (subreq == state->subreqs[i]) {
     101          16 :                         break;
     102             :                 }
     103             :         }
     104          16 :         if (i < state->num_domains) {
     105             :                 /* found one */
     106             : 
     107          16 :                 state->subreqs[i] = NULL;
     108          16 :                 state->statuses[i] = status;
     109          16 :                 if (NT_STATUS_IS_OK(status)) {
     110          16 :                         state->seqnums[i] = seqnum;
     111             : 
     112             :                         /*
     113             :                          * This first assignment might be removed
     114             :                          * later
     115             :                          */
     116          16 :                         state->domains[i]->sequence_number = seqnum;
     117             : 
     118          32 :                         if (!wcache_store_seqnum(state->domains[i]->name,
     119          16 :                                                  state->seqnums[i],
     120             :                                                  time(NULL))) {
     121           0 :                                 DEBUG(1, ("wcache_store_seqnum failed for "
     122             :                                           "domain %s\n",
     123             :                                           state->domains[i]->name));
     124             :                         }
     125             :                 }
     126             :         }
     127             : 
     128          16 :         TALLOC_FREE(subreq);
     129             : 
     130          16 :         state->num_received += 1;
     131             : 
     132          16 :         if (state->num_received >= state->num_domains) {
     133           4 :                 tevent_req_done(req);
     134             :         }
     135          16 : }
     136             : 
     137           4 : NTSTATUS wb_seqnums_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
     138             :                          int *num_domains, struct winbindd_domain ***domains,
     139             :                          NTSTATUS **statuses, uint32_t **seqnums)
     140             : {
     141           4 :         struct wb_seqnums_state *state = tevent_req_data(
     142             :                 req, struct wb_seqnums_state);
     143           0 :         NTSTATUS status;
     144             : 
     145           4 :         if (tevent_req_is_nterror(req, &status)) {
     146           0 :                 return status;
     147             :         }
     148           4 :         *num_domains = state->num_domains;
     149           4 :         *domains = talloc_move(mem_ctx, &state->domains);
     150           4 :         *statuses = talloc_move(mem_ctx, &state->statuses);
     151           4 :         *seqnums = talloc_move(mem_ctx, &state->seqnums);
     152           4 :         return NT_STATUS_OK;
     153             : }