tweak

[krsd] / src / tools / remove-token.c

diff --git a/src/tools/remove-token.c b/src/tools/remove-token.c
index 0fba01b..4143b3f 100644 (file)
--- a/src/tools/remove-token.c
+++ b/src/tools/remove-token.c
@@ -10,11 +10,16 @@
  * along with this program. If not, see <http://www.gnu.org/licenses/>.
  */
 
+#define _GNU_SOURCE
+
 #include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
 #include <fcntl.h>
 #include <errno.h>
+#include <string.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <stdint.h>
+#include <db.h>
 
 #include "common/auth.h"
 
@@ -31,7 +36,8 @@ int main(int argc, char **argv) {
   struct rs_authorization auth;
   auth.username = argv[1];
   auth.token = argv[2];
-  auth.scopes = NULL;
-  remove_authorization(&auth);
+  int success = remove_authorization(&auth);
   close_authorizations();
+  fprintf(stderr, (success == DB_NOTFOUND) ? "Token not found!\n" : (success == 0 ? "Token removed.\n" : "Error removing token!\n"));
+  return success;
 }