浏览代码

applied Jukka's stdinread patch

arg@mig29 19 年之前
父节点
当前提交
27ef73507b
共有 1 个文件被更改,包括 3 次插入2 次删除
  1. 3 2
      main.c

+ 3 - 2
main.c

@@ -274,9 +274,10 @@ main(int argc, char *argv[]) {
 				eprint("select failed\n");
 		}
 		if(FD_ISSET(STDIN_FILENO, &rd)) {
-			switch(r = read(STDIN_FILENO, stext, sizeof(stext))) {
+			switch(r = read(STDIN_FILENO, stext, sizeof(stext) - 1)) {
 			case -1:
 				strncpy(stext, strerror(errno), sizeof(stext));
+				stext[sizeof(stext) - 1] = '\0';
 				readin = False;
 				break;
 			case 0:
@@ -284,7 +285,7 @@ main(int argc, char *argv[]) {
 				readin = False;
 				break;
 			default:
-				stext[r-1] = 0;
+				stext[r - (stext[r - 1] == '\n' ? 1 : 0)] = '\0';
 			}
 			drawstatus();
 		}