democlient: handle input through QUERY_STRING if stdin doesn't work

nomick · Nov 28, 2012 · 7190443 · 7190443
1 parent 40719c5
commit 7190443
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/democlient/democlient.cpp b/democlient/democlient.cpp
@@ -67,10 +67,16 @@ class CppcheckExecutor : public ErrorLogger {
 
 int main()
 {
-    const char *lenstr = getenv("CONTENT_LENGTH");
     char data[4096] = {0};
-    int len = std::min(1 + atoi(lenstr), (int)(sizeof(data) - 2));
-    fgets(data, len, stdin);
+
+    const char *lenstr = getenv("CONTENT_LENGTH");
+    if (lenstr) {
+        int len = std::min(1 + atoi(lenstr), (int)(sizeof(data) - 2));
+        fgets(data, len, stdin);
+    } else {
+	const char *s = getenv("QUERY_STRING");
+        strncpy(data, s?s:"", sizeof(data)-2);
+    }
 
     char code[4096] = {0};
     unencode(data, code);