57 static const char* buffer_str =
"buffer";
67 if (!allocator || !capacity) {
74 buffer->
data = (uint8_t*) calloc(capacity,
sizeof(uint8_t));
76 buffer->
limit = capacity;
91 buffer->
data = (uint8_t*) data;
186 get_bit(uint8_t bits[],
size_t index)
188 return bits[index / 8] & (1 << (7 - index % 8));
197 set_bit(uint8_t bits[],
size_t index)
199 bits[index / 8] |= (1 << (7 - index % 8));
209 label_is_pointer(
const uint8_t* label)
212 return (label[0] & 0xc0) == 0xc0;
221 label_pointer_location(
const uint8_t* label)
225 return ((uint16_t) (label[0] & ~0xc0) << 8) | (uint16_t) label[1];
234 label_is_normal(
const uint8_t* label)
237 return (label[0] & 0xc0) == 0;
245 label_is_root(
const uint8_t* label)
248 return label[0] == 0;
257 label_length(
const uint8_t* label)
274 size_t dname_length = 0;
275 const uint8_t *label = NULL;
290 if (label_is_pointer(label)) {
292 if (!allow_pointers) {
298 pointer = label_pointer_location(label);
307 }
else if (label_is_normal(label)) {
308 size_t length = label_length(label) + 1;
309 done = label_is_root(label);
317 dname_length += length;
338 uint8_t label_size = 0;
343 if (label_size == 0) {
345 }
else if ((label_size & 0xc0) != 0) {
400 return buffer->
limit;
413 buffer->
limit = limit;
442 return buffer->
data + at;
487 buffer_remaining_at(
buffer_type* buffer,
size_t at)
491 return buffer->
limit - at;
503 return buffer_remaining_at(buffer, buffer->
position);
512 buffer_available_at(
buffer_type *buffer,
size_t at,
size_t count)
515 return count <= buffer_remaining_at(buffer, at);
527 return buffer_available_at(buffer, buffer->
position, count);
536 buffer_write_u8_at(
buffer_type* buffer,
size_t at, uint8_t data)
540 buffer->
data[at] = data;
554 write_uint16(buffer->
data + at, data);
564 buffer_write_u32_at(
buffer_type* buffer,
size_t at, uint32_t data)
568 write_uint32(buffer->
data + at, data);
596 buffer_write_u8_at(buffer, buffer->
position, data);
624 buffer_write_u32_at(buffer, buffer->
position, data);
639 buffer_write(buffer, ldns_rdf_data(rdf), ldns_rdf_size(rdf));
654 size_t rdlength_pos = 0;
655 uint16_t rdlength = 0;
666 sizeof(uint32_t) +
sizeof(rdlength))) {
676 for (i=0; i < ldns_rr_rd_count(rr); i++) {
703 return buffer->
data[at];
716 return read_uint16(buffer->
data + at);
728 return read_uint32(buffer->
data + at);
756 result = buffer_read_u8_at(buffer, buffer->
position);
757 buffer->
position +=
sizeof(uint8_t);
771 result = buffer_read_u16_at(buffer, buffer->
position);
772 buffer->
position +=
sizeof(uint16_t);
786 result = buffer_read_u32_at(buffer, buffer->
position);
787 buffer->
position +=
sizeof(uint32_t);
800 return buffer_read_u16_at(buffer, 0);
810 return ldns_get_random();
836 return (uint16_t) buffer_read_u16_at(buffer, 2);
861 return (
int)
QR(buffer);
899 return (ldns_pkt_opcode)
OPCODE(buffer);
924 return (
int)
AA(buffer);
949 return (
int)
TC(buffer);
961 return (
int)
RD(buffer);
973 return (
int)
RA(buffer);
985 return (
int)
AD(buffer);
997 return (
int)
CD(buffer);
1009 return (ldns_pkt_rcode)
RCODE(buffer);
1034 return buffer_read_u16_at(buffer, 4);
1059 return buffer_read_u16_at(buffer, 6);
1084 return buffer_read_u16_at(buffer, 8);
1109 return buffer_read_u16_at(buffer, 10);
1131 buffer_pkt_new(
buffer_type* buffer, ldns_rdf* qname, ldns_rr_type qtype,
1132 ldns_rr_class qclass, ldns_pkt_opcode opcode)
1163 ldns_rr_class qclass)
1165 buffer_pkt_new(buffer, qname, qtype, qclass, LDNS_PACKET_QUERY);
1178 buffer_pkt_new(buffer, qname, LDNS_RR_TYPE_SOA, qclass,
1179 LDNS_PACKET_NOTIFY);
1191 buffer_pkt_new(buffer, qname, LDNS_RR_TYPE_AXFR, qclass,
1205 ldns_status status = LDNS_STATUS_OK;
1206 ldns_pkt* pkt = NULL;
1211 if (status == LDNS_STATUS_OK) {
1213 ldns_pkt_print(fd, pkt);
1216 fprintf(fd,
";;\n");
1217 fprintf(fd,
";; Bogus packet: %s\n", ldns_get_errorstr_by_id(status));
1218 fprintf(fd,
";;\n");
1219 fprintf(fd,
";;\n");
1233 if (!buffer || !allocator) {
1236 free((
void*)buffer->
data);
ldns_pkt_opcode buffer_pkt_opcode(buffer_type *buffer)
int buffer_pkt_rd(buffer_type *buffer)
void buffer_pkt_set_random_id(buffer_type *buffer)
uint8_t * buffer_at(buffer_type *buffer, size_t at)
uint16_t buffer_pkt_arcount(buffer_type *buffer)
#define BUFFER_PKT_HEADER_SIZE
void * allocator_alloc(allocator_type *allocator, size_t size)
uint16_t buffer_pkt_qdcount(buffer_type *buffer)
void buffer_skip(buffer_type *buffer, ssize_t count)
void buffer_pkt_set_flags(buffer_type *buffer, uint16_t flags)
uint16_t buffer_read_u16(buffer_type *buffer)
int buffer_skip_rr(buffer_type *buffer, unsigned qrr)
void buffer_flip(buffer_type *buffer)
void buffer_pkt_set_qdcount(buffer_type *buffer, uint16_t count)
void buffer_clear(buffer_type *buffer)
#define OPCODE_SET(packet, opcode)
uint8_t buffer_read_u8(buffer_type *buffer)
void buffer_write_u8(buffer_type *buffer, uint8_t data)
void ods_log_error(const char *format,...)
int buffer_pkt_qr(buffer_type *buffer)
void buffer_pkt_print(FILE *fd, buffer_type *buffer)
void buffer_write(buffer_type *buffer, const void *data, size_t count)
uint16_t buffer_pkt_id(buffer_type *buffer)
uint8_t * buffer_current(buffer_type *buffer)
buffer_type * buffer_create(allocator_type *allocator, size_t capacity)
uint16_t buffer_pkt_ancount(buffer_type *buffer)
size_t buffer_limit(buffer_type *buffer)
void buffer_rewind(buffer_type *buffer)
void buffer_pkt_set_ancount(buffer_type *buffer, uint16_t count)
void buffer_pkt_set_nscount(buffer_type *buffer, uint16_t count)
void buffer_set_limit(buffer_type *buffer, size_t limit)
uint16_t buffer_pkt_flags(buffer_type *buffer)
void buffer_pkt_axfr(buffer_type *buffer, ldns_rdf *qname, ldns_rr_class qclass)
size_t buffer_capacity(buffer_type *buffer)
int buffer_skip_dname(buffer_type *buffer)
uint32_t buffer_read_u32(buffer_type *buffer)
void buffer_pkt_set_opcode(buffer_type *buffer, ldns_pkt_opcode opcode)
void buffer_pkt_query(buffer_type *buffer, ldns_rdf *qname, ldns_rr_type qtype, ldns_rr_class qclass)
void buffer_write_u16(buffer_type *buffer, uint16_t data)
size_t buffer_read_dname(buffer_type *buffer, uint8_t *dname, unsigned allow_pointers)
void buffer_write_u32(buffer_type *buffer, uint32_t data)
void buffer_pkt_set_aa(buffer_type *buffer)
void buffer_read(buffer_type *buffer, void *data, size_t count)
int buffer_pkt_aa(buffer_type *buffer)
uint16_t buffer_pkt_nscount(buffer_type *buffer)
void buffer_write_u16_at(buffer_type *buffer, size_t at, uint16_t data)
void buffer_set_position(buffer_type *buffer, size_t pos)
void buffer_pkt_notify(buffer_type *buffer, ldns_rdf *qname, ldns_rr_class qclass)
int buffer_available(buffer_type *buffer, size_t count)
int buffer_pkt_ad(buffer_type *buffer)
#define RCODE_SET(packet, rcode)
void buffer_pkt_clear_qr(buffer_type *buffer)
size_t buffer_remaining(buffer_type *buffer)
void buffer_pkt_set_rcode(buffer_type *buffer, ldns_pkt_rcode rcode)
void buffer_write_rdf(buffer_type *buffer, ldns_rdf *rdf)
int buffer_write_rr(buffer_type *buffer, ldns_rr *rr)
int buffer_pkt_ra(buffer_type *buffer)
void allocator_deallocate(allocator_type *allocator, void *data)
void buffer_create_from(buffer_type *buffer, void *data, size_t size)
size_t buffer_position(buffer_type *buffer)
void buffer_cleanup(buffer_type *buffer, allocator_type *allocator)
void buffer_pkt_set_arcount(buffer_type *buffer, uint16_t count)
#define ods_log_assert(x)
int buffer_pkt_cd(buffer_type *buffer)
void buffer_pkt_set_qr(buffer_type *buffer)
uint8_t * buffer_begin(buffer_type *buffer)
ldns_pkt_rcode buffer_pkt_rcode(buffer_type *buffer)
uint8_t * buffer_end(buffer_type *buffer)
int buffer_pkt_tc(buffer_type *buffer)