strstr的安全版本

最新推荐文章于 2023-10-15 17:50:10 发布

小麻侬

最新推荐文章于 2023-10-15 17:50:10 发布

阅读量426

点赞数

文章标签： c++ strstr 安全

本文链接：https://blog.csdn.net/wutaozhao/article/details/127802730

版权

字符串处理内存安全子串查找 C语言编程算法

关键词由CSDN通过智能技术生成

inline char* findSubStr(char* targetData, int dataLen, const char* subStr, int subStrLen){
	if (dataLen < subStrLen || dataLen <= 0 || subStrLen <= 0){
		return NULL;
	}

	char lastChar = targetData[dataLen - 1];
	targetData[dataLen - 1] = '\0';

	char* pFind = NULL;
	pFind = strstr(targetData, subStr);
	targetData[dataLen - 1] = lastChar;

	// 纯二进制下，尾部是否一致
	if (!pFind){
		bool hasNotSameChar = false;
		for (int i = 0; i < subStrLen; i++){
			if (subStr[i] != targetData[dataLen - subStrLen + i]){
				hasNotSameChar = true;
				break;
			}
		}

		if (!hasNotSameChar){
			pFind = &targetData[dataLen - subStrLen];
		}
	}

	return pFind;
}

由于strstr查找的目标地址可能不是以'\0'结束，这样就可能有越界的风险，上面代码则不会越界。