《C语言 —— strncat、strncpy、snprintf、fgets》
在对缓冲区(一般为数组)进行拷贝前,要保证复制的长度不要超过缓冲区的空间大小。比如在 memcpy 前,要检查目的地址是否有足够的空间。
使用宏或 sizeof 可保证缓冲长度的一致性:
char dst_buf[64]; memcpy(dst_buf, src_buf,64)
当缓冲大小改变为 32 的时候,需要改动两处代码。一旦忘记修改 memcpy 处的拷贝长度,就会造成内存越界。
对上面的代码进行改善
#define BUF_SIZE 64 char dst_buf[BUF_SIZE]; memcpy(dst_buf, src_buf, BUF_SIZE);
或
char dst_buf[64]; memcpy(dst_buf, src_buf, sizeof(dst_buf));
这样就可以做到缓存大小和复制长度的同步修改。
使用安全的库函数也可以保证复制的长度不超过缓冲区的空间,下面来介绍 4 种库函数。
1. 使用 strncat 代替 strcat
#include <string.h> char *strncat(char *dest, const char *src, size_t n); char dest[20] = "hello"; strncat(dest, src, sizeof(dest)-strlen(dest)-1);
从 src 中最多追加 n 个字符到 dest 字符串的后面。需要注意的是,当 src 包含 n 个以上的字符时, dest 的空间至少为 strlen ( dest ) +n+1 ,因为该函数还会追加字符串结束符 '\0' 到 dest 后面。
一定要记住给 '\0' 留下空间。
2. 使用 strncpy 代替 strcpy
#include <string.h> char *strncpy(char *dest, const char *src, size_t n); char dest[20]; strncpy(dest, src, sizeof(dest)-1);
从 src 中最多复制 n 个字符到 dest 字符串中。与 strncat 相同的是,当 src 包含 n 个以上的字符时, dest 的空间需要为 n+1 ,因为该函数还会再复制一个字符串结束符 '\0' 。
3.使用 snprintf 代替 sprintf
#include <stdio.h> int snprintf(char *str, size_t size, const char *format, ...); char str[20]; snprintf(str, sizeof(str), "%s", dest0);
4.使用 fgets 代替 gets
#include <stdio.h> char *fgets(char *s, int size, FILE *stream); char str[20]; fgets(str, sizeof(str), stdin)
危险的 gets 函数从来不检查缓冲区的大小,并且还是从标准输入中读取数据,这是极其危险的行为。再大的缓存空间也无法满足永无终止的标准输入,因此一定要使用 fgets 代替。
fgets 最多会复制 size-1 字节到缓存 s 中,并且会在最后一个字符后面追加 '\0' 。因此如果要读取标准输入。
5.strstr函数原理实现
(34条消息) C语言力扣第28题之实现Strstr。KMP算法_管二狗绝不摆烂的博客-CSDN博客_力扣28题
浙公网安备 33010602011771号