mirror of
https://github.com/google/sandboxed-api.git
synced 2024-03-22 13:11:30 +08:00
cb8efdc270
Try to move the affected FDs transparently to avoid conflict. PiperOrigin-RevId: 480105375 Change-Id: I0cd093fce120505d1cd4a1d081b3c0e63bf0210a
111 lines
3.3 KiB
C++
111 lines
3.3 KiB
C++
// Copyright 2019 Google LLC
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// https://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
// A binary that uses comms and client, to receive FDs by name, communicate
|
|
// with them, sandboxed or not.
|
|
|
|
#include <cstdio>
|
|
#include <cstdlib>
|
|
#include <string>
|
|
|
|
#include "absl/strings/numbers.h"
|
|
#include "sandboxed_api/sandbox2/client.h"
|
|
#include "sandboxed_api/sandbox2/comms.h"
|
|
#include "sandboxed_api/util/raw_logging.h"
|
|
|
|
int main(int argc, char* argv[]) {
|
|
if (argc < 2) {
|
|
printf("argc < 2\n");
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
sandbox2::Comms default_comms(sandbox2::Comms::kDefaultConnection);
|
|
sandbox2::Client client(&default_comms);
|
|
|
|
int testno;
|
|
SAPI_RAW_CHECK(absl::SimpleAtoi(argv[1], &testno), "testno is not a number");
|
|
switch (testno) {
|
|
case 1:
|
|
break;
|
|
case 2:
|
|
client.SandboxMeHere();
|
|
break;
|
|
case 3:
|
|
// In case 3, we're running without a mapped fd. This is to test that the
|
|
// Client object parses the environment variable properly in that case.
|
|
return EXIT_SUCCESS;
|
|
default:
|
|
printf("Unknown test: %d\n", testno);
|
|
return EXIT_FAILURE;
|
|
}
|
|
int expected_fd = -1;
|
|
if (argc >= 3) {
|
|
SAPI_RAW_CHECK(absl::SimpleAtoi(argv[2], &expected_fd),
|
|
"expected_fd is not a number");
|
|
}
|
|
|
|
int fd = client.GetMappedFD("ipc_test");
|
|
if (expected_fd >= 0 && fd != expected_fd) {
|
|
fprintf(stderr, "error mapped fd not as expected, got: %d, want: %d", fd,
|
|
expected_fd);
|
|
return EXIT_FAILURE;
|
|
}
|
|
sandbox2::Comms comms(fd);
|
|
std::string resp;
|
|
if (!default_comms.SendString("start")) {
|
|
fputs("error on default_comms.RecvString(\"start\")", stderr);
|
|
return EXIT_FAILURE;
|
|
}
|
|
if (!default_comms.RecvString(&resp)) {
|
|
fputs("error on default_comms.RecvString(&resp)", stderr);
|
|
return EXIT_FAILURE;
|
|
}
|
|
if (resp != "started") {
|
|
fprintf(stderr, "unexpected response \"%s\" (expected \"started\")\n",
|
|
resp.c_str());
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
if (!comms.RecvString(&resp)) {
|
|
fputs("error on comms.RecvString(&resp)", stderr);
|
|
return EXIT_FAILURE;
|
|
}
|
|
if (resp != "hello") {
|
|
fprintf(stderr, "unexpected response \"%s\" (expected \"hello\")\n",
|
|
resp.c_str());
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
if (!comms.SendString("world")) {
|
|
fputs("error on comms.SendString(\"world\")", stderr);
|
|
return EXIT_FAILURE;
|
|
}
|
|
if (!default_comms.SendString("finish")) {
|
|
fputs("error on default_comms.RecvString(\"finish\")", stderr);
|
|
return EXIT_FAILURE;
|
|
}
|
|
if (!default_comms.RecvString(&resp)) {
|
|
fputs("error on default_comms.RecvString(&resp)", stderr);
|
|
return EXIT_FAILURE;
|
|
}
|
|
if (resp != "finished") {
|
|
fprintf(stderr, "unexpected response \"%s\" (expected \"finished\")\n",
|
|
resp.c_str());
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
printf("OK: All tests went OK\n");
|
|
return EXIT_SUCCESS;
|
|
}
|