X-Git-Url: http://andersk.mit.edu/gitweb/moira.git/blobdiff_plain/7ac48069b111a991ee5975cb6088c4563b57b670..a3bbcc2b4dda82b95c21041349bcfbada22c45ca:/lib/mr_query.c

diff --git a/lib/mr_query.c b/lib/mr_query.c
index f9374d01..af22ef89 100644
--- a/lib/mr_query.c
+++ b/lib/mr_query.c
@@ -1,4 +1,4 @@
-/* $Id $
+/* $Id$
  *
  * Perform a Moira query
  *
@@ -12,6 +12,7 @@
 #include <moira.h>
 #include "mr_private.h"
 
+#include <errno.h>
 #include <stdlib.h>
 #include <string.h>
 
@@ -34,6 +35,8 @@ int mr_query(char *name, int argc, char **argv,
   char **nargv = malloc(sizeof(char *) * (argc + 1));
   int status = 0;
 
+  if (!nargv)
+    return ENOMEM;
   nargv[0] = name;
   memcpy(nargv + 1, argv, sizeof(char *) * argc);
   status = mr_query_internal(argc + 1, nargv, callproc, callarg);