广

MSSQL

  • MYSQL
  • MSSQL
  • Redis
  • MongoDB
  • oracle数据库
  • 数据管理

    编写安全的SQL Server扩展存储过程

    2018-04-05 20:15:00 次阅读 稿源:互联网
    零七网广告
    全网推广平台,软文发布

      SQL Server 的扩展存储过程,其实就是一个普通的 Windows DLL,只不过按照某种规则实现了某些函数而已。

      近日在写一个扩展存储过程时,发现再写这类动态库时,还是有一些需要特别注意的地方。之所以会特别注意,是因为DLL运行于SQL Server的地址空间,而SQL Server到底是怎么进行线程调度的,却不是我们能了解的,即便了解也无法控制。

      我们写动态库一般是自己用,即便给别人用,也很少像SQL Server这样,一个动态库很有可能加载多次,并且都是加载到一个进程的地址空间中。我们知道,当一个动态库加载到进程的地址空间时,DLL所有全局与局部变量初始化且仅初始化一次,以后再次调用 LoadLibrary函数时,仅仅增加其引用计数而已,那么很显然,假如有一全局 int ,初始化为0,调用一个函数另其自加,此时其值为1,然后再调用LoadLibray,并利用返回的句柄调用输出函数输出该值,虽然调用者觉得自己加载后立即输出,然后该值确实1而不是0。windows是进程独立的,而在线程方面,假如不注意,上面的情况很可能会程序员带来麻烦。

      介绍一下我的扩展存储过程,该动态库导出了三个函数: Init,work,Final,Init读文件,存储信息于内存,work简单的只是向该内存检索信息,Final回收内存。如上所说,假如不考虑同一进程空间多次加载问题,两次调用Init将造成无谓的浪费,因为我第一次已经读进了内存,要是通过堆分配内存,还会造成内存泄露。

      我使用的引用计数解决的该问题,代码很短,直接贴上来:

    #include "stdafx.h"
    #include <string>

    using namespace std;

    extern "C" {
    RETCODE __declspec(dllexport) xp_part_init(SRV_PROC *srvproc);
    RETCODE __declspec(dllexport) xp_part_process(SRV_PROC *srvproc);
    RETCODE __declspec(dllexport) xp_part_finalize(SRV_PROC *srvproc);
    }

    #define XP_NOERROR 0
    #define XP_ERROR 1

    HINSTANCE hInst = NULL;
    int nRef = 0;

    void printError (SRV_PROC *pSrvProc, CHAR* szErrorMsg);

    ULONG __GetXpVersion(){ return ODS_VERSION;}

    SRVRETCODE xp_part_init(SRV_PROC* pSrvProc){
    typedef bool (*Func)();

    if(nRef == 0){
    hInst = ::LoadLibrary("part.dll");
    if(hInst == NULL){
    printError(pSrvProc,"不能加载part.dll");
    return XP_ERROR;
    }
    Func theFunc = (Func)::GetProcAddress(hInst,"Init");
    if(!theFunc()){
    ::FreeLibrary(hInst);
    printError(pSrvProc,"不能获得分类号与专辑的对应表");
    return XP_ERROR;
    }
    }

    ++ nRef;
    return (XP_NOERROR);
    }

    SRVRETCODE xp_part_process(SRV_PROC* pSrvProc){
    typedef bool (*Func)(char*);

    if(nRef == 0){
    printError(pSrvProc,"函数尚未初始化,请首先调用xp_part_init");
    return XP_ERROR;
    }
    Func theFunc = (Func)::GetProcAddress(hInst,"Get");

    BYTE bType;
    ULONG cbMaxLen,cbActualLen;
    BOOL fNull;
    char szInput[256] = {0};

    if (srv_paraminfo(pSrvProc, 1, &bType, (ULONG*)&cbMaxLen, (ULONG*)&cbActualLen, (BYTE*)szInput, &fNull) == FAIL){
    printError(pSrvProc,"srv_paraminfo 返回 FAIL");
    return XP_ERROR;
    }
    szInput[cbActualLen] = 0;

    string strInput = szInput;
    string strOutput = ";";
    int cur,old = 0;
    while(string::npos != (cur = strInput.find(’;’,old)) ){
    strncpy(szInput,strInput.c_str() + old,cur - old);
    szInput[cur - old] = 0;
    old = cur + 1;
    theFunc(szInput);

    零七网部分新闻及文章转载自互联网,供读者交流和学习,若有涉及作者版权等问题请及时与我们联系,以便更正、删除或按规定办理。感谢所有提供资讯的网站,欢迎各类媒体与零七网进行文章共享合作。

    零七网广告
    零七网广告
    零七网广告
    零七网广告