/* $NetBSD: util.h,v 1.2.6.1 2024/02/29 12:30:17 martin Exp $ */ /* * Copyright (C) Internet Systems Consortium, Inc. ("ISC") * * SPDX-License-Identifier: MPL-2.0 AND ISC * * This Source Code Form is subject to the terms of the Mozilla Public * License, v. 2.0. If a copy of the MPL was not distributed with this * file, you can obtain one at https://mozilla.org/MPL/2.0/. * * See the COPYRIGHT file distributed with this work for additional * information regarding copyright ownership. */ /* * Copyright (C) Red Hat * * Permission to use, copy, modify, and/or distribute this software for any * purpose with or without fee is hereby granted, provided that the above * copyright notice and this permission notice appear in all copies. * * THE SOFTWARE IS PROVIDED "AS IS" AND AUTHORS DISCLAIMS ALL WARRANTIES WITH * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY * AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT, * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR * PERFORMANCE OF THIS SOFTWARE. */ /* * Memory allocation and error handling utilities. */ #pragma once #include #include #include "log.h" #define CLEANUP_WITH(result_code) \ do { \ result = (result_code); \ goto cleanup; \ } while (0) #define CHECK(op) \ do { \ result = (op); \ if (result != ISC_R_SUCCESS) \ goto cleanup; \ } while (0) #define CHECKED_MEM_GET(m, target_ptr, s) \ do { \ (target_ptr) = isc_mem_get((m), (s)); \ if ((target_ptr) == NULL) { \ result = ISC_R_NOMEMORY; \ log_error("Memory allocation failed"); \ goto cleanup; \ } \ } while (0) #define CHECKED_MEM_GET_PTR(m, target_ptr) \ CHECKED_MEM_GET(m, target_ptr, sizeof(*(target_ptr))) #define CHECKED_MEM_STRDUP(m, source, target) \ do { \ (target) = isc_mem_strdup((m), (source)); \ if ((target) == NULL) { \ result = ISC_R_NOMEMORY; \ log_error("Memory allocation failed"); \ goto cleanup; \ } \ } while (0) #define ZERO_PTR(ptr) memset((ptr), 0, sizeof(*(ptr))) #define MEM_PUT_AND_DETACH(target_ptr) \ isc_mem_putanddetach(&(target_ptr)->mctx, target_ptr, \ sizeof(*(target_ptr)))