From f215e02bf85f68d3a6106c2a1f4f7f063f819064 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Thu, 11 Apr 2024 10:17:27 +0200 Subject: Adding upstream version 7.0.14-dfsg. Signed-off-by: Daniel Baumann --- src/VBox/Runtime/testcase/tstRTSystemQueryDmi.cpp | 135 ++++++++++++++++++++++ 1 file changed, 135 insertions(+) create mode 100644 src/VBox/Runtime/testcase/tstRTSystemQueryDmi.cpp (limited to 'src/VBox/Runtime/testcase/tstRTSystemQueryDmi.cpp') diff --git a/src/VBox/Runtime/testcase/tstRTSystemQueryDmi.cpp b/src/VBox/Runtime/testcase/tstRTSystemQueryDmi.cpp new file mode 100644 index 00000000..67c68406 --- /dev/null +++ b/src/VBox/Runtime/testcase/tstRTSystemQueryDmi.cpp @@ -0,0 +1,135 @@ +/* $Id: tstRTSystemQueryDmi.cpp $ */ +/** @file + * IPRT Testcase - RTSystemQueryDmi*. + */ + +/* + * Copyright (C) 2006-2023 Oracle and/or its affiliates. + * + * This file is part of VirtualBox base platform packages, as + * available from https://www.virtualbox.org. + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation, in version 3 of the + * License. + * + * This program is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, see . + * + * The contents of this file may alternatively be used under the terms + * of the Common Development and Distribution License Version 1.0 + * (CDDL), a copy of it is provided in the "COPYING.CDDL" file included + * in the VirtualBox distribution, in which case the provisions of the + * CDDL are applicable instead of those of the GPL. + * + * You may elect to license modified versions of this file under the + * terms and conditions of either the GPL or the CDDL or both. + * + * SPDX-License-Identifier: GPL-3.0-only OR CDDL-1.0 + */ + + +/********************************************************************************************************************************* +* Header Files * +*********************************************************************************************************************************/ +#include + +#include +#include +#include +#include + + +int main() +{ + RTTEST hTest; + int rc = RTTestInitAndCreate("tstRTSystemQueryDmi", &hTest); + if (rc) + return rc; + RTTestBanner(hTest); + + /* + * Simple stuff. + */ + char szInfo[_4K]; + + rc = RTSystemQueryDmiString(RTSYSDMISTR_PRODUCT_NAME, szInfo, sizeof(szInfo)); + RTTestIPrintf(RTTESTLVL_ALWAYS, "PRODUCT_NAME: \"%s\", rc=%Rrc\n", szInfo, rc); + + rc = RTSystemQueryDmiString(RTSYSDMISTR_PRODUCT_VERSION, szInfo, sizeof(szInfo)); + RTTestIPrintf(RTTESTLVL_ALWAYS, "PRODUCT_VERSION: \"%s\", rc=%Rrc\n", szInfo, rc); + + rc = RTSystemQueryDmiString(RTSYSDMISTR_PRODUCT_UUID, szInfo, sizeof(szInfo)); + RTTestIPrintf(RTTESTLVL_ALWAYS, "PRODUCT_UUID: \"%s\", rc=%Rrc\n", szInfo, rc); + + rc = RTSystemQueryDmiString(RTSYSDMISTR_PRODUCT_SERIAL, szInfo, sizeof(szInfo)); + RTTestIPrintf(RTTESTLVL_ALWAYS, "PRODUCT_SERIAL: \"%s\", rc=%Rrc\n", szInfo, rc); + + rc = RTSystemQueryDmiString(RTSYSDMISTR_MANUFACTURER, szInfo, sizeof(szInfo)); + RTTestIPrintf(RTTESTLVL_ALWAYS, "MANUFACTURER: \"%s\", rc=%Rrc\n", szInfo, rc); + + /* + * Check that unsupported stuff is terminated correctly. + */ + for (int i = RTSYSDMISTR_INVALID + 1; i < RTSYSDMISTR_END; i++) + { + memset(szInfo, ' ', sizeof(szInfo)); + rc = RTSystemQueryDmiString((RTSYSDMISTR)i, szInfo, sizeof(szInfo)); + if ( (rc == VERR_NOT_SUPPORTED || rc == VERR_ACCESS_DENIED) + && szInfo[0] != '\0') + RTTestIFailed("level=%d; unterminated buffer on VERR_NOT_SUPPORTED\n", i); + else if (RT_SUCCESS(rc) || rc == VERR_BUFFER_OVERFLOW) + RTTESTI_CHECK(RTStrEnd(szInfo, sizeof(szInfo)) != NULL); + else if (rc != VERR_NOT_SUPPORTED && rc != VERR_ACCESS_DENIED) + RTTestIFailed("level=%d unexpected rc=%Rrc\n", i, rc); + } + + /* + * Check buffer overflow + */ + RTAssertSetQuiet(true); + RTAssertSetMayPanic(false); + for (int i = RTSYSDMISTR_INVALID + 1; i < RTSYSDMISTR_END; i++) + { + RTTESTI_CHECK_RC(RTSystemQueryDmiString((RTSYSDMISTR)i, szInfo, 0), VERR_INVALID_PARAMETER); + + /* Get the length of the info and check that we get overflow errors for + everything less that it. */ + rc = RTSystemQueryDmiString((RTSYSDMISTR)i, szInfo, sizeof(szInfo)); + if (RT_FAILURE(rc)) + continue; + size_t const cchInfo = strlen(szInfo); + + for (size_t cch = 1; cch < sizeof(szInfo) && cch < cchInfo; cch++) + { + memset(szInfo, 0x7f, sizeof(szInfo)); + RTTESTI_CHECK_RC(RTSystemQueryDmiString((RTSYSDMISTR)i, szInfo, cch), VERR_BUFFER_OVERFLOW); + + /* check the padding. */ + for (size_t off = cch; off < sizeof(szInfo); off++) + if (szInfo[off] != 0x7f) + { + RTTestIFailed("level=%d, rc=%Rrc, cch=%zu, off=%zu: Wrote too much!\n", i, rc, cch, off); + break; + } + + /* check for zero terminator. */ + if (!RTStrEnd(szInfo, cch)) + RTTestIFailed("level=%d, rc=%Rrc, cch=%zu: Buffer not terminated!\n", i, rc, cch); + } + + /* Check that the exact length works. */ + rc = RTSystemQueryDmiString((RTSYSDMISTR)i, szInfo, cchInfo + 1); + if (rc != VINF_SUCCESS) + RTTestIFailed("level=%d: rc=%Rrc when specifying exactly right buffer length (%zu)\n", i, rc, cchInfo + 1); + } + + return RTTestSummaryAndDestroy(hTest); +} + -- cgit v1.2.3