util-src/windows.c

   1 /* Prosody IM\r
   2 -- Copyright (C) 2008-2010 Matthew Wild\r
   3 -- Copyright (C) 2008-2010 Waqas Hussain\r
   4 -- \r
   5 -- This project is MIT/X11 licensed. Please see the\r
   6 -- COPYING file in the source package for more information.\r
   7 --\r
   8 */\r
   9 \r
  10 /*\r
  11 * windows.c\r
  12 * Windows support functions for Lua\r
  13 */\r
  14 \r
  15 #include <stdio.h>\r
  16 #include <windows.h>\r
  17 #include <windns.h>\r
  18 \r
  19 #include "lua.h"\r
  20 #include "lauxlib.h"\r
  21 \r
  22 static int Lget_nameservers(lua_State *L) {\r
  23         char stack_buffer[1024]; // stack allocated buffer\r
  24         IP4_ARRAY* ips = (IP4_ARRAY*) stack_buffer;\r
  25         DWORD len = sizeof(stack_buffer);\r
  26         DNS_STATUS status;\r
  27 \r
  28         status = DnsQueryConfig(DnsConfigDnsServerList, FALSE, NULL, NULL, ips, &len);\r
  29         if (status == 0) {\r
  30                 DWORD i;\r
  31                 lua_createtable(L, ips->AddrCount, 0);\r
  32                 for (i = 0; i < ips->AddrCount; i++) {\r
  33                         DWORD ip = ips->AddrArray[i];\r
  34                         char ip_str[16] = "";\r
  35                         sprintf_s(ip_str, sizeof(ip_str), "%d.%d.%d.%d", (ip >> 0) & 255, (ip >> 8) & 255, (ip >> 16) & 255, (ip >> 24) & 255);\r
  36                         lua_pushstring(L, ip_str);\r
  37                         lua_rawseti(L, -2, i+1);\r
  38                 }\r
  39                 return 1;\r
  40         } else {\r
  41                 luaL_error(L, "DnsQueryConfig returned %d", status);\r
  42                 return 0; // unreachable, but prevents a compiler warning\r
  43         }\r
  44 }\r
  45 \r
  46 static const luaL_Reg Reg[] =\r
  47 {\r
  48         { "get_nameservers",    Lget_nameservers        },\r
  49         { NULL,         NULL    }\r
  50 };\r
  51 \r
  52 LUALIB_API int luaopen_util_windows(lua_State *L) {\r
  53         luaL_register(L, "windows", Reg);\r
  54         lua_pushliteral(L, "version");                  /** version */\r
  55         lua_pushliteral(L, "-3.14");\r
  56         lua_settable(L,-3);\r
  57         return 1;\r
  58 }\r