开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 游戏 > 模拟服务器 > 查看源码
KLuaScript.cpp
资源名称：SwordOnline.rar　[点击查看]
上传用户：dzyhzl
上传日期：2019-04-29
资源大小：56270k
文件大小：17k
源码类别：

模拟服务器

开发平台：

C/C++

KLuaScript.cpp：源码内容
  1. //---------------------------------------------------------------------------
  2. // Sword3 Engine (c) 1999-2000 by Kingsoft
  3. // 
  4. // File: KLuaScript.cpp
  5. // Date: 2001-9-13 10:33:29
  6. // Code: Romandou
  7. // Desc:
  8. //---------------------------------------------------------------------------
  9. #include "KWin32.h"
  10. #include "KDebug.h"
  11. #include "KPakFile.h"
  12. #include "KLuaScript.h"
  13. #include "LuaLib.h"
  14. #include "KMemClass.h"
  15. //---------------------------------------------------------------------------
  16. // 函数: KLuaScript::KLuaScript
  17. // 功能:
  18. // 参数: void
  19. // 返回:
  20. //---------------------------------------------------------------------------
  21. KLuaScript::KLuaScript(void)
  22. {
  23. m_LuaState = lua_open(100);
  25. if (m_LuaState == NULL)
  26. {
  27. ScriptError(LUA_CREATE_ERROR);
  28. m_IsRuning = FALSE;
  29. return ;
  30. }
  32. m_IsRuning = TRUE;
  33. m_szScriptName[0] = '';
  34. }
  36. //---------------------------------------------------------------------------
  37. // 函数: KLuaScript::KLuaScript
  38. // 功能:
  39. // 参数: int StackSize
  40. // 返回:
  41. //---------------------------------------------------------------------------
  42. KLuaScript::KLuaScript(int StackSize = 0)
  43. {
  44. m_LuaState = Lua_Create(StackSize);
  46. if (m_LuaState == NULL )
  47. {
  48. ScriptError(LUA_CREATE_ERROR);
  49. m_IsRuning = FALSE;
  50. return ;
  51. }
  52. m_IsRuning = TRUE;
  53. m_szScriptName[0] = '';
  54. }
  57. //---------------------------------------------------------------------------
  58. // 函数: KLuaScript::~KLuaScript
  59. // 功能:
  60. // 参数: void
  61. // 返回:
  62. //---------------------------------------------------------------------------
  63. KLuaScript::~KLuaScript(void)
  64. {
  65. Exit();
  66. }
  68. //---------------------------------------------------------------------------
  69. // 函数: KLuaScript::LoadBuffer()
  70. // 功能:
  71. // 参数: PBYTE pBuffer
  72. // 参数: DWORD dwLen
  73. // 返回: BOOL 
  74. //---------------------------------------------------------------------------
  75. BOOL KLuaScript::LoadBuffer(PBYTE pBuffer, DWORD dwLen )
  76. {
  77. if (dwLen < 0)
  78. {
  79. ScriptError(LUA_SCRIPT_LEN_ERROR);
  80. return FALSE;
  81. }
  83. if (Lua_CompileBuffer(m_LuaState, (char *) pBuffer, dwLen, NULL) != 0)
  84. {
  85. ScriptError(LUA_SCRIPT_COMPILE_ERROR);
  86. return FALSE;
  87. }
  88. return TRUE;
  89. }
  90. //---------------------------------------------------------------------------
  91. // 函数: KLuaScript::Load
  92. // 功能:
  93. // 参数: LPSTR Filename
  94. // 返回: BOOL 
  95. //---------------------------------------------------------------------------
  96. BOOL KLuaScript::Load(char * Filename)
  97. {
  98. KPakFile File;
  99. DWORD Size;
  102. // open file
  103. if (!File.Open(Filename)) return FALSE;
  105. // get file size
  106. Size = File.Size();
  108. KMemClass Memory;
  109. // alloc memory
  110. if (! Memory.Alloc(Size + 4))
  111. return FALSE;
  113. // read file
  114. if (File.Read(Memory.GetMemPtr(), Size) != Size)
  115. return FALSE;
  116. char * pszMem = (char *)Memory.GetMemPtr();
  117. pszMem[Size + 1] = 0;
  119. File.Close();
  120. try
  121. {
  122. if (!LoadBuffer((PBYTE)Memory.GetMemPtr(), Size ))
  123. {
  124. ScriptError(LUA_SCRIPT_COMPILE_ERROR);
  125. return FALSE;
  126. }
  127. }
  128. catch(...)
  129. {
  130. printf("Load Script %s 出现异常，请检查!!n", Filename);
  131. return FALSE;
  132. }
  134. if (!ExecuteCode()) return FALSE;
  136. return TRUE;
  137. }
  139. //---------------------------------------------------------------------------
  140. // 函数: KLuaScript::Execute
  141. // 功能:
  142. // 返回: BOOL 
  143. //---------------------------------------------------------------------------
  144. BOOL KLuaScript::Execute()
  145. {
  146. if (m_IsRuning && m_LuaState)
  147. return CallFunction(MAINFUNCTIONNAME,0,"");
  149. return FALSE;
  150. }
  153. //---------------------------------------------------------------------------
  154. // 函数: KLuaScript::ExecuteCode
  155. // 功能:
  156. // 返回: BOOL 
  157. //---------------------------------------------------------------------------
  158. BOOL KLuaScript::ExecuteCode()
  159. {
  160. if (!(m_IsRuning && m_LuaState))
  161. {
  162. ScriptError(LUA_SCRIPT_EXECUTE_ERROR);
  163. //if (!ExecuteCode()) return FALSE; ZHANGPENG 发现这里可能有错
  164. return FALSE;
  165. }
  167. int state;
  168. if (state = Lua_Execute(m_LuaState) != 0)
  169. {
  170. ScriptError(LUA_SCRIPT_EXECUTE_ERROR, state);
  171. return FALSE;
  172. }
  174. return TRUE;
  175. }
  178. //---------------------------------------------------------------------------
  179. // 函数: KLuaScript::CallFunction
  180. // 功能: 调用Lua脚本内的函数
  181. // 参数: LPSTR cFuncName
  182. // 参数: int nResults
  183. // 参数: LPSTR cFormat  调用时所传参数的类型 
  184. // n:数字型(double) d:整形(int) s:字符串型 f:C函数型  n:Nil v:Value p:Point
  185. //        v形为Lua支持的，参数为整形的数index，指明将index所指堆栈的变量作为
  186. //  该函数的调用参数。
  187. // 注意：由于该函数有不定参数…,对于数字，系统并不确定数是以double还是以int
  188. //  存在，两种保存形式是不同的。因此需要注意当传入的数是整形时，格式符应用d
  189. //  而不能用n,或者强行改变为double形。否则会出现计算的错误。
  190. //   
  191. // 参数: ...
  192. // 返回: BOOL 
  193. //---------------------------------------------------------------------------
  194. BOOL KLuaScript::CallFunction(LPSTR cFuncName, int nResults, LPSTR cFormat, va_list vlist)
  195. {
  197. double nNumber;
  198. char * cString = NULL;
  199. void * pPoint = NULL;
  200. Lua_CFunction CFunc;
  201. int i=0;
  202. int nArgnum = 0;
  203. int nIndex = 0;
  204. int nRetcode; //调用脚本函数后的返回码
  206. if (! (m_IsRuning && m_LuaState))
  207. {
  208. ScriptError(LUA_SCRIPT_STATES_IS_NULL);
  209. return FALSE;
  210. }
  212. {
  213. Lua_GetGlobal(m_LuaState, cFuncName); //在堆栈中加入需要调用的函数名
  215. while (cFormat[i] != '')
  216. {
  217. switch(cFormat[i])
  218. {
  219. case 'n'://输入的数据是double形 NUMBER，Lua来说是Double型
  220. { 
  221. nNumber = va_arg(vlist, double );
  222. Lua_PushNumber(m_LuaState, nNumber);
  223. nArgnum ++;
  225. }
  226. break;
  228. case 'd'://输入的数据为整形
  229. {
  230. nNumber = (double)(va_arg(vlist, int));
  231. Lua_PushNumber(m_LuaState, (double) nNumber);
  232. nArgnum ++;
  233. }
  234. break;
  236. case 's'://字符串型
  237. {
  238. cString = va_arg(vlist, char *);
  239. Lua_PushString(m_LuaState, cString);
  240. nArgnum ++;
  241. }
  242. break;
  243. case 'N'://NULL
  244. {
  245. Lua_PushNil(m_LuaState);
  246. nArgnum ++;
  247. }
  248. break;
  250. case 'f'://输入的是CFun形，即内部函数形
  251. {
  252. CFunc = va_arg(vlist, Lua_CFunction);
  253. Lua_PushCFunction(m_LuaState, CFunc) ;
  254. nArgnum ++;
  255. }
  256. break;
  258. case 'v'://输入的是堆栈中Index为nIndex的数据类型
  259. {
  260. nNumber = va_arg(vlist, int);
  261. int nIndex1 = (int) nNumber;
  262. Lua_PushValue(m_LuaState, nIndex1);
  263. nArgnum ++;
  264. }
  265. break;
  266. case 't'://输入为一Table类型
  267. {
  271. }
  272. break;
  274. case 'p':
  275. {
  276. pPoint = va_arg(vlist, void *);
  278. Lua_PushUserTag(m_LuaState, pPoint,m_UserTag);
  279. nArgnum ++;
  280. }
  281. break;
  282. }
  284. i++;
  285. }
  287. }  
  288.     
  289. nRetcode = Lua_Call(m_LuaState, nArgnum, nResults);
  291. if (nRetcode != 0)
  292. {
  293. ScriptError(LUA_SCRIPT_EXECUTE_ERROR, nRetcode);
  294. return FALSE;
  295. }
  298. return TRUE;
  299. }
  302. //---------------------------------------------------------------------------
  303. // 函数: KLuaScript::CallFunction
  304. // 功能:
  305. // 参数: LPSTR cFuncName
  306. // 参数: int nResults
  307. // 参数: LPSTR cFormat
  308. // 参数: ...
  309. // 返回: BOOL 
  310. //---------------------------------------------------------------------------
  311. BOOL KLuaScript::CallFunction(LPSTR cFuncName, int nResults, LPSTR cFormat, ...)
  312. {
  313. BOOL bResult  = FALSE;
  314. va_list vlist;
  315. va_start(vlist, cFormat);
  316. bResult = CallFunction(cFuncName, nResults, cFormat, vlist);
  317. va_end(vlist);
  318. return bResult;
  319. }
  322. //---------------------------------------------------------------------------
  323. // 函数: KLuaScript::GetValuesFromStack
  324. // 功能: 从堆栈中获得变量
  325. // 参数: char * cFormat
  326. // 参数: ...
  327. // 返回: BOOL 
  328. //---------------------------------------------------------------------------
  329. BOOL KLuaScript::GetValuesFromStack(char * cFormat, ...)
  330. {
  331. va_list vlist;
  332. double* pNumber = NULL;
  333. const char **   pString ;
  334. int * pInt = NULL;
  335. int i = 0;
  336. int nTopIndex = 0;
  337. int nIndex = 0;
  338. int nValueNum = 0;//cFormat的字符长度，表示需要取的参数数量
  340. if (! m_LuaState)
  341. return FALSE;
  343. nTopIndex = Lua_GetTopIndex(m_LuaState);
  344. nValueNum = strlen(cFormat);
  346. if (nTopIndex == 0 || nValueNum == 0)//当堆栈中无数据或不取参数是返回FALSE
  347. return FALSE;
  349. if (nTopIndex < nValueNum)
  350. return FALSE;
  352. nIndex = nTopIndex - nValueNum +1;
  354. {
  355. va_start(vlist, cFormat);     
  357. while (cFormat[i] != '')
  358. {
  360. switch(cFormat[i])
  361. {
  362. case 'n'://返回值为数值形,Number,此时Lua只传递double形的值
  363. {
  364. pNumber = va_arg(vlist, double *);
  366. if (pNumber == NULL)
  367. return FALSE;
  369. if (Lua_IsNumber(m_LuaState, nIndex ))
  370. {
  371. * pNumber = Lua_ValueToNumber(m_LuaState, nIndex ++ );
  373. }
  374. else
  375. {
  376. ScriptError(LUA_SCRIPT_NOT_NUMBER_ERROR);
  377. return FALSE;
  378. }
  381. }
  382. break;
  383. case 'd':
  384. {
  385. pInt = va_arg(vlist, int *);
  386. if (pInt == NULL)
  387. return FALSE;
  388. if ( Lua_IsNumber(m_LuaState, nIndex))
  389. {
  390. * pInt = (int ) Lua_ValueToNumber(m_LuaState, nIndex ++);
  391. }
  392. else
  393. {
  394. ScriptError(LUA_SCRIPT_NOT_NUMBER_ERROR);
  395. return FALSE;
  396. }
  398. }
  399. break;
  400. case 's'://字符串形
  401. {
  402. pString = va_arg(vlist, const char **);
  404. if (pString == NULL)
  405. return FALSE;
  407. if (Lua_IsString(m_LuaState, nIndex))
  408. {
  409. (*pString) = (const char *)Lua_ValueToString(m_LuaState, nIndex++);
  411. }
  412. else
  413. {
  414. ScriptError(LUA_SCRIPT_NOT_STRING_ERROR);
  415. return FALSE;
  416. }
  417. }
  418. break;
  420. }
  423. i ++;
  424. }
  425. va_end(vlist);      /* Reset variable arguments.      */
  427. }
  428. return TRUE;
  429. }
  431. //---------------------------------------------------------------------------
  432. // 函数: KLuaScript::Init
  433. // 功能: 初始化脚本对象，注册系统标准函数库
  434. // 返回: BOOL 
  435. //---------------------------------------------------------------------------
  436. BOOL KLuaScript::Init()
  437. {
  438. if (! m_LuaState)
  439. {
  440. m_LuaState = Lua_Create(0);
  442. if (m_LuaState == NULL)
  443. {
  444. ScriptError(LUA_CREATE_ERROR);
  445. m_IsRuning = FALSE;
  446. return FALSE;
  447. }
  449. m_IsRuning = TRUE;
  450. m_szScriptName[0] = '';
  451. m_UserTag = lua_newtag(m_LuaState) ;
  452. }
  454. RegisterStandardFunctions();
  455. return TRUE;
  456. }
  458. //---------------------------------------------------------------------------
  459. // 函数: KLuaScript::RegisterFunction
  460. // 功能: 注册某内部C函数至脚本中
  461. // 参数: LPSTR FuncName  在脚本中使用的函数名
  462. // 参数: void* Func    实际相应的C函数指针
  463. // 参数: int Args = 0 //与KScript接口相容，无用
  464. // 参数: int Flag = 0 //与KScript接口相容, 无用
  465. // 返回: BOOL 
  466. //---------------------------------------------------------------------------
  467. BOOL KLuaScript::RegisterFunction(LPSTR FuncName , void* Func)
  468. {
  469. if (! m_LuaState)
  470. return FALSE;
  471. Lua_Register(m_LuaState, FuncName, (Lua_CFunction)Func);
  472. return TRUE;
  473. }
  475. //---------------------------------------------------------------------------
  476. // 函数: KLuaScript::Compile
  477. // 功能:
  478. // 参数: char *
  479. // 返回: BOOL 
  480. //---------------------------------------------------------------------------
  481. BOOL KLuaScript::Compile(char *)
  482. {
  483. return TRUE;
  484. }
  486. //---------------------------------------------------------------------------
  487. // 函数: KLuaScript::RegisterFunctions
  488. // 功能: 批量注册Lua的内部C函数，各个函数的信息保存在TLua_Funcs的数据中
  489. // 参数: TLua_Funcs *Funcs 数组的指针
  490. // 参数: int n 函数数量。可以为零，由系统计算得到。
  491. // 返回: BOOL 
  492. //---------------------------------------------------------------------------
  493. BOOL KLuaScript::RegisterFunctions(TLua_Funcs Funcs[], int n)
  494. {
  495. if (! m_LuaState) return FALSE;
  496. if (n == 0) n = sizeof(Funcs) / sizeof(Funcs[0]);
  497. for (int i = 0; i < n; i ++) Lua_Register(m_LuaState, Funcs[i].name, Funcs[i].func);
  498. return TRUE;
  499. }
  501. //---------------------------------------------------------------------------
  502. // 函数: KLuaScript::RegisterStandardFunctions
  503. // 功能: 注册Lua系统标准的函数库
  504. // 返回: void 
  505. //---------------------------------------------------------------------------
  506. void KLuaScript::RegisterStandardFunctions()
  507. {
  508. if (! m_LuaState) return ;
  509. Lua_OpenBaseLib(m_LuaState);//Lua基本库
  510. Lua_OpenIOLib(m_LuaState);//输入输出库
  511. Lua_OpenStrLib(m_LuaState);//字符串处理库
  512. Lua_OpenMathLib(m_LuaState);//数值运算库
  513. //Lua_OpenDBLib(m_LuaState);//调试库
  514. return;
  515. }
  518. //---------------------------------------------------------------------------
  519. // 函数: KLuaScript::ReleaseScript
  520. // 功能: 释放该脚本资源。
  521. // 返回: BOOL 
  522. //---------------------------------------------------------------------------
  523. void KLuaScript::Exit()
  524. {
  526. if (! m_LuaState) return ;
  527. Lua_Release(m_LuaState);
  528. m_LuaState = NULL;
  529. m_IsRuning = FALSE;
  531. }
  533. //---------------------------------------------------------------------------
  534. // 函数: KLuaScript::ScriptError
  535. // 功能:
  536. // 参数: int Error
  537. // 返回: void 
  538. //---------------------------------------------------------------------------
  539. void KLuaScript::ScriptError(int Error)
  540. {
  541. char lszErrMsg[200];
  542. sprintf(lszErrMsg, "ScriptError %d. (%s) n", Error, m_szScriptName);
  543. lua_outerrmsg(lszErrMsg);
  544. return;
  545. }
  547. //---------------------------------------------------------------------------
  548. // 函数: KLuaScript::ScriptError
  549. // 功能:
  550. // 参数: int Error1
  551. // 参数: int Error2
  552. // 返回: void 
  553. //---------------------------------------------------------------------------
  554. void KLuaScript::ScriptError(int Error1 ,int Error2)
  555. {
  556. char lszErrMsg[200];
  557. sprintf(lszErrMsg, "ScriptError %d:[%d] (%s) n", Error1, Error2, m_szScriptName);
  558. lua_outerrmsg(lszErrMsg);
  559. return;
  560. }
  562. //---------------------------------------------------------------------------
  563. // 函数: KLuaScript::SafeCallBegin
  564. // 功能:
  565. // 参数: int * pIndex
  566. // 返回: void 
  567. //---------------------------------------------------------------------------
  568. //---------------------------------------------------------------------------
  569. // SafeCallBegin与SafeCallEnd两函数应搭配使用，以防止在调用Lua的外部函数之后，
  570. //有多余数据在堆栈中未被清除。达到调用前与调用后堆栈的占用大小不变。
  571. //上述情况只需用在调用外部函数时，内部函数不需如此处理。
  572. // Romandou
  573. //---------------------------------------------------------------------------
  574. void KLuaScript::SafeCallBegin(int * pIndex)
  575. {
  576. if (! m_LuaState) return ;
  577. Lua_SafeBegin(m_LuaState, pIndex);
  578. }
  580. //---------------------------------------------------------------------------
  581. // 函数: KLuaScript::SafeCallEnd
  582. // 功能:
  583. // 参数: int nIndex
  584. // 返回: void 
  585. //---------------------------------------------------------------------------
  586. void KLuaScript::SafeCallEnd(int nIndex)
  587. {
  588. if (! m_LuaState) return;
  589. Lua_SafeEnd(m_LuaState, nIndex);
  590. }
  592. //---------------------------------------------------------------------------
  593. // 函数: KLuaScript::StopScript
  594. // 功能: 中止脚本
  595. // 参数: void
  596. // 返回: BOOL 
  597. //---------------------------------------------------------------------------
  598. BOOL KLuaScript::Stop(void)
  599. {
  600. if (! m_IsRuning) return TRUE;
  601. if (! m_LuaState) return FALSE;
  602. m_IsRuning =  FALSE;
  603. return TRUE;
  604. }
  606. //---------------------------------------------------------------------------
  607. // 函数: KLuaScript::ResumeScript
  608. // 功能: 恢复已中止的脚本
  609. // 参数: void
  610. // 返回: BOOL 
  611. //---------------------------------------------------------------------------
  612. BOOL KLuaScript::Resume(void)
  613. {
  614. if ((! m_IsRuning) && (m_LuaState))
  615. {
  616. m_IsRuning = TRUE;
  617. return TRUE;
  618. }
  619. return FALSE;
  620. }
  623. //---------------------------------------------------------------------------
  624. // 函数: KLuaScript::CreateTable
  625. // 功能: 建立一个Lua的Table，在调用该函数并设置Table各个成员之后，必须调用
  626. // SetGlobalName()来给这个Table指定一个名字。
  627. // 返回: DWORD 
  628. //---------------------------------------------------------------------------
  629. DWORD KLuaScript::CreateTable()
  630. {
  631.  int nIndex = 0;
  633. nIndex = Lua_GetTopIndex(m_LuaState) ;
  634. Lua_NewTable(m_LuaState);
  635. if (Lua_GetTopIndex(m_LuaState) != ++nIndex ) 
  636. return -1;
  638. return nIndex;
  639. }
  641. //---------------------------------------------------------------------------
  642. // 函数: KLuaScript::SetGlobalName
  643. // 功能: 设置Lua堆栈顶部的数据一个名字
  644. // 参数: LPSTR szName
  645. // 返回: void 
  646. //---------------------------------------------------------------------------
  647. void KLuaScript::SetGlobalName(LPSTR szName)
  648. {
  649. if (!szName) return ;
  650. Lua_SetGlobal(m_LuaState, szName);
  651. }
  653. //---------------------------------------------------------------------------
  654. // 函数: KLuaScript::ModifyTable
  655. // 功能: 将指定名称的LuaTable置堆栈顶端，并返回顶端Index
  656. // 参数: LPSTR szTableName
  657. // 返回: DWORD 若Lua中不存在该Table则返回-1
  658. //---------------------------------------------------------------------------
  659. DWORD KLuaScript::ModifyTable(LPSTR szTableName) 
  660. {
  661. if (! szTableName[0]) return -1;
  663. int nIndex = Lua_GetTopIndex(m_LuaState);
  665. Lua_GetGlobal(m_LuaState, szTableName);
  667. if (Lua_GetTopIndex(m_LuaState) != ++nIndex) return -1;
  669. return nIndex;
  670. }