从用户那里获取字符串的更好方法

Question

我想从用户那里得到一个字符串（其实我是想控制长度，但是没办法），然后输出。我尝试了一种可行的方法（但我认为这不是一个好的解决方案，因为它很容易溢出或占用太多内存。任何人都可以提供更好的方法来做到这一点？

#include<stdio.h>
#include<stdlib.h>

int main(){
  char p[1024];   //there if i use 'char *p=(char*)malloc(sizeof(char)*1024)' error too.
  int i=0;
  fgets(p,1024,stdin); //there if i use 'scanf("%s",p)', wrong.
  while(p[i]!='\0'&&i<1024){
    printf("%c",p[i]);
    i++;
  }
  return 0;
}

谁能提供更好的方法来解释我的错误？ 提前致谢。

Answer 1

我想控制长度，但是没办法

确实，代码无法通过魔术臂接触到用户并限制用户打字，但代码可以（并且应该 IMO）将输入 保存 限制在一定数量，比如 @ 987654321@ 阅读名称。

超出理智的输入最好被视为攻击而不是有效的极端。此类输入不需要完整保存，只需检测和处理即可。

使用fgets()

int main(void) {
  char p[1024];
  if (fgets(p, sizeof p,stdin) == NULL) {
    puts("Input closed");
    return -1;
  }
  size_t len = strlen(p);
  if (len > 0 && p[len - 1] == '\n') {  // lop off potential \n
    p[--len] = '\0';
  }
  if (len + 1 == sizeof p) {  // too long a line
    int ch;
    while ((ch = fgetc(stdin)) != '\n' && ch != EOF) {
      ; // consume rest of line
    }
    printf("Input too long, first part <%s>\n", p);
    return -1;
  }
  printf("Input OK <%s>\n", p);
  return 0;
}

Answer 2

如果我理解正确，您想读取字符串而不需要硬编码大小。 fgets 不可能做到这一点。你给它一个硬编码的大小，希望输入适合，如果输入不适合，它会被截断。

如果您的编译器与 POSIX 兼容，您可以使用 new(er) getline：

#include <stdio.h>
#include <stdlib.h>
int main() {
    char *p = NULL;
    size_t size = 0;
    getline(&p, &size, stdin);
    for (int i = 0; p[i] != '\0' &&i < size; i++)
        printf("%c", p[i]);
    free(p);
    printf("size %zu",size);
    return 0;
}

这样您就不必担心缓冲区的大小。 getline 负责处理。完成后只需free 内存即可。