1 /*
2 * Copyright (c) 2023-2023 Huawei Device Co., Ltd. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without modification,
5 * are permitted provided that the following conditions are met:
6 *
7 * 1. Redistributions of source code must retain the above copyright notice, this list of
8 * conditions and the following disclaimer.
9 *
10 * 2. Redistributions in binary form must reproduce the above copyright notice, this list
11 * of conditions and the following disclaimer in the documentation and/or other materials
12 * provided with the distribution.
13 *
14 * 3. Neither the name of the copyright holder nor the names of its contributors may be used
15 * to endorse or promote products derived from this software without specific prior written
16 * permission.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
20 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
21 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
22 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
23 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
24 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
25 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
26 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
27 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
28 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 */
30 #include <string>
31 #include <iostream>
32 #include <regex>
33 #include "It_container_test.h"
34
GenNetLinkPath(int pid)35 static std::string GenNetLinkPath(int pid)
36 {
37 std::ostringstream buf;
38 buf << "/proc/" << pid << "/container/net";
39 return buf.str();
40 }
41
ReadlinkNet(int pid)42 static std::string ReadlinkNet(int pid)
43 {
44 auto path = GenNetLinkPath(pid);
45 struct stat sb;
46
47 int ret = lstat(path.data(), &sb);
48 if (ret == -1) {
49 perror("lstat");
50 return std::string();
51 }
52
53 auto bufsiz = sb.st_size + 1;
54 if (sb.st_size == 0) {
55 bufsiz = PATH_MAX;
56 }
57
58 std::vector<char> buf(bufsiz);
59 auto nbytes = readlink(path.c_str(), buf.data(), bufsiz);
60 if (nbytes == -1) {
61 perror("readlink");
62 return std::string();
63 }
64
65 return buf.data();
66 }
67
ItNetContainer003(void)68 void ItNetContainer003(void)
69 {
70 std::string zerolink("'net:[0]'");
71 auto netlink = ReadlinkNet(getpid());
72 int ret = zerolink.compare(netlink);
73 ASSERT_NE(ret, 0);
74
75 std::regex reg("'net:\\[[0-9]+\\]'");
76 ret = std::regex_match(netlink, reg);
77 ASSERT_EQ(ret, 1);
78 }
79