// 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;
}