现在的位置: 首页 > 综合 > 正文

《Windows API巡礼》—StringCbCopy和StringCchCopy

2012年11月26日 ⁄ 综合 ⁄ 共 2740字 ⁄ 字号 评论关闭

StringCbCopy函数用于拷贝string,且提供了目标缓冲区的大小作为参数,防止缓冲区溢出等安全问题。

这个函数可用于替换以下函数的使用:

strcpy, wcscpy, _tcscpy

lstrcpy

StrCpy

StringCbCopy函数原型如下:

HRESULT StringCbCopy(

  __out  LPTSTR pszDest, //目标字符串缓冲区

  __in   size_t cbDest, //目标缓冲区大小(字节),这个值必须考虑pszSrc加上空结束符’/0’的大小;

                            //最大运行的字节数是STRSAFE_MAX_CCH * sizeof(TCHAR)

  __in   LPCTSTR pszSrc //源字符串缓冲区,必须以’/0’结尾

);

函数返回值如下(强烈建议使用SUCCEEDEDFAILED宏来测试返回值):

S_OK                 //一切OK

STRSAFE_E_INVALID_PARAMETER    //目标缓冲区中值的大小要么是0,要么大于最大允许值

STRSAFE_E_INSUFFICIENT_BUFFER  //目标缓冲区大小不够,数据被截断;

//当允许数据截断时,这不算是错误

StringCbCopy函数根据传入的字符串参数类型而自动调用不同的函数:

//字符串类型                    字符串实例                       自动调用的函数

char                             "ASCE"                                 StringCbCopyA

TCHAR                               TEXT("ASCE")                            StringCbCopy

WCHAR                    L"ASCE"                     StringCbCopyW

 

扩展函数StringCbCopyEx原型如下:

HRESULT StringCbCopyEx(

  __out      LPTSTR pszDest, //目标字符串缓冲区

  __in       size_t cbDest, //pszDest的大小(字节

  __in       LPCTSTR pszSrc, //源字符串缓冲区

  __out_opt  LPTSTR *ppszDestEnd, //指向pszDest结尾的指针(指向空结束符处)

  __out_opt  size_t *pcbRemaining, //pszDest未使用的字节数,包括空结束符占用的字节

  __in       DWORD dwFlags //取值见下文

);

其中dwFlags可取值如下:

STRSAFE_FILL_BEHIND_NULL             //函数执行成功时,dwFlag(0)的低字节将被用于填充

//pszDest'/0'之后的未被填充的空间

STRSAFE_IGNORE_NULLS  //将空字符串指针(null)以空字符串TEXT("")对待,可用于模拟lstrcpy

STRSAFE_FILL_ON_FAILURE       //函数执行失败时,dwFlag(0)的低字节将被用于填充

//pszDest所有的空间,且pszDest以空结束符结束

STRSAFE_NULL_ON_FAILURE //函数执行失败时,pszDest被设置为空字符串TEXT("")

STRSAFE_NO_TRUNCATION     //不截断

 

函数返回值同StringCbCopy

 

还有两个相似的函数分别是StringCchCopyStringCchCopyEx,它们和上面两个函数唯一的区别StringCchCopy函数是以字符数计数的,而StringCbCopy函数是以字节数计数的。

ch表示characterb表示byte),这两个函数原型如下:

HRESULT StringCchCopy(

  __out  LPTSTR pszDest,

  __in   size_t cchDest, //pszDest的大小(字符数

  __in   LPCTSTR pszSrc

);

 

HRESULT StringCchCopyEx(

  __out      LPTSTR pszDest,

  __in       size_t cchDest, //pszDest的大小(字符数

  __in       LPCTSTR pszSrc,

  __out_opt  LPTSTR *ppszDestEnd,

  __out_opt  size_t *pcchRemaining,

  __in       DWORD dwFlags

);

 

用法举例:

获取当前目录,在没有使用strSafe系列函数之前,经典的用法是:

void UnsafeFunc(LPTSTR szPath,DWORD cchPath) {

         TCHAR szCWD[MAX_PATH];

         GetCurrentDirectory(ARRAYSIZE(szCWD), szCWD);

         strncpy(szPath, szCWD, cchPath);       //没有检查返回值

         strncat(szPath, TEXT("//"), cchPath);

         strncat(szPath, TEXT("asce.ini"),cchPath);

}

 

使用strSafe系列函数之后,提高了字符串处理的安全性:

bool SaferFunc(LPTSTR szPath,DWORD cchPath) {

         TCHAR szCWD[MAX_PATH];

         if (GetCurrentDirectory(ARRAYSIZE(szCWD), szCWD) &&

                            SUCCEEDED(StringCchCopy(szPath, cchPath, szCWD)) &&

                            SUCCEEDED(StringCchCat(szPath, cchPath, TEXT("//"))) &&

                            SUCCEEDED(StringCchCat(szPath, cchPath, TEXT("asce.ini")))) {

                            return true;

         }

         return false;

}

抱歉!评论已关闭.