lib/util/server_id.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Samba utility functions
   4    Copyright (C) Andrew Bartlett 2011
   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 "librpc/gen_ndr/server_id.h"
  22
  23 bool server_id_equal(const struct server_id *p1, const struct server_id *p2)
  24 {
  25         if (p1->pid != p2->pid) {
  26                 return false;
  27         }
  28
  29         if (p1->task_id != p2->task_id) {
  30                 return false;
  31         }
  32
  33         if (p1->vnn != p2->vnn) {
  34                 return false;
  35         }
  36
  37         if (p1->unique_id != p2->unique_id) {
  38                 return false;
  39         }
  40
  41         return true;
  42 }
  43
  44 char *server_id_str_buf(struct server_id id, struct server_id_buf *dst)
  45 {
  46         if (server_id_is_disconnected(&id)) {
  47                 strlcpy(dst->buf, "disconnected", sizeof(dst->buf));
  48         } else if ((id.vnn == NONCLUSTER_VNN) && (id.task_id == 0)) {
  49                 snprintf(dst->buf, sizeof(dst->buf), "%llu",
  50                          (unsigned long long)id.pid);
  51         } else if (id.vnn == NONCLUSTER_VNN) {
  52                 snprintf(dst->buf, sizeof(dst->buf), "%llu.%u",
  53                          (unsigned long long)id.pid, (unsigned)id.task_id);
  54         } else if (id.task_id == 0) {
  55                 snprintf(dst->buf, sizeof(dst->buf), "%u:%llu",
  56                          (unsigned)id.vnn, (unsigned long long)id.pid);
  57         } else {
  58                 snprintf(dst->buf, sizeof(dst->buf), "%u:%llu.%u",
  59                          (unsigned)id.vnn,
  60                          (unsigned long long)id.pid,
  61                          (unsigned)id.task_id);
  62         }
  63         return dst->buf;
  64 }
  65
  66 char *server_id_str(TALLOC_CTX *mem_ctx, const struct server_id *id)
  67 {
  68         struct server_id_buf tmp;
  69         char *result;
  70
  71         result = talloc_strdup(mem_ctx, server_id_str_buf(*id, &tmp));
  72         if (result == NULL) {
  73                 return NULL;
  74         }
  75
  76         /*
  77          * beautify the talloc_report output
  78          */
  79         talloc_set_name_const(result, result);
  80         return result;
  81 }
  82
  83 struct server_id server_id_from_string(uint32_t local_vnn,
  84                                        const char *pid_string)
  85 {
  86         struct server_id result;
  87         unsigned long long pid;
  88         unsigned int vnn, task_id = 0;
  89
  90         ZERO_STRUCT(result);
  91
  92         /*
  93          * We accept various forms with 1, 2 or 3 component forms
  94          * because the server_id_str() can print different forms, and
  95          * we want backwards compatibility for scripts that may call
  96          * smbclient.
  97          */
  98         if (sscanf(pid_string, "%u:%llu.%u", &vnn, &pid, &task_id) == 3) {
  99                 result.vnn = vnn;
 100                 result.pid = pid;
 101                 result.task_id = task_id;
 102         } else if (sscanf(pid_string, "%u:%llu", &vnn, &pid) == 2) {
 103                 result.vnn = vnn;
 104                 result.pid = pid;
 105         } else if (sscanf(pid_string, "%llu.%u", &pid, &task_id) == 2) {
 106                 result.vnn = local_vnn;
 107                 result.pid = pid;
 108                 result.task_id = task_id;
 109         } else if (sscanf(pid_string, "%llu", &pid) == 1) {
 110                 result.vnn = local_vnn;
 111                 result.pid = pid;
 112         } else if (strcmp(pid_string, "disconnected") ==0) {
 113                 server_id_set_disconnected(&result);
 114         } else {
 115                 result.vnn = NONCLUSTER_VNN;
 116                 result.pid = UINT64_MAX;
 117         }
 118         return result;
 119 }
 120
 121 /**
 122  * Set the serverid to the special value that represents a disconnected
 123  * client for (e.g.) durable handles.
 124  */
 125 void server_id_set_disconnected(struct server_id *id)
 126 {
 127         SMB_ASSERT(id != NULL);
 128
 129         id->pid = UINT64_MAX;
 130         id->task_id = UINT32_MAX;
 131         id->vnn = NONCLUSTER_VNN;
 132         id->unique_id = SERVERID_UNIQUE_ID_NOT_TO_VERIFY;
 133
 134         return;
 135 }
 136
 137 /**
 138  * check whether a serverid is the special placeholder for
 139  * a disconnected client
 140  */
 141 bool server_id_is_disconnected(const struct server_id *id)
 142 {
 143         struct server_id dis;
 144
 145         SMB_ASSERT(id != NULL);
 146
 147         server_id_set_disconnected(&dis);
 148
 149         return server_id_equal(id, &dis);
 150 }