update

Jenni-Foued · Oct 27, 2020 · dcbc8ca · dcbc8ca
1 parent 3dedb05
commit dcbc8ca
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/print_rot13.c b/print_rot13.c
@@ -1,5 +1,6 @@
 #include "holberton.h"
 #include <stdarg.h>
+#include <stdlib.h>
 /**
  * print_rot13 - Prints a string in rot13.
  * @arg: String address.
@@ -8,13 +9,16 @@
 int print_rot13(va_list arg)
 {
 	int i, j, count = 0;
-	char a[] = "AaBbCcDdEeFfGgHhIiJjKkLlMm";
-	char b[] = "NnOoPpQqRrSsTtUuVvWwXxYyZz";
+	char a[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz ";
+	char b[] = "NOPQRSTUVWXYZABCDEFGHIJKLMnopqrstuvwxyzabcdefghijklm";
 	char *s = va_arg(arg, char *);
 
+	if (!s)
+		s = "empty";
+
 	for (i = 0; s[i]; i++)
 	{
-		for (j = 0; j < 26; j++)
+		for (j = 0; j < 52; j++)
 		{
 			if (s[i] == a[j])
 			{