Update set builtin too

diff --git a/src/builtins/set.c b/src/builtins/set.c
index df28cc8f209ab8b6569cf60d6eabfe153b279937..f8df49eddfebce95f15322ed2a98e36467cfbced 100644 (file)
--- a/src/builtins/set.c
+++ b/src/builtins/set.c
@@ -4,12 +4,17 @@
 #include "utils.h"
 
 BUILTIN(set) {
-   if (argc != 3) return usage(argv[0], "name value");
-
-   if (setenv(argv[1], argv[2], 1) == -1) {
-       note("Unable to set %s to %s", argv[1], argv[2]);
-       return EXIT_FAILURE;
+   switch (argc) {
+   case 3:
+       if (setenv(argv[1], argv[2], 1) == -1) {
+           note("Unable to set %s to %s", argv[1], argv[2]);
+           return EXIT_FAILURE;
+       }
+   case 2:
+       break;
+   default:
+       return usage(argv[0], "name [value]");
    }
-   
+
    return EXIT_SUCCESS;
 }

diff --git a/src/builtins/unset.c b/src/builtins/unset.c
index c7eb78fc8cdc104cd6a2d472ae473c8c211d8d8f..d7a5679e628c648edc3a2e6d726794942b8c2478 100644 (file)
--- a/src/builtins/unset.c
+++ b/src/builtins/unset.c
@@ -6,10 +6,8 @@
 BUILTIN(unset) {
    if (argc != 2) return usage(argv[0], "name");
 
-   if (unsetenv(argv[1]) == -1) {
-       note("Unable to unset $%s$", argv[1]);
-       return EXIT_FAILURE;
-   }
+   if (unsetenv(argv[1]) != -1) return EXIT_SUCCESS;
 
-   return EXIT_SUCCESS;
+   note("Unable to unset $%s$", argv[1]);
+   return EXIT_FAILURE;
 }