1/*	$NetBSD: rw_oom.c,v 1.1.1.1 2019/12/22 12:34:06 skrll Exp $	*/
2
3// SPDX-License-Identifier: LGPL-2.1-or-later
4/*
5 * libfdt - Flat Device Tree manipulation
6 *	Testcase for fdt_nop_node()
7 * Copyright (C) 2006 David Gibson, IBM Corporation.
8 */
9
10#include <stdlib.h>
11#include <stdio.h>
12#include <string.h>
13#include <ctype.h>
14#include <stdint.h>
15
16#include <libfdt.h>
17
18#include "tests.h"
19#include "testdata.h"
20
21/* This is not derived programmatically. May require adjustment to changes. */
22#define SPACE	285
23
24#define CHECK(code) \
25	{ \
26		err = (code); \
27		if (err) \
28			FAIL(#code ": %s", fdt_strerror(err)); \
29	}
30
31#define OFF_CHECK(off, code) \
32	{ \
33		(off) = (code); \
34		if (off < 0) \
35			FAIL(#code ": %s", fdt_strerror(off)); \
36	}
37
38int main(int argc, char *argv[])
39{
40	void *fdt;
41	int err;
42	int offset, s1;
43	int strsize1, strsize2;
44
45	/*
46	 * Check OOM path, and check that property is cleaned up if it fails
47	 * with OOM, rather than adding an orphan name.
48	 *
49	 * SW OOM is tested with the realloc/resize strategies.
50	 */
51	test_init(argc, argv);
52
53	fdt = xmalloc(SPACE);
54
55	/* First create empty tree with SW */
56	CHECK(fdt_create_empty_tree(fdt, SPACE));
57
58	CHECK(fdt_add_mem_rsv(fdt, TEST_ADDR_1, TEST_SIZE_1));
59	CHECK(fdt_add_mem_rsv(fdt, TEST_ADDR_2, TEST_SIZE_2));
60
61	CHECK(fdt_setprop_string(fdt, 0, "compatible", "test_oom"));
62	CHECK(fdt_setprop_u32(fdt, 0, "prop-int", TEST_VALUE_1));
63	CHECK(fdt_setprop_u64(fdt, 0, "prop-int64", TEST_VALUE64_1));
64	CHECK(fdt_setprop_string(fdt, 0, "prop-str", TEST_STRING_1));
65
66	OFF_CHECK(offset, fdt_add_subnode(fdt, 0, "subnode@1"));
67	s1 = offset;
68
69	strsize1 = fdt_size_dt_strings(fdt);
70	err = fdt_setprop_string(fdt, s1, "unique", "subnode1");
71	if (err != -FDT_ERR_NOSPACE)
72		FAIL("fdt_setprop_string(fdt, s1, \"compatible\", \"subnode1\"): %s", fdt_strerror(err));
73	strsize2 = fdt_size_dt_strings(fdt);
74
75	if (strsize1 != strsize2)
76		FAIL("fdt_setprop NOSPACE error failed to clean up allocated string\n");
77	err = 0;
78
79	/* Ensure we failed in the right place */
80	CHECK(fdt_setprop_string(fdt, s1, "unique", ""));
81
82	CHECK(fdt_pack(fdt));
83
84	PASS();
85}
86