librpc/tests/test_ndr.c

   1 /*
   2  * Tests for librpc ndr functions
   3  *
   4  * Copyright (C) Catalyst.NET Ltd 2020
   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
  21 /*
  22  * from cmocka.c:
  23  * These headers or their equivalents should be included prior to
  24  * including
  25  * this header file.
  26  *
  27  * #include <stdarg.h>
  28  * #include <stddef.h>
  29  * #include <setjmp.h>
  30  *
  31  * This allows test applications to use custom definitions of C standard
  32  * library functions and types.
  33  *
  34  */
  35 #include <stdarg.h>
  36 #include <stddef.h>
  37 #include <stdint.h>
  38 #include <setjmp.h>
  39 #include <cmocka.h>
  40
  41 #include "librpc/ndr/libndr.h"
  42
  43 /*
  44  * Test NDR_PULL_NEED_BYTES integer overflow handling.
  45  */
  46 static enum ndr_err_code wrap_NDR_PULL_NEED_BYTES(
  47         struct ndr_pull *ndr,
  48         uint32_t bytes) {
  49
  50         NDR_PULL_NEED_BYTES(ndr, bytes);
  51         return NDR_ERR_SUCCESS;
  52 }
  53
  54 static void test_NDR_PULL_NEED_BYTES(void **state)
  55 {
  56         struct ndr_pull ndr = {0};
  57         enum ndr_err_code err;
  58
  59         ndr.data_size = UINT32_MAX;
  60         ndr.offset = UINT32_MAX -1;
  61
  62         /*
  63          * This will not cause an overflow
  64          */
  65         err = wrap_NDR_PULL_NEED_BYTES(&ndr, 1);
  66         assert_int_equal(NDR_ERR_SUCCESS, err);
  67
  68         /*
  69          * This will cause an overflow
  70          * and (offset + n) will be less than data_size
  71          */
  72         err = wrap_NDR_PULL_NEED_BYTES(&ndr, 2);
  73         assert_int_equal(NDR_ERR_BUFSIZE, err);
  74 }
  75
  76 /*
  77  * Test NDR_PULL_ALIGN integer overflow handling.
  78  */
  79 static enum ndr_err_code wrap_NDR_PULL_ALIGN(
  80         struct ndr_pull *ndr,
  81         uint32_t bytes) {
  82
  83         NDR_PULL_ALIGN(ndr, bytes);
  84         return NDR_ERR_SUCCESS;
  85 }
  86
  87 static void test_NDR_PULL_ALIGN(void **state)
  88 {
  89         struct ndr_pull ndr = {0};
  90         enum ndr_err_code err;
  91
  92         ndr.data_size = UINT32_MAX;
  93         ndr.offset = UINT32_MAX -1;
  94
  95         /*
  96          * This will not cause an overflow
  97          */
  98         err = wrap_NDR_PULL_ALIGN(&ndr, 2);
  99         assert_int_equal(NDR_ERR_SUCCESS, err);
 100
 101         /*
 102          * This will cause an overflow
 103          * and (offset + n) will be less than data_size
 104          */
 105         err = wrap_NDR_PULL_ALIGN(&ndr, 4);
 106         assert_int_equal(NDR_ERR_BUFSIZE, err);
 107 }
 108
 109 /*
 110  * Test ndr_pull_advance integer overflow handling.
 111  */
 112 static void test_ndr_pull_advance(void **state)
 113 {
 114         struct ndr_pull ndr = {0};
 115         enum ndr_err_code err;
 116
 117         ndr.data_size = UINT32_MAX;
 118         ndr.offset = UINT32_MAX -1;
 119
 120         /*
 121          * This will not cause an overflow
 122          */
 123         err = ndr_pull_advance(&ndr, 1);
 124         assert_int_equal(NDR_ERR_SUCCESS, err);
 125
 126         /*
 127          * This will cause an overflow
 128          * and (offset + n) will be less than data_size
 129          */
 130         err = ndr_pull_advance(&ndr, 2);
 131         assert_int_equal(NDR_ERR_BUFSIZE, err);
 132 }
 133
 134 int main(int argc, const char **argv)
 135 {
 136         const struct CMUnitTest tests[] = {
 137                 cmocka_unit_test(test_NDR_PULL_NEED_BYTES),
 138                 cmocka_unit_test(test_NDR_PULL_ALIGN),
 139                 cmocka_unit_test(test_ndr_pull_advance),
 140         };
 141
 142         cmocka_set_message_output(CM_OUTPUT_SUBUNIT);
 143         return cmocka_run_group_tests(tests, NULL, NULL);
 144 }