fix memory leaks

main.c

@@ -12,8 +12,9 @@ const long DEFAULT_DASH_PAUSE = 300000;
 const long DEFAULT_DOT_PAUSE = 150000;
 
 int main(int argc, char *argv[]){
-  char* input_string = malloc(sizeof(char) * MAX_LINE_SIZE);
-  char* output_string = malloc(sizeof(char) * MAX_LINE_SIZE);
+  int string_len;
+  char* input_string = NULL;
+  char* output_string = (char*) malloc(sizeof(char) * MAX_LINE_SIZE);
   // output_char tracks the current printing character when slow printing is enabled
   char output_char;
 
@@ -48,12 +49,10 @@ int main(int argc, char *argv[]){
 
 	if (convert_from_morse){
 		// for every line in the input, until the end of file
-		while ((getline(&input_string, &buflen, stdin))!=EOF){
-            int stringLen = strlen(input_string);
-
+		while ((string_len = getline(&input_string, &buflen, stdin))!=EOF){
       // This effectively strips the newline off the string
-            if (input_string[stringLen-1] == '\n'){
-                input_string[stringLen-1] = '\0';
+      if (input_string[string_len-1] == '\n'){
+          input_string[string_len-1] = '\0';
       }
             
       morse_to_string(input_string, output_string);
@@ -63,12 +62,10 @@ int main(int argc, char *argv[]){
     }	
 	} else {
 		// for each line of input
-		while ((getline(&input_string, &buflen, stdin))!=EOF){
-            int stringLen = strlen(input_string);
-
+		while ((string_len = getline(&input_string, &buflen, stdin))!=EOF){
       // This effectively strips the newline off the string
-            if (input_string[stringLen-1] == '\n'){
-                input_string[stringLen-1] = '\0';
+      if (input_string[string_len-1] == '\n'){
+          input_string[string_len-1] = '\0';
       }
 
       string_to_morse(input_string, output_string);
@@ -95,6 +92,7 @@ int main(int argc, char *argv[]){
       }
 		}
 	}
+
   free(output_string);
   free(input_string);
 	return 0;