util-src/windows.c

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