main.cpp File Reference

ascii-query-server entry point More...

Go to the source code of this file.

Functions
int	main (int argc, char *argv[])

Detailed Description

ascii-query-server entry point

This is the controller process for the query tool. It attaches to a target process via LLDB and serves HTTP requests for variable inspection.

Usage: ascii-query-server –attach <pid> –port 9999 ascii-query-server –attach-name ascii-chat –port 9999

See also

docs/tooling/QUERY_TOOL_PLAN.md

Definition in file main.cpp.

Function Documentation

main()

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 392 of file main.cpp.

                                 {
  // Parse command line arguments
  pid_t attach_pid = 0;
  std::string attach_name;
  bool wait_for = false;
  uint16_t port = 9999;
 
  for (int i = 1; i < argc; i++) {
    if (strcmp(argv[i], "--help") == 0 || strcmp(argv[i], "-h") == 0) {
      printUsage(argv[0]);
      return 0;
    } else if (strcmp(argv[i], "--attach") == 0 && i + 1 < argc) {
      // Parse PID using safe parsing utility
      unsigned long pid_val = 0;
      if (parse_ulong(argv[++i], &pid_val, 1, ULONG_MAX) != ASCIICHAT_OK) {
        fprintf(stderr, "Error: Invalid PID value\n");
        return 1;
      }
      attach_pid = static_cast<pid_t>(pid_val);
    } else if (strcmp(argv[i], "--attach-name") == 0 && i + 1 < argc) {
      attach_name = argv[++i];
    } else if (strcmp(argv[i], "--wait") == 0) {
      wait_for = true;
    } else if (strcmp(argv[i], "--port") == 0 && i + 1 < argc) {
      // Parse port using safe parsing utility with proper range validation
      if (parse_port(argv[++i], &port) != ASCIICHAT_OK) {
        fprintf(stderr, "Error: Invalid port number (must be 1-65535)\n");
        return 1;
      }
    } else {
      fprintf(stderr, "Unknown option: %s\n", argv[i]);
      printUsage(argv[0]);
      return 1;
    }
  }
 
  // Validate arguments
  if (attach_pid == 0 && attach_name.empty()) {
    fprintf(stderr, "Error: Must specify --attach <pid> or --attach-name <name>\n\n");
    printUsage(argv[0]);
    return 1;
  }
 
  // Setup signal handlers
  signal(SIGINT, signalHandler);
  signal(SIGTERM, signalHandler);
#ifndef _WIN32
  signal(SIGPIPE, SIG_IGN);
#endif
 
  // Initialize LLDB controller
  ascii_query::LLDBController controller;
  g_controller = &controller;
 
  if (!controller.initialize()) {
    fprintf(stderr, "Error: Failed to initialize LLDB: %s\n", controller.lastError().c_str());
    return 1;
  }
 
  // Attach to target
  fprintf(stderr, "[ascii-query-server] Attaching to ");
  if (attach_pid > 0) {
    fprintf(stderr, "PID %llu...\n", static_cast<unsigned long long>(attach_pid));
    if (!controller.attach(attach_pid)) {
      fprintf(stderr, "Error: Failed to attach: %s\n", controller.lastError().c_str());
      return 1;
    }
  } else {
    fprintf(stderr, "process '%s'%s...\n", attach_name.c_str(), wait_for ? " (waiting)" : "");
    if (!controller.attachByName(attach_name, wait_for)) {
      fprintf(stderr, "Error: Failed to attach: %s\n", controller.lastError().c_str());
      return 1;
    }
  }
 
  fprintf(stderr, "[ascii-query-server] Attached to %s (PID %llu)\n", controller.targetName().c_str(),
          static_cast<unsigned long long>(controller.targetPid()));
 
  // Resume the target (it's stopped after attach)
  if (controller.state() == ascii_query::ProcessState::Stopped) {
    fprintf(stderr, "[ascii-query-server] Resuming target...\n");
    controller.resume();
  }
 
  // Setup HTTP server
  ascii_query::HttpServer server;
  g_server = &server;
 
  setupRoutes(server, controller);
 
  // Start HTTP server
  if (!server.start(port)) {
    fprintf(stderr, "Error: Failed to start HTTP server: %s\n", server.lastError().c_str());
    controller.detach();
    return 1;
  }
 
  fprintf(stderr, "[ascii-query-server] HTTP server listening on http://localhost:%d\n", port);
  fprintf(stderr, "[ascii-query-server] Press Ctrl+C to stop\n");
 
  // Wait for shutdown
  while (!g_shutdown_requested && controller.isAttached()) {
#ifdef _WIN32
    Sleep(100);
#else
    usleep(100000);
#endif
    ascii_query::ProcessState state = controller.state();
    if (state == ascii_query::ProcessState::Exited || state == ascii_query::ProcessState::Crashed) {
      fprintf(stderr, "[ascii-query-server] Target %s, shutting down\n",
              state == ascii_query::ProcessState::Exited ? "exited" : "crashed");
      break;
    }
  }
 
  // Cleanup
  fprintf(stderr, "[ascii-query-server] Shutting down...\n");
  server.stop();
  controller.detach();
  controller.shutdown();
 
  fprintf(stderr, "[ascii-query-server] Done\n");
  return 0;
}

References ascii_query::LLDBController::attach(), ascii_query::LLDBController::attachByName(), ascii_query::Crashed, ascii_query::LLDBController::detach(), ascii_query::Exited, ascii_query::LLDBController::initialize(), ascii_query::LLDBController::isAttached(), ascii_query::HttpServer::lastError(), ascii_query::LLDBController::lastError(), parse_port(), parse_ulong(), ascii_query::LLDBController::resume(), ascii_query::LLDBController::shutdown(), ascii_query::HttpServer::start(), ascii_query::LLDBController::state(), ascii_query::HttpServer::stop(), ascii_query::Stopped, ascii_query::LLDBController::targetName(), and ascii_query::LLDBController::targetPid().