[已解决]C语言给字符串赋值的问题

rt3

如果 使用scanf() 或 gets() 赋值可能造成缓冲区溢出，
但是 简单的使用 fgets() 代替 又会把 '\n' 输入进去。
我知道 可以 使用 循环 ，但是 有没有 更好的办法呢。
谢谢

最佳答案

月排行榜 / 总排行榜

永恒的蓝色梦想

2020-7-31 21:26:52

楼上的效率不行，用这个：

1. char* safe_gets(char* const _Buffer, size_t _Size) {
   
2.     if (_Buffer && _Size) {
   
3.         char* _Str = _Buffer;
   
4. 
   
5.         while (--_Size) {
   
6.             switch (*_Str = getchar()) {
   
7.                 case EOF:
   
8.                 case '\0':
   
9.                 case '\r':
   
10.                 case '\n':
    
11.                     goto _End;
    
12.             }
    
13. 
    
14.             ++_Str;
    
15.         }
    
16. 
    
17.     _End:
    
18.         *_Str = '\0';
    
19.     }
    
20. 
    
21.     return _Buffer;
    
22. }

复制代码

跳转到最佳答案楼层

永恒的蓝色梦想

楼上的效率不行，用这个：

1. char* safe_gets(char* const _Buffer, size_t _Size) {
   
2.     if (_Buffer && _Size) {
   
3.         char* _Str = _Buffer;
   
4. 
   
5.         while (--_Size) {
   
6.             switch (*_Str = getchar()) {
   
7.                 case EOF:
   
8.                 case '\0':
   
9.                 case '\r':
   
10.                 case '\n':
    
11.                     goto _End;
    
12.             }
    
13. 
    
14.             ++_Str;
    
15.         }
    
16. 
    
17.     _End:
    
18.         *_Str = '\0';
    
19.     }
    
20. 
    
21.     return _Buffer;
    
22. }

复制代码

urva

自己写一个吧，替代fgets();
#include<string.h>
char * s_gets(char *st, int n) //只需要传递一个字符串指针和字符串长度
{
    char *temp;
    char *find_n;
    temp = fgets(st, n, stdin);
    if(temp)
       {
           find_n = strchr(temp, '\n');
          if(find_n)
            *find_n = '\0';
          else
             while(getchar()!='\n')
                 continue;
       }
     return temp;
}