#include <stdlib.h>

 #include <errno.h>

 #include <windows.h>

 #include <setupapi.h>

 #include <lua.h>

 #include <lualib.h>

 #include <lauxlib.h>

 #include "_whelk.h"

 #ifndef SUOI_FORCEDELETE

 #define SUOI_FORCEDELETE	1

 #endif

 int whelk_win32_error(lua_State *L,DWORD err)

 {

     DWORD n;

     WCHAR *buf;

     char *utf8;

     lua_pushnil(L);

     buf=NULL;

     n=FormatMessageW(FORMAT_MESSAGE_ALLOCATE_BUFFER|\

       FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS,0,err,0,

       (LPWSTR)&buf,0,NULL);

     if (n)

     {

 	if (n>2 && buf[n-2]=='\r' && buf[n-1]=='\n')

 	    n-=2;

 	utf8=whelk_utf16_to_utf8(buf,n);

 	LocalFree(buf);

 	lua_pushstring(L,utf8);

 	free(utf8);

     }

     else

 	lua_pushstring(L,"Unknown error");

     lua_pushinteger(L,err);

     return 3;

 }

 /*

  * SetupCopyOEMInf(source_inf_file_name,oem_source_media_location,

  * 	oem_source_media_type,copy_style)

  *

  * oem_source_media_type is one of:

  * 	SPOST_NONE

  * 		No source media information is stored.

  *

  * 	SPOST_PATH

  * 		oem_source_media_location contains a path to the source media

  * 		(eg., "A:\").

  *

  *	SPOST_URL

  *		oem_source_media_locationcontains a universal resource locator.

  *

  * copy_style is one of:

  * 	SP_COPY_DELETESOURCE

  * 		Delete source file on successful copy.

  * 	SP_COPY_REPLACEONLY

  * 		Copy only if this file already exists in the Inf directory.

  *	SP_COPY_NOOVERWRITE

  * 		Copy only if this file does not already exist in the Inf

  * 		directory.

  *	SP_COPY_OEMINF_CATALOG_ONLY

  *		The specified .inf file's corresponding catalog files is

  *		copied to %windir%\Inf.

  *

  * Returns two strings: the destination .inf file path (including file name)

  * and just the file name component.

  */

 int whelk_setup_copy_oem_inf(lua_State *L)

 {

     WCHAR path[MAX_PATH];

     BOOL result;

     size_t n;

     const char *s;

     char *utf8;

     WCHAR *source_inf_file_name=whelk_utf8_to_utf16(luaL_checkstring(L,1),-1);

     WCHAR *oem_source_media_location;

     int oem_source_media_type=luaL_checkint(L,3);

     int copy_style=luaL_checkint(L,4);

     WCHAR *dest_in_file_name_component;

     s=luaL_checklstring(L,2,&n);

     oem_source_media_location=s?whelk_utf8_to_utf16(s,n):NULL;

     path[0]='\0';

     dest_in_file_name_component=NULL;

     result=SetupCopyOEMInfW(source_inf_file_name,oem_source_media_location,

       oem_source_media_type,copy_style,path,sizeof(path)/sizeof(*path),NULL,

       &dest_in_file_name_component);

     free(source_inf_file_name);

     free(oem_source_media_location);

     if (!result)

 	return whelk_win32_error(L,GetLastError());

     utf8=whelk_utf16_to_utf8(path,-1);

     lua_pushstring(L,utf8);

     free(utf8);

     if (dest_in_file_name_component)

     {

 	utf8=whelk_utf16_to_utf8(dest_in_file_name_component,-1);

 	lua_pushstring(L,utf8);

 	free(utf8);

     }

     else

 	lua_pushnil(L);

     return 2;

 }

 /*

  * SetupUninstallOEMInf(inf_file_name[,flags])

  *

  * The following flags are recognised:

  *	SUOI_FORCEDELETE

  *		Delete the .inf file even if an installed device uses it.

  *

  * Returns non-nil on success.

  */

 int whelk_setup_uninstall_oem_inf(lua_State *L)

 {

     BOOL result;

     WCHAR *inf_file_name=whelk_utf8_to_utf16(luaL_checkstring(L,1),-1);

     int flags=lua_isnone(L,2)?0:luaL_checkint(L,2);

 #ifdef HAVE_SETUPUNINSTALLOEMINFW

     result=SetupUninstallOEMInfW(inf_file_name,flags,NULL);

 #else

     HMODULE lib;

     BOOL (WINAPI *SetupUninstallOEMInfW)(WCHAR *inf_file_name,DWORD flags,

       VOID *reserved);

     lib=LoadLibrary("setupapi.dll");

     if (lib)

     {

 	SetupUninstallOEMInfW=GetProcAddress(lib,"SetupUninstallOEMInfW");

 	if (SetupUninstallOEMInfW)

 	    result=SetupUninstallOEMInfW(inf_file_name,flags,NULL);

 	else

 	{

 	    FreeLibrary(lib);

 	    free(inf_file_name);

 	    return whelk_win32_error(L,ERROR_NOT_SUPPORTED);

 	}

 	FreeLibrary(lib);

     }

     else

     {

 	free(inf_file_name);

 	return whelk_win32_error(L,ERROR_NOT_SUPPORTED);

     }

 #endif

     free(inf_file_name);

     if (!result)

 	return whelk_win32_error(L,GetLastError());

     lua_pushinteger(L,1);		/* Something other than nil */

     return 1;

 }

 void whelk_open_setup(lua_State *L)

 {

     whelk_reg_const(L,SPOST_NONE);

     whelk_reg_const(L,SPOST_PATH);

     whelk_reg_const(L,SPOST_URL);

     whelk_reg_const(L,SP_COPY_DELETESOURCE);

     whelk_reg_const(L,SP_COPY_REPLACEONLY);

     whelk_reg_const(L,SP_COPY_NOOVERWRITE);

     whelk_reg_const(L,SP_COPY_OEMINF_CATALOG_ONLY);

     whelk_reg_const(L,SUOI_FORCEDELETE);

 }