Permalink
Cannot retrieve contributors at this time
Name already in use
A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
glibc/inet/test-inet6_opt.c
Go to fileThis commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
211 lines (196 sloc)
4.25 KB
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#include <netinet/in.h> | |
#include <stdio.h> | |
#include <stdlib.h> | |
#include <string.h> | |
#include <stdint.h> | |
#define OPT_X 42 | |
#define OPT_Y 43 | |
#define OPT_Z 44 | |
static void * | |
encode_inet6_opt (socklen_t *elp) | |
{ | |
void *eb = NULL; | |
socklen_t el; | |
int cl; | |
void *db; | |
int offset; | |
uint8_t val1; | |
uint16_t val2; | |
uint32_t val4; | |
uint64_t val8; | |
*elp = 0; | |
#define CHECK() \ | |
if (cl == -1) \ | |
{ \ | |
printf ("cl == -1 on line %d\n", __LINE__); \ | |
free (eb); \ | |
return NULL; \ | |
} | |
/* Estimate the length */ | |
cl = inet6_opt_init (NULL, 0); | |
CHECK (); | |
cl = inet6_opt_append (NULL, 0, cl, OPT_X, 12, 8, NULL); | |
CHECK (); | |
cl = inet6_opt_append (NULL, 0, cl, OPT_Y, 7, 4, NULL); | |
CHECK (); | |
cl = inet6_opt_append (NULL, 0, cl, OPT_Z, 7, 1, NULL); | |
CHECK (); | |
cl = inet6_opt_finish (NULL, 0, cl); | |
CHECK (); | |
el = cl; | |
eb = malloc (el + 8); | |
if (eb == NULL) | |
{ | |
puts ("malloc failed"); | |
return NULL; | |
} | |
/* Canary. */ | |
memcpy (eb + el, "deadbeef", 8); | |
cl = inet6_opt_init (eb, el); | |
CHECK (); | |
cl = inet6_opt_append (eb, el, cl, OPT_X, 12, 8, &db); | |
CHECK (); | |
val4 = 0x12345678; | |
offset = inet6_opt_set_val (db, 0, &val4, sizeof (val4)); | |
val8 = 0x0102030405060708LL; | |
inet6_opt_set_val (db, offset, &val8, sizeof (val8)); | |
cl = inet6_opt_append (eb, el, cl, OPT_Y, 7, 4, &db); | |
CHECK (); | |
val1 = 0x01; | |
offset = inet6_opt_set_val (db, 0, &val1, sizeof (val1)); | |
val2 = 0x1331; | |
offset = inet6_opt_set_val (db, offset, &val2, sizeof (val2)); | |
val4 = 0x01020304; | |
inet6_opt_set_val (db, offset, &val4, sizeof (val4)); | |
cl = inet6_opt_append (eb, el, cl, OPT_Z, 7, 1, &db); | |
CHECK (); | |
inet6_opt_set_val (db, 0, (void *) "abcdefg", 7); | |
cl = inet6_opt_finish (eb, el, cl); | |
CHECK (); | |
if (memcmp (eb + el, "deadbeef", 8) != 0) | |
{ | |
puts ("Canary corrupted"); | |
free (eb); | |
return NULL; | |
} | |
*elp = el; | |
return eb; | |
} | |
int | |
decode_inet6_opt (void *eb, socklen_t el) | |
{ | |
int ret = 0; | |
int seq = 0; | |
int cl = 0; | |
int offset; | |
uint8_t type; | |
socklen_t len; | |
uint8_t val1; | |
uint16_t val2; | |
uint32_t val4; | |
uint64_t val8; | |
void *db; | |
char buf[8]; | |
while ((cl = inet6_opt_next (eb, el, cl, &type, &len, &db)) != -1) | |
switch (type) | |
{ | |
case OPT_X: | |
if (seq++ != 0) | |
{ | |
puts ("OPT_X is not first"); | |
ret = 1; | |
} | |
if (len != 12) | |
{ | |
printf ("OPT_X's length %d != 12\n", len); | |
ret = 1; | |
} | |
offset = inet6_opt_get_val (db, 0, &val4, sizeof (val4)); | |
if (val4 != 0x12345678) | |
{ | |
printf ("OPT_X's val4 %x != 0x12345678\n", val4); | |
ret = 1; | |
} | |
offset = inet6_opt_get_val (db, offset, &val8, sizeof (val8)); | |
if (offset != len || val8 != 0x0102030405060708LL) | |
{ | |
printf ("OPT_X's val8 %llx != 0x0102030405060708\n", | |
(long long) val8); | |
ret = 1; | |
} | |
break; | |
case OPT_Y: | |
if (seq++ != 1) | |
{ | |
puts ("OPT_Y is not second"); | |
ret = 1; | |
} | |
if (len != 7) | |
{ | |
printf ("OPT_Y's length %d != 7\n", len); | |
ret = 1; | |
} | |
offset = inet6_opt_get_val (db, 0, &val1, sizeof (val1)); | |
if (val1 != 0x01) | |
{ | |
printf ("OPT_Y's val1 %x != 0x01\n", val1); | |
ret = 1; | |
} | |
offset = inet6_opt_get_val (db, offset, &val2, sizeof (val2)); | |
if (val2 != 0x1331) | |
{ | |
printf ("OPT_Y's val2 %x != 0x1331\n", val2); | |
ret = 1; | |
} | |
offset = inet6_opt_get_val (db, offset, &val4, sizeof (val4)); | |
if (offset != len || val4 != 0x01020304) | |
{ | |
printf ("OPT_Y's val4 %x != 0x01020304\n", val4); | |
ret = 1; | |
} | |
break; | |
case OPT_Z: | |
if (seq++ != 2) | |
{ | |
puts ("OPT_Z is not third"); | |
ret = 1; | |
} | |
if (len != 7) | |
{ | |
printf ("OPT_Z's length %d != 7\n", len); | |
ret = 1; | |
} | |
offset = inet6_opt_get_val (db, 0, buf, 7); | |
if (offset != len || memcmp (buf, "abcdefg", 7) != 0) | |
{ | |
buf[7] = '\0'; | |
printf ("OPT_Z's buf \"%s\" != \"abcdefg\"\n", buf); | |
ret = 1; | |
} | |
break; | |
default: | |
printf ("Unknown option %d\n", type); | |
ret = 1; | |
break; | |
} | |
if (seq != 3) | |
{ | |
puts ("Didn't see all of OPT_X, OPT_Y and OPT_Z"); | |
ret = 1; | |
} | |
return ret; | |
} | |
static int | |
do_test (void) | |
{ | |
void *eb; | |
socklen_t el; | |
eb = encode_inet6_opt (&el); | |
if (eb == NULL) | |
return 1; | |
if (decode_inet6_opt (eb, el)) | |
return 1; | |
return 0; | |
} | |
#define TEST_FUNCTION do_test () | |
#include "../test-skeleton.c" |