Line data    Source code

       1             : /*
       2             :    Unix SMB/CIFS implementation.
       3             :    Wrap unix errno around tevent_req
       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 "../replace/replace.h"
      21             : #include "tevent_ntstatus.h"
      22             : #include "libcli/util/error.h"
      23             : 
      24             : #define TEVENT_NTERROR_MAGIC (0x917b5acd)
      25             : 
      26     3296694 : bool _tevent_req_nterror(struct tevent_req *req,
      27             :                          NTSTATUS status,
      28             :                          const char *location)
      29             : {
      30             :         uint64_t err;
      31             : 
      32     3296694 :         if (NT_STATUS_IS_OK(status)) {
      33     3076011 :                 return false;
      34             :         }
      35             : 
      36             :         /*
      37             :          * I've put this variable here, because I'm not 100% certain
      38             :          * how to correctly assign a 64-bit constant and left-shift it
      39             :          * by 32 bits in a single expression. If anyone knows, feel
      40             :          * free :-)
      41             :          */
      42      220683 :         err = TEVENT_NTERROR_MAGIC;
      43      220683 :         err <<= 32;
      44      220683 :         err |= NT_STATUS_V(status);
      45             : 
      46      220683 :         return _tevent_req_error(req, err, location);
      47             : }
      48             : 
      49     6034487 : bool tevent_req_is_nterror(struct tevent_req *req, NTSTATUS *status)
      50             : {
      51             :         enum tevent_req_state state;
      52             :         uint64_t err;
      53             : 
      54     6034487 :         if (!tevent_req_is_error(req, &state, &err)) {
      55     5814257 :                 return false;
      56             :         }
      57      220230 :         switch (state) {
      58          64 :         case TEVENT_REQ_TIMED_OUT:
      59          64 :                 *status = NT_STATUS_IO_TIMEOUT;
      60          64 :                 break;
      61           0 :         case TEVENT_REQ_NO_MEMORY:
      62           0 :                 *status = NT_STATUS_NO_MEMORY;
      63           0 :                 break;
      64      220154 :         case TEVENT_REQ_USER_ERROR:
      65      220154 :                 if ((err >> 32) != TEVENT_NTERROR_MAGIC) {
      66           0 :                         abort();
      67             :                 }
      68      220154 :                 *status = NT_STATUS(err & 0xffffffff);
      69      220154 :                 break;
      70          12 :         default:
      71          12 :                 *status = NT_STATUS_INTERNAL_ERROR;
      72          12 :                 break;
      73             :         }
      74      220230 :         return true;
      75             : }
      76             : 
      77      376354 : NTSTATUS tevent_req_simple_recv_ntstatus(struct tevent_req *req)
      78             : {
      79      376354 :         NTSTATUS status = NT_STATUS_OK;
      80             : 
      81             :         /*
      82             :          * Ignore result of tevent_req_is_nterror, we're only interested in
      83             :          * the status
      84             :          */
      85      376354 :         tevent_req_is_nterror(req, &status);
      86      376354 :         tevent_req_received(req);
      87      376354 :         return status;
      88             : }
      89             : 
      90       23762 : void tevent_req_simple_finish_ntstatus(struct tevent_req *subreq,
      91             :                                        NTSTATUS subreq_status)
      92             : {
      93       23762 :         struct tevent_req *req = tevent_req_callback_data(
      94             :                 subreq, struct tevent_req);
      95             : 
      96       23762 :         TALLOC_FREE(subreq);
      97             : 
      98       23762 :         if (!NT_STATUS_IS_OK(subreq_status)) {
      99         644 :                 tevent_req_nterror(req, subreq_status);
     100         644 :                 return;
     101             :         }
     102       23118 :         tevent_req_done(req);
     103             : }
     104             : 
     105      300253 : bool tevent_req_poll_ntstatus(struct tevent_req *req,
     106             :                               struct tevent_context *ev,
     107             :                               NTSTATUS *status)
     108             : {
     109      300253 :         bool ret = tevent_req_poll(req, ev);
     110      300253 :         if (!ret) {
     111           0 :                 *status = map_nt_error_from_unix_common(errno);
     112             :         }
     113      300253 :         return ret;
     114             : }