LUA调用 Dll 添加收藏夹操作！！！IE Chrome

// AddFavorate.cpp : Defines the exported functions for the DLL application.
//

#include "stdafx.h"
#include "atlbase.h"
#include <iostream>
#include <fstream>
#include <string>
#include "stdlib.h"
#include "stdio.h"
#include "io.h"
extern "C"
{    
#include "lua.h"  
#include "lualib.h"
#include "lauxlib.h"
}
#if defined(_WIN32)
    extern "C" __declspec(dllexport) int luaopen_MyLuaDLL(lua_State* L);
#else
    extern "C" int luaopen_MyLuaDLL(lua_State* L);
#endif
#pragma comment(lib,"lualib.lib")
lua_State * L;
using namespace std;
char* UnicodeToAnsi(TCHAR* wstr)     //转换字符编码
{
    if( !wstr )
        return NULL;

    int strlen = ::WideCharToMultiByte(CP_ACP, NULL, wstr, wcslen(wstr), NULL, 0, NULL, FALSE);
    char* str = new char[strlen + 1];
    ::WideCharToMultiByte(CP_ACP, NULL, wstr, wcslen(wstr), str, strlen, NULL, FALSE);
    str[strlen] = ‘\0‘;

    return str;
}
char* read_reg_sz()//读取操作表,其类型为REG_SZ
{
    HKEY hkey;
    LPCTSTR data_set = _T("\Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders");  //\Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders" /v "Favorites"‘

        if (ERROR_SUCCESS == ::RegOpenKeyEx(HKEY_CURRENT_USER, data_set, 0, KEY_READ, &hkey))
        {
            TCHAR dwValue[256];
            DWORD dwSzType = REG_SZ;
            DWORD dwSize = sizeof(dwValue);
            _tsetlocale(LC_ALL, _T("chs"));
            if (::RegQueryValueEx(hkey, _T("Favorites"), 0, &dwSzType, (LPBYTE)&dwValue, &dwSize) != ERROR_SUCCESS)
            {
                cout << "无法查询有关的注册表信息" << endl;
            }
            char* reValue;
            reValue = UnicodeToAnsi(dwValue);
            ::RegCloseKey(hkey);
            return reValue;
        }
        ::RegCloseKey(hkey);
}



int WriteFile(char* ljx,char *url,char* icon,char* name)  //写文件操作;
{
    FILE *stream;
    char lj[128]={};
    char l1[256] = "[{000214A0-0000-0000-C000-000000000046}]\r\nProp3=19,11\r\n[InternetShortcut]\r\nIDList=\r\nURL=";
    char l3[256] = "\r\nIconFile=";
    char l5[] = "\r\nIconIndex=1\r\n";
    strcat(lj,ljx);
    strcat(lj,"\\");

    strcat(lj,name);
    if ((stream = fopen(lj, "wb")) == NULL) /* open file TEST.$$$ */
    {
        fprintf(stderr, "Cannot open output file.\n");
        return -1;
    }
    strcat(l1,url);
    strcat(l3,icon);
    fwrite(l1, sizeof(l1), 1, stream);
    fwrite(l3, sizeof(l3), 1, stream);
    fwrite(l5, sizeof(l5), 1, stream);
    fclose(stream); /*关闭文件*/
    return 0;
}
LUALIB_API int AddFavorateIE(lua_State* L)   //添加 IE 主函数;
{    
    string url;
    string icon;
    string name;
    int res = 0;
//拾取栈
if( lua_isstring(L,-1)){           
    name = lua_tostring(L,-1);
}  
if( lua_isstring(L,-2)){           
    icon = lua_tostring(L,-2);
}  
if( lua_isstring(L,-3)){           
    url = lua_tostring(L,-3);
}  
    char* lj = read_reg_sz();
    if (strlen((char*)url.c_str()) && strlen((char*)name.c_str()) && strlen((char*)icon.c_str()))
    {
        res = WriteFile(lj,(char*)url.c_str(),(char*)icon.c_str(),(char*)name.c_str());
    }
    else
    {
        res = -1;
        lua_pushnumber(L,res);
        return 1;
    }
    lua_pushnumber(L,res);
    return 1;
}


int AddPage(char* url,char* name) // 写书签
{
    //Win7 及以上目录;
    char Lj[128] = "C:";
    strcat(Lj,getenv("homepath"));
    char i[]= "\\AppData\\Local\\Google\\Chrome\\User Data\\Default\\Bookmarks";
    char j[]= "\\AppData\\Local\\Google\\Chrome\\User Data\\Default\\Bookmarks1";
    char Lj1[128]={0};
    strcpy(Lj1,Lj);
    strcat(Lj,i);
    strcat(Lj1,j);

    //  XP 兼容目录;
    FILE *fp;
    if((fp=fopen(Lj,"r"))==NULL)
    {
        char Ljx[128] = "C:";
        strcat(Ljx,getenv("homepath"));
        char ix[]= "\\Local Settings\\Application Data\\Google\\Chrome\\User Data\\Default\\Bookmarks";
        char jx[]= "\\Local Settings\\Application Data\\Google\\Chrome\\User Data\\Default\\Bookmarks1";
        char Lj1x[128]={0};
        strcpy(Lj1x,Ljx);
        strcat(Ljx,ix);
        strcat(Lj1x,jx);
        if((fp=fopen(Ljx,"r"))==NULL)
        {
            return -1;
        }
        strcpy(Ljx,Lj);
        strcpy(Lj1x,Lj1);
    }
    fclose(fp);
    ///////////////////////////////////////////////////////////


    ifstream in(Lj);
    ofstream out(Lj1);
    string filename;
    string line;
    string str="children";
    string str1 = "],";
    bool tof = true;

    //创建拷贝文件;
    while (getline (in, line))
    {
        if (tof && strstr(line.c_str(),str.c_str()))
        {
            if (strstr(line.c_str(),str1.c_str()))
            {
                out << "         \"children\": [ " << endl;
                out<<"            \"date_added\": \"13087947012000000\","<<endl;
                out << "            \"id\": \"2\"," << endl;    
                out << "            \"name\": \""<<name<<"\"," << endl;    
                out << "            \"type\": \"url\"," << endl;    
                out << "            \"url\": \""<<url<<"\"" << endl;    
                out << "         } ]," << endl;    
            }
            else
            {
                out << line << endl;
                out<<"            \"date_added\": \"13087947012000000\","<<endl;
                out << "            \"id\": \"2\"," << endl;    
                out << "            \"name\": \""<<name<<"\"," << endl;    
                out << "            \"type\": \"url\"," << endl;    
                out << "            \"url\": \""<<url<<"\"" << endl;
                out << "         }, {" << endl;
            }
            tof = false;
        }
        else
        {
            out << line << endl;        
        }
    }
    in.close();
    out.close();
    //删除重命名;
    remove(Lj);
    rename(Lj1,Lj);
    return 0 ;
}
LUALIB_API int AddFavorateCh(lua_State* L)
{    
    string url;
    string icon;
    string name;
    int res = 0;
    //拾取栈
    if( lua_isstring(L,-1)){           
        name = lua_tostring(L,-1);
    }  
    if( lua_isstring(L,-2)){           
        url = lua_tostring(L,-2);
    }  
    char* lj = read_reg_sz();
    if (strlen((char*)url.c_str()) && strlen((char*)name.c_str()))
    {
        res = AddPage((char*)url.c_str(),(char*)name.c_str());
    }
    else
    {    
        res = -1;
        lua_pushnumber(L,res);
        return 1;
    }
    lua_pushnumber(L,res);
    return 1;
}
static const struct luaL_reg MyLuaDLLFunctions[]=
{
    {"AddFavorateIE",AddFavorateIE},
    {"AddFavorateCh",AddFavorateCh},
    {NULL, NULL}
};
int __declspec(dllexport) luaopen_MyLuaDLL(lua_State* L)
{
    luaL_openlib(L, "MyLuaDLLFunctions", MyLuaDLLFunctions, 0);
    return 0;
}

function main()                        --入口
local testlib = package.loadlib("C:\\Users\\丁宁\\Documents\\Visual Studio 2010\\Projects\\AddFavorate\\Debug\\AddFavorate.dll", "luaopen_MyLuaDLL")
print(testlib)--, "Can not open testlib.")
testlib()
--A = MyLuaDLLFunctions.AddFavorateIE("http://www.baidu.com/","https://www.baidu.com/favicon.ico/","baxxxxxxxxxxxxxxxxxxxxxxxxxxxxidu.url")
A = MyLuaDLLFunctions.AddFavorateCh("http://www.baidu.com/","baidu")
print(A)
end

void tiaoshi()
{
    cout<<"调取LUA中函数"<<endl;//LUA入口
    lua_getglobal(L,"main");
    if(!lua_isfunction(L,-1))
    {
        const char* error = lua_tostring(L,-1);
        cout<<string(error)<<endl;
    }
    if(lua_pcall(L,0,0,0))
    {
        const char* error = lua_tostring(L,-1);
        return;
    }
}

int main ( int argc, char *argv[] )
{    
    L = luaL_newstate();        //加载lua标准库;
    luaL_openlibs(L);    //加载脚本    ;
    char* str = "E:\\tiaoshi.lua";
    if (luaL_loadfile(L,str)||lua_pcall(L,0,0,0))
    {
        const char* error = lua_tostring(L,-1);
        cout<< string(error)<<endl;
        return 0;
    }
    tiaoshi();
    lua_close(L);    //系统命令，暂停;    
    system("pause");
    return 0;
}