llvm-project/libc/test/loader/linux/args_test.cpp

41 lines
1.2 KiB
C++

//===-- Loader test to check args to main ---------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#undef NDEBUG
#include "src/assert/assert.h"
static bool my_streq(const char *lhs, const char *rhs) {
const char *l, *r;
for (l = lhs, r = rhs; *l != '\0' && *r != '\0'; ++l, ++r)
if (*l != *r)
return false;
return *l == '\0' && *r == '\0';
}
int main(int argc, char **argv, char **envp) {
assert(argc == 4 && "Unexpected argc.");
assert(my_streq(argv[1], "1") && "Unexpected argv[1].");
assert(my_streq(argv[2], "2") && "Unexpected argv[2].");
assert(my_streq(argv[3], "3") && "Unexpected argv[3].");
bool found_france = false;
bool found_germany = false;
for (; *envp != nullptr; ++envp) {
if (my_streq(*envp, "FRANCE=Paris"))
found_france = true;
if (my_streq(*envp, "GERMANY=Berlin"))
found_germany = true;
}
assert(found_france && found_germany &&
"Did not find whats expected in envp.");
return 0;
}