You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
112 lines
4.0 KiB
112 lines
4.0 KiB
/**
|
|
* Copyright 2020 Huawei Technologies Co., Ltd
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#define SECUREC_INLINE_DO_MEMCPY 1
|
|
|
|
#include "securecutil.h"
|
|
|
|
static errno_t SecDoWcsncpy(wchar_t *strDest, size_t destMax, const wchar_t *strSrc, size_t count)
|
|
{
|
|
size_t srcStrLen;
|
|
if (count < destMax) {
|
|
SECUREC_CALC_WSTR_LEN(strSrc, count, &srcStrLen);
|
|
} else {
|
|
SECUREC_CALC_WSTR_LEN(strSrc, destMax, &srcStrLen);
|
|
}
|
|
if (srcStrLen == destMax) {
|
|
strDest[0] = '\0';
|
|
SECUREC_ERROR_INVALID_RANGE("wcsncpy_s");
|
|
return ERANGE_AND_RESET;
|
|
}
|
|
if (strDest == strSrc) {
|
|
return EOK;
|
|
}
|
|
if (SECUREC_STRING_NO_OVERLAP(strDest, strSrc, srcStrLen)) {
|
|
/* performance optimization srcStrLen not include '\0' */
|
|
SecDoMemcpy(strDest, strSrc, srcStrLen * sizeof(wchar_t));
|
|
*(strDest + srcStrLen) = L'\0';
|
|
return EOK;
|
|
} else {
|
|
strDest[0] = L'\0';
|
|
SECUREC_ERROR_BUFFER_OVERLAP("wcsncpy_s");
|
|
return EOVERLAP_AND_RESET;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* <FUNCTION DESCRIPTION>
|
|
* The wcsncpy_s function copies not more than n successive wide characters
|
|
* (not including the terminating null wide character)
|
|
* from the array pointed to by strSrc to the array pointed to by strDest
|
|
*
|
|
* <INPUT PARAMETERS>
|
|
* strDest Destination string.
|
|
* destMax The size of the destination string, in characters.
|
|
* strSrc Source string.
|
|
* count Number of characters to be copied.
|
|
*
|
|
* <OUTPUT PARAMETERS>
|
|
* strDest is updated
|
|
*
|
|
* <RETURN VALUE>
|
|
* EOK Success
|
|
* EINVAL strDest is NULL and destMax != 0 and destMax <= SECUREC_WCHAR_STRING_MAX_LEN
|
|
* EINVAL_AND_RESET strDest != NULL and strSrc is NULLL and destMax != 0
|
|
* and destMax <= SECUREC_WCHAR_STRING_MAX_LEN
|
|
* ERANGE destMax > SECUREC_WCHAR_STRING_MAX_LEN or destMax is 0
|
|
* ERANGE_AND_RESET count > SECUREC_WCHAR_STRING_MAX_LEN or
|
|
* (destMax <= length of strSrc and destMax <= count and strDest != strSrc
|
|
* and strDest != NULL and strSrc != NULL and destMax != 0 and
|
|
* destMax <= SECUREC_WCHAR_STRING_MAX_LEN and not overlap)
|
|
* EOVERLAP_AND_RESET dest buffer and source buffer are overlapped and all parameters are valid
|
|
*
|
|
*
|
|
* If there is a runtime-constraint violation, strDest[0] will be set to the '\0' when strDest and destMax valid
|
|
*/
|
|
errno_t wcsncpy_s(wchar_t *strDest, size_t destMax, const wchar_t *strSrc, size_t count)
|
|
{
|
|
if (destMax == 0 || destMax > SECUREC_WCHAR_STRING_MAX_LEN) {
|
|
SECUREC_ERROR_INVALID_RANGE("wcsncpy_s");
|
|
return ERANGE;
|
|
}
|
|
if (strDest == NULL || strSrc == NULL) {
|
|
SECUREC_ERROR_INVALID_PARAMTER("wcsncpy_s");
|
|
if (strDest != NULL) {
|
|
strDest[0] = '\0';
|
|
return EINVAL_AND_RESET;
|
|
}
|
|
return EINVAL;
|
|
}
|
|
if (count > SECUREC_WCHAR_STRING_MAX_LEN) {
|
|
#ifdef SECUREC_COMPATIBLE_WIN_FORMAT
|
|
if (count == (size_t)(-1)) {
|
|
return SecDoWcsncpy(strDest, destMax, strSrc, destMax - 1);
|
|
}
|
|
#endif
|
|
strDest[0] = '\0'; /* clear dest string */
|
|
SECUREC_ERROR_INVALID_RANGE("wcsncpy_s");
|
|
return ERANGE_AND_RESET;
|
|
}
|
|
|
|
if (count == 0) {
|
|
strDest[0] = '\0';
|
|
return EOK;
|
|
}
|
|
|
|
return SecDoWcsncpy(strDest, destMax, strSrc, count);
|
|
}
|
|
|