a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt/*
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt * Memory allocation and error handling utilities.
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt *
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt * Copyright (C) 2009-2015 Red Hat ; see COPYRIGHT for license
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt */
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt#ifndef _LD_UTIL_H_
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt#define _LD_UTIL_H_
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt#include <isc/mem.h>
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt#include <dns/types.h>
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt#include "log.h"
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt#define CLEANUP_WITH(result_code) \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt do { \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt result = (result_code); \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt goto cleanup; \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt } while(0)
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt#define CHECK(op) \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt do { \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt result = (op); \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt if (result != ISC_R_SUCCESS) \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt goto cleanup; \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt } while (0)
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt#define CHECKED_MEM_GET(m, target_ptr, s) \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt do { \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt (target_ptr) = isc_mem_get((m), (s)); \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt if ((target_ptr) == NULL) { \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt result = ISC_R_NOMEMORY; \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt log_error("Memory allocation failed"); \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt goto cleanup; \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt } \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt } while (0)
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt#define CHECKED_MEM_GET_PTR(m, target_ptr) \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt CHECKED_MEM_GET(m, target_ptr, sizeof(*(target_ptr)))
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt#define CHECKED_MEM_STRDUP(m, source, target) \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt do { \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt (target) = isc_mem_strdup((m), (source)); \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt if ((target) == NULL) { \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt result = ISC_R_NOMEMORY; \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt log_error("Memory allocation failed"); \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt goto cleanup; \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt } \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt } while (0)
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt#define ZERO_PTR(ptr) memset((ptr), 0, sizeof(*(ptr)))
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt#define MEM_PUT_AND_DETACH(target_ptr) \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt isc_mem_putanddetach(&(target_ptr)->mctx, target_ptr, \
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt sizeof(*(target_ptr)))
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt
a00f9e2f50675bd43cc6a9fe2669709162a2ccb4Evan Hunt#endif /* !_LD_UTIL_H_ */