《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题

 

posted @ 2022-09-15 19:13  一个不知道干嘛的小萌新  阅读(212)  评论(0)    收藏  举报