lib/util/tevent_ntstatus.c

   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
  23 #define TEVENT_NTERROR_MAGIC (0x917b5acd)
  24
  25 bool _tevent_req_nterror(struct tevent_req *req,
  26                          NTSTATUS status,
  27                          const char *location)
  28 {
  29         uint64_t err;
  30
  31         if (NT_STATUS_IS_OK(status)) {
  32                 return false;
  33         }
  34
  35         /*
  36          * I've put this variable here, because I'm not 100% certain
  37          * how to correctly assign a 64-bit constant and left-shift it
  38          * by 32 bits in a single expression. If anyone knows, feel
  39          * free :-)
  40          */
  41         err = TEVENT_NTERROR_MAGIC;
  42         err <<= 32;
  43         err |= NT_STATUS_V(status);
  44
  45         return _tevent_req_error(req, err, location);
  46 }
  47
  48 bool tevent_req_is_nterror(struct tevent_req *req, NTSTATUS *status)
  49 {
  50         enum tevent_req_state state;
  51         uint64_t err;
  52
  53         if (!tevent_req_is_error(req, &state, &err)) {
  54                 return false;
  55         }
  56         switch (state) {
  57         case TEVENT_REQ_TIMED_OUT:
  58                 *status = NT_STATUS_IO_TIMEOUT;
  59                 break;
  60         case TEVENT_REQ_NO_MEMORY:
  61                 *status = NT_STATUS_NO_MEMORY;
  62                 break;
  63         case TEVENT_REQ_USER_ERROR:
  64                 if ((err >> 32) != TEVENT_NTERROR_MAGIC) {
  65                         abort();
  66                 }
  67                 *status = NT_STATUS(err & 0xffffffff);
  68                 break;
  69         default:
  70                 *status = NT_STATUS_INTERNAL_ERROR;
  71                 break;
  72         }
  73         return true;
  74 }
  75
  76 NTSTATUS tevent_req_simple_recv_ntstatus(struct tevent_req *req)
  77 {
  78         NTSTATUS status;
  79
  80         if (tevent_req_is_nterror(req, &status)) {
  81                 return status;
  82         }
  83         return NT_STATUS_OK;
  84 }
  85
  86 void tevent_req_simple_finish_ntstatus(struct tevent_req *subreq,
  87                                        NTSTATUS subreq_status)
  88 {
  89         struct tevent_req *req = tevent_req_callback_data(
  90                 subreq, struct tevent_req);
  91
  92         TALLOC_FREE(subreq);
  93
  94         if (!NT_STATUS_IS_OK(subreq_status)) {
  95                 tevent_req_nterror(req, subreq_status);
  96                 return;
  97         }
  98         tevent_req_done(req);
  99 }