librpc/ndr/ndr_dnsp.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    Manually parsed structures found in DNSP
   5
   6    Copyright (C) Andrew Tridgell 2010
   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 "librpc/gen_ndr/ndr_dnsp.h"
  24
  25 /*
  26   print a dnsp_name
  27 */
  28 _PUBLIC_ void ndr_print_dnsp_name(struct ndr_print *ndr, const char *name,
  29                                   const char *dns_name)
  30 {
  31         ndr->print(ndr, "%-25s: %s", name, dns_name);
  32 }
  33
  34 /*
  35   pull a dnsp_name
  36 */
  37 _PUBLIC_ enum ndr_err_code ndr_pull_dnsp_name(struct ndr_pull *ndr, int ndr_flags, const char **name)
  38 {
  39         uint8_t len, count, final;
  40         int i;
  41         uint32_t total_len;
  42         char *ret;
  43
  44         NDR_CHECK(ndr_pull_uint8(ndr, ndr_flags, &len));
  45         NDR_CHECK(ndr_pull_uint8(ndr, ndr_flags, &count));
  46
  47         ret = talloc_strdup(ndr->current_mem_ctx, "");
  48         if (!ret) {
  49                 return ndr_pull_error(ndr, NDR_ERR_ALLOC, "Failed to pull dnsp");
  50         }
  51         total_len = 1;
  52
  53         for (i=0; i<count; i++) {
  54                 uint8_t sublen, newlen;
  55                 NDR_CHECK(ndr_pull_uint8(ndr, ndr_flags, &sublen));
  56                 newlen = total_len + sublen;
  57                 if (i != count-1) {
  58                         newlen++; /* for the '.' */
  59                 }
  60                 ret = talloc_realloc(ndr->current_mem_ctx, ret, char, newlen);
  61                 if (!ret) {
  62                         return ndr_pull_error(ndr, NDR_ERR_ALLOC, "Failed to pull dnsp");
  63                 }
  64                 NDR_CHECK(ndr_pull_bytes(ndr, (uint8_t *)&ret[total_len-1], sublen));
  65                 if (i != count-1) {
  66                         ret[newlen-2] = '.';
  67                 }
  68                 ret[newlen-1] = 0;
  69                 total_len = newlen;
  70         }
  71         NDR_CHECK(ndr_pull_uint8(ndr, ndr_flags, &final));
  72         (*name) = ret;
  73         return NDR_ERR_SUCCESS;
  74 }
  75
  76 enum ndr_err_code ndr_push_dnsp_name(struct ndr_push *ndr, int ndr_flags, const char *name)
  77 {
  78         int count, total_len, i;
  79
  80         /* count the dots */
  81         for (count=i=0; name[i]; i++) {
  82                 if (name[i] == '.') count++;
  83         }
  84         total_len = strlen(name) + 1;
  85         if (total_len > 255 || count > 255) {
  86                 return ndr_push_error(ndr, NDR_ERR_BUFSIZE,
  87                                       "dns_name of length %d larger than 255", total_len);
  88         }
  89         NDR_CHECK(ndr_push_uint8(ndr, ndr_flags, (uint8_t)total_len));
  90         NDR_CHECK(ndr_push_uint8(ndr, ndr_flags, (uint8_t)count));
  91         for (i=0; i<count; i++) {
  92                 const char *p = strchr(name, '.');
  93                 size_t sublen = p?(p-name):strlen(name);
  94                 NDR_CHECK(ndr_push_uint8(ndr, ndr_flags, (uint8_t)sublen));
  95                 NDR_CHECK(ndr_push_bytes(ndr, (const uint8_t *)name, sublen));
  96                 name += sublen + 1;
  97         }
  98         NDR_CHECK(ndr_push_uint8(ndr, ndr_flags, 0));
  99
 100         return NDR_ERR_SUCCESS;
 101 }
 102
 103 /*
 104   print a dnsp_string
 105 */
 106 _PUBLIC_ void ndr_print_dnsp_string(struct ndr_print *ndr, const char *name,
 107                                     const char *dns_string)
 108 {
 109         ndr->print(ndr, "%-25s: %s", name, dns_string);
 110 }
 111
 112 /*
 113   pull a dnsp_string
 114 */
 115 _PUBLIC_ enum ndr_err_code ndr_pull_dnsp_string(struct ndr_pull *ndr, int ndr_flags, const char **string)
 116 {
 117         uint8_t len;
 118         uint32_t total_len;
 119         char *ret;
 120
 121         NDR_CHECK(ndr_pull_uint8(ndr, ndr_flags, &len));
 122
 123         ret = talloc_strdup(ndr->current_mem_ctx, "");
 124         if (!ret) {
 125                 return ndr_pull_error(ndr, NDR_ERR_ALLOC, "Failed to pull dnsp");
 126         }
 127         total_len = 1;
 128         ret = talloc_zero_array(ndr->current_mem_ctx, char, len+1);
 129         if (!ret) {
 130                 return ndr_pull_error(ndr, NDR_ERR_ALLOC, "Failed to pull dnsp");
 131         }
 132         NDR_CHECK(ndr_pull_bytes(ndr, (uint8_t *)&ret[total_len-1], len));
 133         total_len = len;
 134
 135         (*string) = ret;
 136         NDR_PULL_ALIGN(ndr, 1);
 137         return NDR_ERR_SUCCESS;
 138 }
 139
 140 enum ndr_err_code ndr_push_dnsp_string(struct ndr_push *ndr, int ndr_flags, const char *string)
 141 {
 142         int total_len;
 143         total_len = strlen(string) + 1;
 144         if (total_len > 255) {
 145                 return ndr_push_error(ndr, NDR_ERR_BUFSIZE,
 146                                       "dns_name of length %d larger than 255", total_len);
 147         }
 148         NDR_CHECK(ndr_push_uint8(ndr, ndr_flags, (uint8_t)total_len));
 149         NDR_CHECK(ndr_push_bytes(ndr, (const uint8_t *)string, total_len - 1));
 150         NDR_PUSH_ALIGN(ndr, 1);
 151
 152         return NDR_ERR_SUCCESS;
 153 }