add length constrain in server

0xuye0 · Sep 8, 2013 · 946512e · 946512e
1 parent 9e598a8
commit 946512e
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/server/ltp_server.cpp b/src/server/ltp_server.cpp
@@ -135,7 +135,28 @@ static int Service(struct mg_connection *conn) {
                 // failed the xml validation check
                 return 0;
             }
+
+            // guanrante the input sentence is not to long.
+            int  num_para = xml4nlp.CountParagraphInDocument();
+            bool flag = true;
+            for (int i = 0; i < num_para; ++ i) {
+                int num_sent = xml4nlp.CountSentenceInParagraph(i);
+                for (int j = 0; j < num_sent; ++ j) { 
+                    if (length(std::string(xml4nlp.GetSentence(i,j))) > 70) {
+                        flag = false;
+                    }
+                }
+            }
+
+            if (!flag) {
+                WARNING_LOG("Input sentence is too long");
+                return 0;
+            }
         } else {
+            if (length(strSentence) > 70) {
+                WARNING_LOG("Input sentence is too long");
+                return 0;
+            }
             xml4nlp.CreateDOMFromString(strSentence);
         }