好吧,你应该确保有足够的空间fname
来存储你的文件名,否则你几乎肯定会导致损坏和“炸毁”:-)
例如,以下代码段:
char fname[10];
gets (fname);
如果您输入的信息多于fname
无法容纳的信息,将会出现问题。那时,您进入了未定义的行为领域,任何事情都可能发生。
但是,由于gets
无法限制用户输入的内容,因此您永远不应该使用它!
可以在此答案中找到一种适当的、受保护的用户输入方法。
它使用fgets
,因为这可以限制用户输入的内容。它还允许提示,在出现问题时提供错误指示,正确处理文件结尾,并删除任何过大行的其余部分,使其不会影响下一个输入操作。
事实上,我会在这里复制它以使这个答案自成一体:
#include <stdio.h>
#include <string.h>
#define OK 0
#define NO_INPUT 1
#define TOO_LONG 2
static int getLine (char *prmpt, char *buff, size_t sz) {
int ch, extra;
// Get line with buffer overrun protection.
if (prmpt != NULL) {
printf ("%s", prmpt);
fflush (stdout);
}
if (fgets (buff, sz, stdin) == NULL)
return NO_INPUT;
// If it was too long, there'll be no newline. In that case, we flush
// to end of line so that excess doesn't affect the next call.
if (buff[strlen(buff)-1] != '\n') {
extra = 0;
while (((ch = getchar()) != '\n') && (ch != EOF))
extra = 1;
return (extra == 1) ? TOO_LONG : OK;
}
// Otherwise remove newline and give string back to caller.
buff[strlen(buff)-1] = '\0';
return OK;
}
您可以按如下方式调用它,指定缓冲区和大小,并在返回时接收错误指示:
// Test program for getLine().
int main (void) {
int rc;
char buff[10];
rc = getLine ("Enter string> ", buff, sizeof(buff));
if (rc == NO_INPUT) {
// Extra NL since my system doesn't output that on EOF.
printf ("\nNo input\n");
return 1;
}
if (rc == TOO_LONG) {
printf ("Input too long [%s]\n", buff);
return 1;
}
printf ("OK [%s]\n", buff);
return 0;
}