| 1 | #!/usr/bin/perl
|
|---|
| 2 | # NDR represent_as() / transmit_as() tests
|
|---|
| 3 | # (C) 2006 Jelmer Vernooij. Published under the GNU GPL
|
|---|
| 4 | use strict;
|
|---|
| 5 |
|
|---|
| 6 | use Test::More tests => 2 * 8;
|
|---|
| 7 | use FindBin qw($RealBin);
|
|---|
| 8 | use lib "$RealBin";
|
|---|
| 9 | use Util qw(test_samba4_ndr);
|
|---|
| 10 |
|
|---|
| 11 | test_samba4_ndr('represent_as-simple',
|
|---|
| 12 | '
|
|---|
| 13 | void bla([in,represent_as(uint32)] uint8 x);
|
|---|
| 14 | ',
|
|---|
| 15 | '
|
|---|
| 16 | uint8_t expected[] = { 0x0D };
|
|---|
| 17 | DATA_BLOB in_blob = { expected, 1 };
|
|---|
| 18 | struct ndr_pull *ndr = ndr_pull_init_blob(&in_blob, NULL, NULL);
|
|---|
| 19 | struct bla r;
|
|---|
| 20 |
|
|---|
| 21 | if (!NDR_ERR_CODE_IS_SUCCESS(ndr_pull_bla(ndr, NDR_SCALARS|NDR_BUFFERS, &r)))
|
|---|
| 22 | return 1;
|
|---|
| 23 |
|
|---|
| 24 | if (r.in.x != 13)
|
|---|
| 25 | return 2;
|
|---|
| 26 | ',
|
|---|
| 27 | '
|
|---|
| 28 | enum ndr_err_code ndr_uint8_to_uint32(uint8_t from, uint32_t *to)
|
|---|
| 29 | {
|
|---|
| 30 | *to = from;
|
|---|
| 31 | return NDR_ERR_SUCCESS;
|
|---|
| 32 | }
|
|---|
| 33 |
|
|---|
| 34 | enum ndr_err_code ndr_uint32_to_uint8(uint32_t from, uint8_t *to)
|
|---|
| 35 | {
|
|---|
| 36 | *to = from;
|
|---|
| 37 | return NDR_ERR_SUCCESS;
|
|---|
| 38 | }
|
|---|
| 39 | '
|
|---|
| 40 | );
|
|---|
| 41 |
|
|---|
| 42 | test_samba4_ndr('transmit_as-simple',
|
|---|
| 43 | '
|
|---|
| 44 | void bla([in,transmit_as(uint32)] uint8 x);
|
|---|
| 45 | ',
|
|---|
| 46 | '
|
|---|
| 47 | uint8_t expected[] = { 0x0D };
|
|---|
| 48 | DATA_BLOB in_blob = { expected, 1 };
|
|---|
| 49 | struct ndr_pull *ndr = ndr_pull_init_blob(&in_blob, NULL, NULL);
|
|---|
| 50 | struct bla r;
|
|---|
| 51 |
|
|---|
| 52 | if (!NDR_ERR_CODE_IS_SUCCESS(ndr_pull_bla(ndr, NDR_SCALARS|NDR_BUFFERS, &r)))
|
|---|
| 53 | return 1;
|
|---|
| 54 |
|
|---|
| 55 | if (r.in.x != 13)
|
|---|
| 56 | return 2;
|
|---|
| 57 | ',
|
|---|
| 58 | '
|
|---|
| 59 | enum ndr_err_code ndr_uint8_to_uint32(uint8_t from, uint32_t *to)
|
|---|
| 60 | {
|
|---|
| 61 | *to = from;
|
|---|
| 62 | return NDR_ERR_SUCCESS;
|
|---|
| 63 | }
|
|---|
| 64 |
|
|---|
| 65 | enum ndr_err_code ndr_uint32_to_uint8(uint32_t from, uint8_t *to)
|
|---|
| 66 | {
|
|---|
| 67 | *to = from;
|
|---|
| 68 | return NDR_ERR_SUCCESS;
|
|---|
| 69 | }
|
|---|
| 70 | '
|
|---|
| 71 | );
|
|---|