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 393 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 ASCIICHAT_OK, 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().