Erik/colosseum-near-rt-ric
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
main
colosseum-near-rt-ric/setup/e2mgr/E2Manager/3rdparty/asn1codec/e2ap_engine/converter-example.c

1034 lines
33 KiB
C
Raw Permalink Normal View History

First commit
2021-12-08 21:17:46 +01:00
/*
* Generic converter template for a selected ASN.1 type.
* Copyright (c) 2005-2017 Lev Walkin <vlm@lionet.info>.
* All rights reserved.
*
* To compile with your own ASN.1 type, redefine the PDU as shown:
*
* cc -DPDU=MyCustomType -o myDecoder.o -c converter-example.c
*/
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#define __EXTENSIONS__
#include <stdio.h>
#include <sys/types.h>
#include <stdlib.h> /* for atoi(3) */
#include <getopt.h> /* for getopt(3) */
#include <string.h> /* for strerror(3) */
#include <sysexits.h> /* for EX_* exit codes */
#include <errno.h> /* for errno */
#include <unistd.h> /* for isatty(3) */
#include <asn_application.h>
#include <asn_internal.h> /* for ASN__DEFAULT_STACK_MAX */
/* Convert "Type" defined by -DPDU into "asn_DEF_Type" */
#ifdef PDU
#define ASN_DEF_PDU(t) asn_DEF_ ## t
#define DEF_PDU_Type(t) ASN_DEF_PDU(t)
#define PDU_Type DEF_PDU_Type(PDU)
extern asn_TYPE_descriptor_t PDU_Type; /* ASN.1 type to be decoded */
#define PDU_Type_Ptr (&PDU_Type)
#else /* !PDU */
#define PDU_Type_Ptr NULL
#endif /* PDU */
#ifdef ASN_PDU_COLLECTION /* Generated by asn1c -pdu=... */
extern asn_TYPE_descriptor_t *asn_pdu_collection[];
#endif
#ifndef NO_ASN_PDU
#if !defined(PDU) && !defined(ASN_PDU_COLLECTION)
#error Define -DPDU to compile this example converter.
#error `asn1c -pdu=...` adds necessary flags automatically.
#endif
#endif
/*
* Open file and parse its contens.
*/
static void *data_decode_from_file(enum asn_transfer_syntax,
asn_TYPE_descriptor_t *asnTypeOfPDU,
FILE *file, const char *name,
ssize_t suggested_bufsize, int first_pdu);
static int write_out(const void *buffer, size_t size, void *key);
static FILE *argument_to_file(char *av[], int idx);
static char *argument_to_name(char *av[], int idx);
int opt_debug; /* -d (or -dd) */
static int opt_check; /* -c (constraints checking) */
static int opt_stack; /* -s (maximum stack size) */
static int opt_nopad; /* -per-nopad (PER input is not padded between msgs) */
static int opt_onepdu; /* -1 (decode single PDU) */
#ifdef JUNKTEST /* Enable -J <probability> */
#define JUNKOPT "J:"
static double opt_jprob; /* Junk bit probability */
static int junk_failures;
static void junk_bytes_with_probability(uint8_t *, size_t, double prob);
#define RANDOPT "R:"
static ssize_t random_max_size = 0; /* Size of the random data */
#if !defined(__FreeBSD__) && !(defined(__APPLE__) && defined(__MACH__))
static void
srandomdev(void) {
FILE *f = fopen("/dev/urandom", "rb");
unsigned seed;
if(f) {
if(fread(&seed, 1, sizeof(seed), f) != sizeof(seed)) {
seed = time(NULL);
}
fclose(f);
} else {
seed = time(NULL);
}
srandom(seed);
}
#endif
#else /* !JUNKTEST */
#define JUNKOPT
#define RANDOPT
#endif /* JUNKTEST */
/* Debug output function */
static void CC_PRINTFLIKE(1, 2)
DEBUG(const char *fmt, ...) {
va_list ap;
if(!opt_debug) return;
fprintf(stderr, "AD: ");
va_start(ap, fmt);
vfprintf(stderr, fmt, ap);
va_end(ap);
fprintf(stderr, "\n");
}
static const char *
ats_simple_name(enum asn_transfer_syntax syntax) {
switch(syntax) {
case ATS_INVALID:
return "/dev/null";
case ATS_NONSTANDARD_PLAINTEXT:
return "plaintext";
case ATS_BER:
return "BER";
case ATS_DER:
return "DER";
case ATS_CER:
return "CER";
case ATS_BASIC_OER:
case ATS_CANONICAL_OER:
return "OER";
case ATS_BASIC_XER:
case ATS_CANONICAL_XER:
return "XER";
case ATS_UNALIGNED_BASIC_PER:
case ATS_UNALIGNED_CANONICAL_PER:
return "PER";
default:
return "<?>";
}
}
#define CODEC_OFFSET(fname) ((ptrdiff_t)&(((asn_TYPE_operation_t *)0)->fname))
typedef struct {
const char *name;
enum asn_transfer_syntax syntax;
ptrdiff_t codec_offset;
const char *full_name;
} syntax_selector;
static syntax_selector input_encodings[] = {
{"ber", ATS_BER, CODEC_OFFSET(ber_decoder),
"Input is in BER (Basic Encoding Rules) or DER"},
{"oer", ATS_BASIC_OER, CODEC_OFFSET(oer_decoder),
"Input is in OER (Octet Encoding Rules)"},
{"per", ATS_UNALIGNED_BASIC_PER, CODEC_OFFSET(uper_decoder),
"Input is in Unaligned PER (Packed Encoding Rules)"},
{"aper", ATS_ALIGNED_BASIC_PER, CODEC_OFFSET(aper_decoder),
"Input is in Aligned PER (Packed Encoding Rules)"},
{"xer", ATS_BASIC_XER, CODEC_OFFSET(xer_decoder),
"Input is in XER (XML Encoding Rules)"},
{0, ATS_INVALID, 0, 0}};
static syntax_selector output_encodings[] = {
{"der", ATS_DER, CODEC_OFFSET(der_encoder),
"Output as DER (Distinguished Encoding Rules)"},
{"oer", ATS_CANONICAL_OER, CODEC_OFFSET(oer_encoder),
"Output as Canonical OER (Octet Encoding Rules)"},
{"per", ATS_UNALIGNED_CANONICAL_PER, CODEC_OFFSET(uper_encoder),
"Output as Unaligned PER (Packed Encoding Rules)"},
{"aper", ATS_ALIGNED_CANONICAL_PER, CODEC_OFFSET(aper_encoder),
"Output as Aligned PER (Packed Encoding Rules)"},
{"xer", ATS_BASIC_XER, CODEC_OFFSET(xer_encoder),
"Output as XER (XML Encoding Rules)"},
{"text", ATS_NONSTANDARD_PLAINTEXT, CODEC_OFFSET(print_struct),
"Output as plain semi-structured text"},
{"null", ATS_INVALID, CODEC_OFFSET(print_struct),
"Verify (decode) input, but do not output"},
{0, ATS_INVALID, 0, 0}};
static int
has_codec_defined(const asn_TYPE_descriptor_t *td,
const syntax_selector *element) {
return *(const void *const *)(const void *)((const char *)td->op
+ element->codec_offset) != 0;
}
/*
* Select ASN.1 Transfer Enocoding Syntax by command line name.
*/
static const syntax_selector *
ats_by_name(const char *name, const asn_TYPE_descriptor_t *td,
const syntax_selector *first_element) {
const syntax_selector *element;
for(element = first_element; element->name; element++) {
if(strcmp(element->name, name) == 0) {
if(td && td->op && has_codec_defined(td, element)) {
return element;
}
}
}
return NULL;
}
int
main(int ac, char *av[]) {
FILE *binary_out;
asn_TYPE_descriptor_t *pduType = PDU_Type_Ptr;
asn_TYPE_descriptor_t *anyPduType = PDU_Type_Ptr;
ssize_t suggested_bufsize = 8192; /* close or equal to stdio buffer */
int number_of_iterations = 1;
int num;
int ch;
const syntax_selector *sel;
enum asn_transfer_syntax isyntax = ATS_INVALID;
enum asn_transfer_syntax osyntax = ATS_BASIC_XER;
if(!anyPduType) {
#ifdef ASN_PDU_COLLECTION
anyPduType = asn_pdu_collection[0];
if(!anyPduType) {
fprintf(stderr,
"Empty PDU collection, no reference PDU to choose from.\n");
exit(EX_SOFTWARE);
}
#else
fprintf(stderr, "Either asn1c -pdu=... or cc -DPDU should be used.\n");
exit(EX_SOFTWARE);
#endif
}
/* Figure out if a specialty decoder needs to be default */
#ifndef ASN_DISABLE_OER_SUPPORT
isyntax = ATS_BASIC_OER;
#endif
#ifndef ASN_DISABLE_PER_SUPPORT
isyntax = ATS_UNALIGNED_BASIC_PER;
#endif
/*
* Pocess the command-line argments.
*/
while((ch = getopt(ac, av, "i:o:1b:cdn:p:hs:" JUNKOPT RANDOPT)) != -1)
switch(ch) {
case 'i':
sel = ats_by_name(optarg, anyPduType, input_encodings);
if(sel) {
isyntax = sel->syntax;
} else {
fprintf(stderr, "-i<format>: '%s': improper format selector\n",
optarg);
exit(EX_UNAVAILABLE);
}
break;
case 'o':
sel = ats_by_name(optarg, anyPduType, output_encodings);
if(sel) {
osyntax = sel->syntax;
} else {
fprintf(stderr, "-o<format>: '%s': improper format selector\n",
optarg);
exit(EX_UNAVAILABLE);
}
break;
case '1':
opt_onepdu = 1;
break;
case 'b':
suggested_bufsize = atoi(optarg);
if(suggested_bufsize < 1
|| suggested_bufsize > 16 * 1024 * 1024) {
fprintf(stderr,
"-b %s: Improper buffer size (1..16M)\n",
optarg);
exit(EX_UNAVAILABLE);
}
break;
case 'c':
opt_check = 1;
break;
case 'd':
opt_debug++; /* Double -dd means ASN.1 debug */
break;
case 'n':
number_of_iterations = atoi(optarg);
if(number_of_iterations < 1) {
fprintf(stderr,
"-n %s: Improper iterations count\n", optarg);
exit(EX_UNAVAILABLE);
}
break;
case 'p':
if(strcmp(optarg, "er-nopad") == 0) {
opt_nopad = 1;
break;
}
#ifdef ASN_PDU_COLLECTION
if(strcmp(optarg, "list") == 0) {
asn_TYPE_descriptor_t **pdu = asn_pdu_collection;
fprintf(stderr, "Available PDU types:\n");
for(; *pdu; pdu++) printf("%s\n", (*pdu)->name);
exit(0);
} else if(optarg[0] >= 'A' && optarg[0] <= 'Z') {
asn_TYPE_descriptor_t **pdu = asn_pdu_collection;
while(*pdu && strcmp((*pdu)->name, optarg)) pdu++;
if(*pdu) { pduType = *pdu; break; }
fprintf(stderr, "-p %s: Unrecognized PDU. Try '-p list'.\n",
optarg);
exit(EX_USAGE);
}
#else /* Without -pdu=auto there's just a single type */
if(strcmp(optarg, "list") == 0) {
fprintf(stderr, "Available PDU types:\n");
if(pduType) {
printf("%s\n", pduType->name);
}
exit(0);
} else if(optarg[0] >= 'A' && optarg[0] <= 'Z') {
if(pduType && strcmp(optarg, pduType->name) == 0) {
break;
}
fprintf(stderr, "-p %s: Unrecognized PDU. Try '-p list'.\n",
optarg);
exit(EX_USAGE);
}
#endif /* ASN_PDU_COLLECTION */
fprintf(stderr, "-p %s: Unrecognized option\n", optarg);
exit(EX_UNAVAILABLE);
case 's':
opt_stack = atoi(optarg);
if(opt_stack < 0) {
fprintf(stderr,
"-s %s: Non-negative value expected\n",
optarg);
exit(EX_UNAVAILABLE);
}
break;
#ifdef JUNKTEST
case 'J':
opt_jprob = strtod(optarg, 0);
if(opt_jprob <= 0.0 || opt_jprob > 1.0) {
fprintf(stderr,
"-J %s: Probability range 0..1 expected \n",
optarg);
exit(EX_UNAVAILABLE);
}
break;
case 'R':
isyntax = ATS_RANDOM;
random_max_size = atoi(optarg);
if(random_max_size < 0) {
fprintf(stderr,
"-R %s: Non-negative value expected\n",
optarg);
exit(EX_UNAVAILABLE);
}
srandomdev();
break;
#endif /* JUNKTEST */
case 'h':
default:
#ifdef ASN_CONVERTER_TITLE
#define _AXS(x) #x
#define _ASX(x) _AXS(x)
fprintf(stderr, "%s\n", _ASX(ASN_CONVERTER_TITLE));
#endif
fprintf(stderr, "Usage: %s [options] <datafile> ...\n", av[0]);
fprintf(stderr, "Where options are:\n");
for(sel = input_encodings; sel->name; sel++) {
if(ats_by_name(sel->name, anyPduType, sel)) {
fprintf(stderr, " -i%s %s%s\n", sel->name,
sel->full_name,
(sel->syntax == isyntax) ? " (DEFAULT)" : "");
}
}
for(sel = output_encodings; sel->name; sel++) {
if(ats_by_name(sel->name, anyPduType, sel)) {
fprintf(stderr, " -o%s%s %s%s\n", sel->name,
strlen(sel->name) > 3 ? "" : " ",
sel->full_name,
(sel->syntax == osyntax) ? " (DEFAULT)" : "");
}
}
if(anyPduType->op->uper_decoder) {
fprintf(stderr,
" -per-nopad Assume PER PDUs are not padded (-iper)\n");
}
#ifdef ASN_PDU_COLLECTION
fprintf(stderr,
" -p <PDU> Specify PDU type to decode\n"
" -p list List available PDUs\n");
#endif /* ASN_PDU_COLLECTION */
fprintf(stderr,
" -1 Decode only the first PDU in file\n"
" -b <size> Set the i/o buffer size (default is %ld)\n"
" -c Check ASN.1 constraints after decoding\n"
" -d Enable debugging (-dd is even better)\n"
" -n <num> Process files <num> times\n"
" -s <size> Set the stack usage limit (default is %d)\n"
#ifdef JUNKTEST
" -J <prob> Set random junk test bit garbaging probability\n"
" -R <size> Generate a random value of roughly the given size,\n"
" instead of parsing the value from file.\n"
#endif
, (long)suggested_bufsize, ASN__DEFAULT_STACK_MAX);
exit(EX_USAGE);
}
ac -= optind;
av += optind;
if(!pduType) {
#ifdef NO_ASN_PDU
fprintf(stderr, "No -DPDU defined during compilation.\n");
exit(0);
#else
fprintf(stderr, "Use '-p <Type>' or '-p list' to select ASN.1 type.\n");
exit(EX_USAGE);
#endif /* NO_ASN_PDU */
}
if(ac < 1 && isyntax != ATS_RANDOM) {
fprintf(stderr, "%s: No input files specified. "
"Try '-h' for more information\n",
av[-optind]);
exit(EX_USAGE);
}
if(isatty(1)) {
const int is_text_output = osyntax == ATS_NONSTANDARD_PLAINTEXT
|| osyntax == ATS_BASIC_XER
|| osyntax == ATS_CANONICAL_XER;
if(is_text_output) {
binary_out = stdout;
} else {
fprintf(stderr, "(Suppressing binary output to a terminal.)\n");
binary_out = fopen("/dev/null", "wb");
if(!binary_out) {
fprintf(stderr, "Can't open /dev/null: %s\n", strerror(errno));
exit(EX_OSERR);
}
}
} else {
binary_out = stdout;
}
setvbuf(stdout, 0, _IOLBF, 0);
for(num = 0; num < number_of_iterations; num++) {
int ac_i;
/*
* Process all files in turn.
*/
for(ac_i = (isyntax == ATS_RANDOM) ? -1 : 0; ac_i < ac; ac_i++) {
asn_enc_rval_t erv = {0,0,0};
void *structure; /* Decoded structure */
FILE *file;
char *name;
int first_pdu;
if(ac_i == -1) {
file = NULL;
name = "<random value generator>";
opt_onepdu = 1;
} else {
file = argument_to_file(av, ac_i);
name = argument_to_name(av, ac_i);
}
for(first_pdu = 1; (first_pdu || !opt_onepdu); first_pdu = 0) {
/*
* Decode the encoded structure from file.
*/
#ifdef JUNKTEST
if(isyntax == ATS_RANDOM) {
structure = NULL;
if(asn_random_fill(pduType, &structure, random_max_size) != 0) {
fprintf(stderr, "Cannot generate a random value.\n");
assert(structure == NULL);
errno = EINVAL;
}
} else {
#endif
structure = data_decode_from_file(isyntax, pduType, file, name,
suggested_bufsize, first_pdu);
#ifdef JUNKTEST
}
#endif
if(!structure) {
if(errno) {
/* Error message is already printed */
exit(EX_DATAERR);
} else {
/* EOF */
break;
}
}
/* Check ASN.1 constraints */
if(opt_check) {
char errbuf[128];
size_t errlen = sizeof(errbuf);
if(asn_check_constraints(pduType, structure, errbuf, &errlen)) {
fprintf(stderr,
"%s: ASN.1 constraint "
"check failed: %s\n",
name, errbuf);
exit(EX_DATAERR);
}
}
if(osyntax == ATS_INVALID) {
#ifdef JUNKTEST
if(opt_jprob == 0.0) {
fprintf(stderr, "%s: decoded successfully\n", name);
}
#else
fprintf(stderr, "%s: decoded successfully\n", name);
#endif
} else {
erv = asn_encode(NULL, osyntax, pduType, structure, write_out,
binary_out);
if(erv.encoded == -1) {
fprintf(stderr, "%s: Cannot convert %s into %s\n", name,
pduType->name, ats_simple_name(osyntax));
DEBUG("Conversion failed for %s:\n", pduType->name);
asn_fprint(stderr, pduType, structure);
exit(EX_UNAVAILABLE);
}
DEBUG("Encoded in %" ASN_PRI_SSIZE " bytes of %s", erv.encoded,
ats_simple_name(osyntax));
}
ASN_STRUCT_FREE(*pduType, structure);
}
if(file && file != stdin) {
fclose(file);
}
}
}
#ifdef JUNKTEST
if(opt_jprob > 0.0) {
fprintf(stderr, "Junked %f OK (%d/%d)\n",
opt_jprob, junk_failures, number_of_iterations);
}
#endif /* JUNKTEST */
return 0;
}
static struct dynamic_buffer {
uint8_t *data; /* Pointer to the data bytes */
size_t offset; /* Offset from the start */
size_t length; /* Length of meaningful contents */
size_t unbits; /* Unused bits in the last byte */
size_t allocated; /* Allocated memory for data */
int nreallocs; /* Number of data reallocations */
off_t bytes_shifted; /* Number of bytes ever shifted */
} DynamicBuffer;
static void
buffer_dump() {
uint8_t *p = DynamicBuffer.data + DynamicBuffer.offset;
uint8_t *e = p + DynamicBuffer.length - (DynamicBuffer.unbits ? 1 : 0);
if(!opt_debug) return;
DEBUG("Buffer: { d=%p, o=%" ASN_PRI_SIZE ", l=%" ASN_PRI_SIZE
", u=%" ASN_PRI_SIZE ", a=%" ASN_PRI_SIZE ", s=%" ASN_PRI_SIZE " }",
(const void *)DynamicBuffer.data,
DynamicBuffer.offset,
DynamicBuffer.length,
DynamicBuffer.unbits,
DynamicBuffer.allocated,
(size_t)DynamicBuffer.bytes_shifted);
for(; p < e; p++) {
fprintf(stderr, " %c%c%c%c%c%c%c%c",
((*p >> 7) & 1) ? '1' : '0',
((*p >> 6) & 1) ? '1' : '0',
((*p >> 5) & 1) ? '1' : '0',
((*p >> 4) & 1) ? '1' : '0',
((*p >> 3) & 1) ? '1' : '0',
((*p >> 2) & 1) ? '1' : '0',
((*p >> 1) & 1) ? '1' : '0',
((*p >> 0) & 1) ? '1' : '0');
}
if(DynamicBuffer.unbits) {
unsigned int shift;
fprintf(stderr, " ");
for(shift = 7; shift >= DynamicBuffer.unbits; shift--)
fprintf(stderr, "%c", ((*p >> shift) & 1) ? '1' : '0');
fprintf(stderr, " %" ASN_PRI_SSIZE ":%" ASN_PRI_SSIZE "\n",
(ssize_t)DynamicBuffer.length - 1,
(ssize_t)8 - DynamicBuffer.unbits);
} else {
fprintf(stderr, " %ld\n", (long)DynamicBuffer.length);
}
}
/*
* Move the buffer content left N bits, possibly joining it with
* preceeding content.
*/
static void
buffer_shift_left(size_t offset, int bits) {
uint8_t *ptr = DynamicBuffer.data + DynamicBuffer.offset + offset;
uint8_t *end = DynamicBuffer.data + DynamicBuffer.offset
+ DynamicBuffer.length - 1;
if(!bits) return;
DEBUG("Shifting left %d bits off %ld (o=%ld, u=%ld, l=%ld)",
bits, (long)offset,
(long)DynamicBuffer.offset,
(long)DynamicBuffer.unbits,
(long)DynamicBuffer.length);
if(offset) {
int right;
right = ptr[0] >> (8 - bits);
DEBUG("oleft: %c%c%c%c%c%c%c%c",
((ptr[-1] >> 7) & 1) ? '1' : '0',
((ptr[-1] >> 6) & 1) ? '1' : '0',
((ptr[-1] >> 5) & 1) ? '1' : '0',
((ptr[-1] >> 4) & 1) ? '1' : '0',
((ptr[-1] >> 3) & 1) ? '1' : '0',
((ptr[-1] >> 2) & 1) ? '1' : '0',
((ptr[-1] >> 1) & 1) ? '1' : '0',
((ptr[-1] >> 0) & 1) ? '1' : '0');
DEBUG("oriht: %c%c%c%c%c%c%c%c",
((ptr[0] >> 7) & 1) ? '1' : '0',
((ptr[0] >> 6) & 1) ? '1' : '0',
((ptr[0] >> 5) & 1) ? '1' : '0',
((ptr[0] >> 4) & 1) ? '1' : '0',
((ptr[0] >> 3) & 1) ? '1' : '0',
((ptr[0] >> 2) & 1) ? '1' : '0',
((ptr[0] >> 1) & 1) ? '1' : '0',
((ptr[0] >> 0) & 1) ? '1' : '0');
DEBUG("mriht: %c%c%c%c%c%c%c%c",
((right >> 7) & 1) ? '1' : '0',
((right >> 6) & 1) ? '1' : '0',
((right >> 5) & 1) ? '1' : '0',
((right >> 4) & 1) ? '1' : '0',
((right >> 3) & 1) ? '1' : '0',
((right >> 2) & 1) ? '1' : '0',
((right >> 1) & 1) ? '1' : '0',
((right >> 0) & 1) ? '1' : '0');
ptr[-1] = (ptr[-1] & (0xff << bits)) | right;
DEBUG("after: %c%c%c%c%c%c%c%c",
((ptr[-1] >> 7) & 1) ? '1' : '0',
((ptr[-1] >> 6) & 1) ? '1' : '0',
((ptr[-1] >> 5) & 1) ? '1' : '0',
((ptr[-1] >> 4) & 1) ? '1' : '0',
((ptr[-1] >> 3) & 1) ? '1' : '0',
((ptr[-1] >> 2) & 1) ? '1' : '0',
((ptr[-1] >> 1) & 1) ? '1' : '0',
((ptr[-1] >> 0) & 1) ? '1' : '0');
}
buffer_dump();
for(; ptr < end; ptr++) {
int right = ptr[1] >> (8 - bits);
*ptr = (*ptr << bits) | right;
}
*ptr <<= bits;
DEBUG("Unbits [%" ASN_PRI_SIZE "=>", DynamicBuffer.unbits);
if(DynamicBuffer.unbits == 0) {
DynamicBuffer.unbits += bits;
} else {
DynamicBuffer.unbits += bits;
if(DynamicBuffer.unbits > 7) {
DynamicBuffer.unbits -= 8;
DynamicBuffer.length--;
DynamicBuffer.bytes_shifted++;
}
}
DEBUG("Unbits =>%" ASN_PRI_SIZE "]", DynamicBuffer.unbits);
buffer_dump();
DEBUG("Shifted. Now (o=%" ASN_PRI_SIZE ", u=%" ASN_PRI_SIZE
" l=%" ASN_PRI_SIZE ")",
DynamicBuffer.offset,
DynamicBuffer.unbits,
DynamicBuffer.length);
}
/*
* Ensure that the buffer contains at least this amount of free space.
*/
static void add_bytes_to_buffer(const void *data2add, size_t bytes) {
if(bytes == 0) return;
DEBUG("=> add_bytes(%" ASN_PRI_SIZE ") { o=%" ASN_PRI_SIZE
" l=%" ASN_PRI_SIZE " u=%" ASN_PRI_SIZE ", s=%" ASN_PRI_SIZE " }",
bytes,
DynamicBuffer.offset,
DynamicBuffer.length,
DynamicBuffer.unbits,
DynamicBuffer.allocated);
if(DynamicBuffer.allocated
>= (DynamicBuffer.offset + DynamicBuffer.length + bytes)) {
DEBUG("\tNo buffer reallocation is necessary");
} else if(bytes <= DynamicBuffer.offset) {
DEBUG("\tContents shifted by %ld", DynamicBuffer.offset);
/* Shift the buffer contents */
memmove(DynamicBuffer.data,
DynamicBuffer.data + DynamicBuffer.offset,
DynamicBuffer.length);
DynamicBuffer.bytes_shifted += DynamicBuffer.offset;
DynamicBuffer.offset = 0;
} else {
size_t newsize = (DynamicBuffer.allocated << 2) + bytes;
void *p = MALLOC(newsize);
if(!p) {
perror("malloc()");
exit(EX_OSERR);
}
if (DynamicBuffer.length) {
memcpy(p,
DynamicBuffer.data + DynamicBuffer.offset,
DynamicBuffer.length);
}
FREEMEM(DynamicBuffer.data);
DynamicBuffer.data = (uint8_t *)p;
DynamicBuffer.offset = 0;
DynamicBuffer.allocated = newsize;
DynamicBuffer.nreallocs++;
DEBUG("\tBuffer reallocated to %ld (%d time)",
newsize, DynamicBuffer.nreallocs);
}
memcpy(DynamicBuffer.data
+ DynamicBuffer.offset + DynamicBuffer.length,
data2add, bytes);
DynamicBuffer.length += bytes;
if(DynamicBuffer.unbits) {
int bits = DynamicBuffer.unbits;
DynamicBuffer.unbits = 0;
buffer_shift_left(DynamicBuffer.length - bytes, bits);
}
DEBUG("<= add_bytes(%" ASN_PRI_SIZE ") { o=%" ASN_PRI_SIZE
" l=%" ASN_PRI_SIZE " u=%" ASN_PRI_SIZE ", s=%" ASN_PRI_SIZE " }",
bytes,
DynamicBuffer.offset,
DynamicBuffer.length,
DynamicBuffer.unbits,
DynamicBuffer.allocated);
}
static int
is_syntax_PER(enum asn_transfer_syntax syntax) {
return (syntax == ATS_UNALIGNED_BASIC_PER
|| syntax == ATS_UNALIGNED_CANONICAL_PER
|| syntax == ATS_ALIGNED_BASIC_PER
|| syntax == ATS_ALIGNED_CANONICAL_PER);
}
static int
restartability_supported(enum asn_transfer_syntax syntax) {
return !is_syntax_PER(syntax);
}
static void *
data_decode_from_file(enum asn_transfer_syntax isyntax, asn_TYPE_descriptor_t *pduType, FILE *file, const char *name, ssize_t suggested_bufsize, int on_first_pdu) {
static uint8_t *fbuf;
static ssize_t fbuf_size;
static asn_codec_ctx_t s_codec_ctx;
asn_codec_ctx_t *opt_codec_ctx = 0;
void *structure = 0;
asn_dec_rval_t rval;
size_t old_offset;
size_t new_offset;
int tolerate_eof;
size_t rd;
if(!file) {
fprintf(stderr, "%s: %s\n", name, strerror(errno));
errno = EINVAL;
return 0;
}
if(opt_stack) {
s_codec_ctx.max_stack_size = opt_stack;
opt_codec_ctx = &s_codec_ctx;
}
DEBUG("Processing %s", name);
/* prepare the file buffer */
if(fbuf_size != suggested_bufsize) {
fbuf = (uint8_t *)REALLOC(fbuf, suggested_bufsize);
if(!fbuf) {
perror("realloc()");
exit(EX_OSERR);
}
fbuf_size = suggested_bufsize;
}
if(on_first_pdu) {
DynamicBuffer.offset = 0;
DynamicBuffer.length = 0;
DynamicBuffer.unbits = 0;
DynamicBuffer.allocated = 0;
DynamicBuffer.bytes_shifted = 0;
DynamicBuffer.nreallocs = 0;
}
old_offset = DynamicBuffer.bytes_shifted + DynamicBuffer.offset;
/* Pretend immediate EOF */
rval.code = RC_WMORE;
rval.consumed = 0;
for(tolerate_eof = 1; /* Allow EOF first time buffer is non-empty */
(rd = fread(fbuf, 1, fbuf_size, file))
|| feof(file) == 0
|| (tolerate_eof && DynamicBuffer.length)
;) {
int ecbits = 0; /* Extra consumed bits in case of PER */
uint8_t *i_bptr;
size_t i_size;
/*
* Copy the data over, or use the original buffer.
*/
if(DynamicBuffer.allocated) {
/* Append new data into the existing dynamic buffer */
add_bytes_to_buffer(fbuf, rd);
i_bptr = DynamicBuffer.data + DynamicBuffer.offset;
i_size = DynamicBuffer.length;
} else {
i_bptr = fbuf;
i_size = rd;
}
DEBUG("Decoding %" ASN_PRI_SIZE " bytes", i_size);
#ifdef JUNKTEST
junk_bytes_with_probability(i_bptr, i_size, opt_jprob);
#endif
if(is_syntax_PER(isyntax) && opt_nopad) {
#ifdef ASN_DISABLE_PER_SUPPORT
rval.code = RC_FAIL;
rval.consumed = 0;
#else
if(isyntax == ATS_UNALIGNED_BASIC_PER)
rval = uper_decode(opt_codec_ctx, pduType, (void **)&structure,
i_bptr, i_size, 0, DynamicBuffer.unbits);
else
rval = aper_decode(opt_codec_ctx, pduType, (void **)&structure,
i_bptr, i_size, 0, DynamicBuffer.unbits);
/* uper_decode() returns bits! */
ecbits = rval.consumed % 8; /* Bits consumed from the last byte */
rval.consumed >>= 3; /* Convert bits into bytes. */
#endif
/* Non-padded PER decoder */
} else {
rval = asn_decode(opt_codec_ctx, isyntax, pduType,
(void **)&structure, i_bptr, i_size);
}
if(rval.code == RC_WMORE && !restartability_supported(isyntax)) {
/* PER does not support restartability */
ASN_STRUCT_FREE(*pduType, structure);
structure = 0;
rval.consumed = 0;
/* Continue accumulating data */
}
DEBUG("decode(%" ASN_PRI_SIZE ") consumed %" ASN_PRI_SIZE
"+%db (%" ASN_PRI_SIZE "), code %d",
DynamicBuffer.length, rval.consumed, ecbits, i_size, rval.code);
if(DynamicBuffer.allocated == 0) {
/*
* Flush remainder into the intermediate buffer.
*/
if(rval.code != RC_FAIL && rval.consumed < rd) {
add_bytes_to_buffer(fbuf + rval.consumed,
rd - rval.consumed);
buffer_shift_left(0, ecbits);
DynamicBuffer.bytes_shifted = rval.consumed;
rval.consumed = 0;
ecbits = 0;
}
}
/*
* Adjust position inside the source buffer.
*/
if(DynamicBuffer.allocated) {
DynamicBuffer.offset += rval.consumed;
DynamicBuffer.length -= rval.consumed;
} else {
DynamicBuffer.bytes_shifted += rval.consumed;
}
switch(rval.code) {
case RC_OK:
if(ecbits) buffer_shift_left(0, ecbits);
DEBUG("RC_OK, finishing up with %ld+%d",
(long)rval.consumed, ecbits);
return structure;
case RC_WMORE:
DEBUG("RC_WMORE, continuing read=%ld, cons=%ld "
" with %ld..%ld-%ld..%ld",
(long)rd,
(long)rval.consumed,
(long)DynamicBuffer.offset,
(long)DynamicBuffer.length,
(long)DynamicBuffer.unbits,
(long)DynamicBuffer.allocated);
if(!rd) tolerate_eof--;
continue;
case RC_FAIL:
break;
}
break;
}
DEBUG("Clean up partially decoded %s", pduType->name);
ASN_STRUCT_FREE(*pduType, structure);
new_offset = DynamicBuffer.bytes_shifted + DynamicBuffer.offset;
/*
* Print a message and return failure only if not EOF,
* unless this is our first PDU (empty file).
*/
if(on_first_pdu
|| DynamicBuffer.length
|| new_offset - old_offset > ((isyntax == ATS_BASIC_XER)?sizeof("\r\n")-1:0)
) {
#ifdef JUNKTEST
/*
* Nothing's wrong with being unable to decode junk.
* Simulate EOF.
*/
if(opt_jprob != 0.0) {
junk_failures++;
errno = 0;
return 0;
}
#endif
DEBUG("ofp %d, no=%ld, oo=%ld, dbl=%ld",
on_first_pdu, (long)new_offset, (long)old_offset,
(long)DynamicBuffer.length);
fprintf(stderr, "%s: "
"Decode failed past byte %ld: %s\n",
name, (long)new_offset,
(rval.code == RC_WMORE)
? "Unexpected end of input"
: "Input processing error");
#ifndef ENOMSG
#define ENOMSG EINVAL
#endif
#ifndef EBADMSG
#define EBADMSG EINVAL
#endif
errno = (rval.code == RC_WMORE) ? ENOMSG : EBADMSG;
} else {
/* Got EOF after a few successful PDUs */
errno = 0;
}
return 0;
}
/* Dump the buffer out to the specified FILE */
static int write_out(const void *buffer, size_t size, void *key) {
FILE *fp = (FILE *)key;
return (fwrite(buffer, 1, size, fp) == size) ? 0 : -1;
}
static int argument_is_stdin(char *av[], int idx) {
if(strcmp(av[idx], "-")) {
return 0; /* Certainly not <stdin> */
} else {
/* This might be <stdin>, unless `./program -- -` */
if(strcmp(av[-1], "--"))
return 1;
else
return 0;
}
}
static FILE *argument_to_file(char *av[], int idx) {
return argument_is_stdin(av, idx) ? stdin : fopen(av[idx], "rb");
}
static char *argument_to_name(char *av[], int idx) {
return argument_is_stdin(av, idx) ? "standard input" : av[idx];
}
#ifdef JUNKTEST
/*
* Fill bytes with some garbage with specified probability (more or less).
*/
static void
junk_bytes_with_probability(uint8_t *buf, size_t size, double prob) {
static int junkmode;
uint8_t *ptr;
uint8_t *end;
if(opt_jprob <= 0.0) return;
for(ptr = buf, end = ptr + size; ptr < end; ptr++) {
int byte = *ptr;
if(junkmode++ & 1) {
if((((double)random() / RAND_MAX) < prob))
byte = random() & 0xff;
} else {
#define BPROB(b) ((((double)random() / RAND_MAX) < prob) ? b : 0)
byte ^= BPROB(0x80);
byte ^= BPROB(0x40);
byte ^= BPROB(0x20);
byte ^= BPROB(0x10);
byte ^= BPROB(0x08);
byte ^= BPROB(0x04);
byte ^= BPROB(0x02);
byte ^= BPROB(0x01);
}
if(byte != *ptr) {
*ptr = byte;
}
}
}
#endif /* JUNKTEST */
Reference in New Issue Copy Permalink