1 files changed, 5 insertions, 4 deletions
diff --git a/main.c b/main.c
index 47cdc0e..5922b3d 100644
--- a/main.c
+++ b/main.c
@@ -97,7 +97,11 @@ chomp(char *inbuf, char *outbuf, int start, int len) {
 		free(rem);
 		rem = NULL;
 	}
-	while((off < len) && (off < MAX_LINE_LEN)) {
+	while(off < len) {
+		if(i > MAX_LINE_LEN) {
+			outbuf[i] = '\0';
+			return ++off;
+		}
 		if(inbuf[off] != '\n') {
 			 outbuf[i++] = inbuf[off++];
 		} else if(inbuf[off] == '\n') {
@@ -106,9 +110,6 @@ chomp(char *inbuf, char *outbuf, int start, int len) {
 		}
 	}
 
-	if(off >= MAX_LINE_LEN)
-		return 0;
-
 	outbuf[i] = '\0';
 	rem = estrdup(outbuf);
 	return 0;