PREPARE/zmap/src/fieldset.h

/*
 * ZMap Copyright 2013 Regents of the University of Michigan
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy
 * of the License at http://www.apache.org/licenses/LICENSE-2.0
 */

#include <stdlib.h>
#include <stdint.h>
#include "types.h"

#ifndef FIELDSET_H
#define FIELDSET_H

// maximum number of records that can be stored in a fieldset
#define MAX_FIELDS 128

// types of data that can be stored in a field
#define FS_STRING 0
#define FS_UINT64 1
#define FS_BINARY 2
#define FS_NULL 3

// definition of a field that's provided by a probe module
// these are used so that users can ask at the command-line
// what fields are available for consumption
typedef struct field_def {
	const char *name;
	const char *type;
	const char *desc;
} fielddef_t;

typedef struct fielddef_set {
	fielddef_t fielddefs[MAX_FIELDS];
	int len;
} fielddefset_t;

typedef union field_val {
	void	  *ptr;
	uint64_t   num;
} field_val_t;

// the internal field type used by fieldset
typedef struct field {
	const char *name;
	int type;
	int free_;
	size_t len;
	union field_val value;
} field_t;

// data structure that is populated by the probe module
// and translated into the data structure that's passed
// to the output module
typedef struct fieldset {
	int len;
	field_t fields[MAX_FIELDS];
} fieldset_t;

// we pass a different fieldset to an output module than
// the probe module generates for us because a user may
// only want certain fields and will expect them in a certain
// order. We generate a translated fieldset that contains
// only the fields we want to export to the output module.
// a translation specifies how to efficiently convert the fs
// povided by the probe module to the fs for the output module.
typedef struct translation {
	int len;
	int translation[MAX_FIELDS];
} translation_t;


fieldset_t *fs_new_fieldset(void);

char* fs_get_string_by_index(fieldset_t *fs, int index);

int fds_get_index_by_name(fielddefset_t *fds, char *name);

void gen_fielddef_set(fielddefset_t *fds, fielddef_t fs[], int len);

void fs_add_null(fieldset_t *fs, const char *name);

void fs_add_uint64(fieldset_t *fs, const char *name, uint64_t value);

void fs_add_string(fieldset_t *fs, const char *name, char *value, int free_);

void fs_add_binary(fieldset_t *fs, const char *name, size_t len,
		void *value, int free_);

// Modify
void fs_modify_null(fieldset_t *fs, const char *name);

void fs_modify_uint64(fieldset_t *fs, const char *name, uint64_t value);

void fs_modify_string(fieldset_t *fs, const char *name, char *value, int free_);

void fs_modify_binary(fieldset_t *fs, const char *name, size_t len,
		void *value, int free_);

uint64_t fs_get_uint64_by_index(fieldset_t *fs, int index);

void fs_free(fieldset_t *fs);

void fs_generate_fieldset_translation(translation_t *t,
		fielddefset_t *avail, char** req, int reqlen);

fieldset_t *translate_fieldset(fieldset_t *fs, translation_t *t);

void fs_generate_full_fieldset_translation(translation_t *t, fielddefset_t *avail);

#endif // FIELDSET_H