Anton (Все сообщения пользователя)

#define STRICT
#define WIN32_LEAN_AND_MEAN
#define VC_EXTRALEAN
#define NOMINMAX
#define WINVER 0x0600
#define _WIN32_WINNT WINVER
#include <Windows.h>
#include <process.h>

extern "C"
{
#include "luaconf.h"
#include "lua.h"
#include "lauxlib.h"
}

#pragma comment(lib, "lua5.1.lib")

// здесь будет храниться ссылка на стейт нового потока.
// глобальная переменная - плохое решение, но для демки сойдет,
// более интересные варианты только добавят ненужной здесь сложности
static int tref = LUA_NOREF;

// это "полезная" функция, выполняемая в новом потоке
static void some_useful_function(lua_State * s)
{
   // показываем сообщение о запуске
   lua_getglobal(s, "message");
   lua_pushliteral(s, "Эта функция выполняется в новом потоке. Выполняется, выполняется...");
   lua_call(s, 1, 0);
   // три секунды имитируем бурную деятельность
   ::Sleep(3000);
   // показываем сообщение об остановке
   lua_getglobal(s, "message");
   lua_pushliteral(s, "    ...кончила выполняться.");
   lua_call(s, 1, 0);
}

// это стартовая функция для нашего нового потока
static unsigned int __stdcall thrd(void * pvoid)
{
   // сохраняем переданный стейт.
   // обратите внимание, что в этом потоке мы работаем только со стейтом,
   // созданным специально для этого потока. мы никогда не трогаем стейты
   // других потоков и другие потоки не должны трогать наш стейт
   lua_State * n = static_cast<lua_State *>(pvoid);
   // вызываем "полезную" функцию, передавая ей свой стейт
   ::some_useful_function(n);
   // освобождаем ссылку на свой стейт, чтобы коллектор мог его прибить. если
   // этого не сделать, стейт доживет до закрытия квика даже после остановки
   // запустившего нас [в огород] скрипта (т.е. будет утечка памяти)
   luaL_unref(n, LUA_REGISTRYINDEX, ::tref);
   ::tref = LUA_NOREF;
   // с этого момента коллектор имеет право прибить наш стейт когда хочет, т.е.
   // для нас его уже не существует, хотя, возможно, где-то в дебрях квика он
   // еще будет болтаться некоторое время
   n = nullptr;
   // сейчас мы вернем управление и поток будет прибит. это приведет к тому, что
   // хэндл потока, на котором ждет наш main, перейдет в состояние signaled,
   // функция WaitForSingleObject вернет управление и main будет разблокирован
   return 0;
}

// это main для квика (внезапно сишный)
static int qmain(lua_State * s)
{
   // создаем новый стейт для потока.
   // стейт также появится на вершине стека s
   lua_State * n = lua_newthread(s);
   // создаем ссылку на стейт, чтобы коллектор случайно не прибил его раньше времени.
   // функция также выдернет стейт n с вершины стека s, вернув стек s в состояние,
   // в котором он был до вызова lua_newthread
   ::tref = luaL_ref(s, LUA_REGISTRYINDEX);
   // создаем нативный поток и передаем ему созданный стейт n
   unsigned int tid;
   HANDLE h = reinterpret_cast<HANDLE>(::_beginthreadex(NULL, 0, ::thrd, n, CREATE_SUSPENDED, &tid));
   if(NULL == h)
   {
      // в случае ошибки поток не создан, поэтому не забываем прибить ссылку на новый стейт,
      // иначе он останется в памяти до завершения квика (утечка памяти)
      luaL_unref(s, LUA_REGISTRYINDEX, ::tref);
      ::tref = LUA_NOREF;
      luaL_error(s, "_beginthreadex failed");
   }
   // обратите внимание, что в текущем потоке трогать стейт n больше нельзя, у него теперь новый хозяин,
   // который позаботится также и о его освобождении. поэтому мы сразу же обнуляем указатель на стейт,
   // чтобы случайные попытки туда влезть вызывали крэш
   n = nullptr;
   // запускаем новый поток на выполнение
   ::ResumeThread(h);
   // здесь main может делать что-то другое, второй поток работает параллельно. но
   // перед тем, как из мейна выйти, нужно убедиться, что все дополнительные потоки уже завершены.
   // по-хорошему им сначала надо как-то посигналить "все бросай и на выход бегом", а уже потом ждать, но
   // мы не будем лезть в дебри в рамках простой демки. поскольку нам в main особо делать нечего, мы сразу
   // переходим к ожиданию завершения второго потока. попробуйте закомментировать следующую строку, если
   // хотите посмотреть, как через три секунды к квику придет внезапный полярный зверек
   ::WaitForSingleObject(h, INFINITE);
   // не забываем закрыть хэндл потока, иначе он доживет до закрытия квика (утечка хэндла)
   ::CloseHandle(h);
   return 0;
}

extern "C" __declspec(dllexport) int qtest(lua_State * s)
{
   // мы хотим, чтобы в качестве колбека main работала наша сишная qmain
   lua_pushcclosure(s, qmain, 0);
   lua_setglobal(s, "main");
   return 0;
}

@echo off
setlocal enabledelayedexpansion
for /f "tokens=1-4" %%1 in ('dumpbin /exports %1') do (
    set /a ordinal=%%1 2>nul
    set /a hint=0x%%2 2>nul
    set /a rva=0x%%3 2>nul
    if !ordinal! equ %%1 if !hint! equ 0x%%2 if !rva! equ 0x%%3 set exports=!exports! /export:%%4
)
for /f %%i in ("%1") do set dllpath=%%~dpni
start lib /out:%dllpath%.lib /machine:x64 /def: %exports%

message("Before body")

t = newproxy(true)
mt = getmetatable(t)
mt["__gc"] = function(self) message("Garbage collector") end

function OnInit()
   message("OnInit")
end

function main()
   message("main")
   error "Error in main()"
end