Add forgotten files 1.4.70b
[polyglot.git] / engine.c
index bf7bd49..5ed6ed7 100644 (file)
--- a/engine.c
+++ b/engine.c
-// engine.c\r
-\r
-// includes\r
-\r
-#include <stdlib.h>\r
-#include <stdarg.h>\r
-#include <string.h>\r
-#include <errno.h>\r
-\r
-\r
-#include "engine.h"\r
-#include "option.h"\r
-#include "pipex.h"\r
-#include "util.h"\r
-\r
-// defines\r
-\r
-#define StringSize 4096\r
-\r
-// variables\r
-\r
-static int write_index = 0;\r
-static char write_buffer[StringSize];\r
-engine_t Engine[1];\r
-\r
-// functions\r
-\r
-// set_affinity()\r
-\r
-void set_affinity(engine_t *engine, int value){\r
-    pipex_set_affinity(engine->pipex,value);\r
-}\r
-\r
-// engine_set_nice_value()\r
-\r
-void engine_set_nice_value(engine_t *engine, int value){\r
-    pipex_set_priority(engine->pipex,value);\r
-}\r
-\r
-// engine_send_queue()\r
-\r
-void engine_send_queue(engine_t * engine, const char *format, ...) {\r
-    va_list arg_list;\r
-    if(write_index>=StringSize){\r
-        my_fatal("engine_send_queue(): write_buffer overflow\n");\r
-    }\r
-    va_start(arg_list,format);\r
-    write_index += vsnprintf(write_buffer + write_index,\r
-                             StringSize-write_index,\r
-                             format,\r
-                             arg_list);\r
-    va_end(arg_list);\r
-}\r
-\r
-// engine_send()\r
-\r
-void engine_send(engine_t * engine, const char *format, ...) {\r
-    va_list arg_list;\r
-    if(write_index>=StringSize){\r
-        my_fatal("engine_send(): write_buffer overflow\n");\r
-    }\r
-    va_start(arg_list,format);\r
-    vsnprintf(write_buffer + write_index,\r
-              StringSize-write_index,\r
-              format,\r
-              arg_list);\r
-    va_end(arg_list);\r
-    pipex_writeln(engine->pipex,write_buffer);\r
-    write_index = 0;\r
-}\r
-\r
-// engine_close()\r
-\r
-void engine_close(engine_t * engine){\r
-    char string[StringSize];\r
-    pipex_send_eof(engine->pipex);\r
-        // TODO: Timeout\r
-    while (!engine_eof(engine)) { \r
-      engine_get(Engine,string);\r
-    }\r
-    pipex_exit(engine->pipex);\r
-}\r
-\r
-// engine_open()\r
-\r
-void engine_open(engine_t * engine){\r
-    int affinity;\r
-    char *my_dir;\r
-    if( (my_dir = my_getcwd( NULL, 0 )) == NULL )\r
-        my_fatal("engine_open(): no current directory: %s\n",strerror(errno));\r
-    if(my_chdir(option_get_string("EngineDir"))){\r
-        my_fatal("engine_open(): cannot change directory: %s\n",strerror(errno));\r
-    }\r
-    pipex_open(engine->pipex,"Engine",option_get_string("EngineCommand"));\r
-    if(pipex_active(engine->pipex)){\r
-            //play with affinity (bad idea)\r
-        affinity=option_get_int("Affinity");\r
-        if(affinity!=-1) set_affinity(engine,affinity); //AAA\r
-            //lets go back\r
-        my_chdir(my_dir);\r
-            // set a low priority\r
-        if (option_get_bool("UseNice")){\r
-            my_log("POLYGLOT Adjust Engine Piority\n");\r
-            engine_set_nice_value(engine, option_get_int("NiceValue"));\r
-        }\r
-    }\r
-    \r
-}\r
-\r
-bool engine_active(engine_t *engine){\r
-    return pipex_active(engine->pipex);\r
-}\r
-\r
-bool engine_eof(engine_t *engine){\r
-    return pipex_eof(engine->pipex);\r
-}\r
-\r
-bool engine_get_non_blocking(engine_t * engine, char *string){\r
-    return pipex_readln_nb(engine->pipex,string);\r
- }\r
-\r
-void engine_get(engine_t * engine, char *string){\r
-    pipex_readln(engine->pipex,string);\r
-}\r
-\r
-\r
+// engine.c
+
+// includes
+
+#include <stdlib.h>
+#include <stdarg.h>
+#include <string.h>
+#include <errno.h>
+
+#include "engine.h"
+#include "option.h"
+#include "pipex.h"
+#include "util.h"
+
+// variables
+
+engine_t Engine[1];
+
+// functions
+
+// set_affinity()
+
+void set_affinity(engine_t *engine, int value){
+    pipex_set_affinity(engine->pipex,value);
+}
+
+// engine_set_nice_value()
+
+void engine_set_nice_value(engine_t *engine, int value){
+    pipex_set_priority(engine->pipex,value);
+}
+
+// engine_send_queue()
+
+void engine_send_queue(engine_t * engine, const char *format, ...) {
+    char buf[FormatBufferSize];
+    CONSTRUCT_ARG_STRING(format,buf);
+    pipex_write(engine->pipex,buf);
+}
+
+// engine_send()
+
+void engine_send(engine_t * engine, const char *format, ...) {
+    char buf[FormatBufferSize];
+    CONSTRUCT_ARG_STRING(format,buf);
+    pipex_writeln(engine->pipex,buf);
+}
+
+// engine_close()
+
+void engine_close(engine_t * engine){
+    char string[StringSize];
+    int elapsed_time;
+    int ret;
+    int close_timeout=500;
+    my_log("POLYGLOT Closing engine.\n");
+    pipex_send_eof(engine->pipex);
+        // TODO: Timeout
+
+    elapsed_time=0;
+    while (!engine_eof(engine) && (elapsed_time<close_timeout)) { 
+      ret=engine_get_non_blocking(engine,string);
+      if(!ret  && !engine_eof(engine)){
+       my_log("POLYGLOT Engine does not reply. Sleeping %dms.\n", WAIT_GRANULARITY);
+       my_sleep(WAIT_GRANULARITY);
+       elapsed_time+=WAIT_GRANULARITY;
+      }
+    }
+    if(elapsed_time>=close_timeout){
+      my_log("POLYGLOT Waited more than %dms. Moving on.\n",close_timeout); 
+    }
+    pipex_exit(engine->pipex,200);
+}
+
+// engine_open()
+
+void engine_open(engine_t * engine){
+    int affinity;
+    pipex_open(engine->pipex,
+               "Engine",
+               option_get_string(Option,"EngineDir"),
+               option_get_string(Option,"EngineCommand"));
+    if(pipex_active(engine->pipex)){
+            //play with affinity (bad idea)
+        affinity=option_get_int(Option,"Affinity");
+        if(affinity!=-1) set_affinity(engine,affinity); //AAA
+            // set a low priority
+        if (option_get_bool(Option,"UseNice")){
+            my_log("POLYGLOT Adjust Engine Piority\n");
+            engine_set_nice_value(engine, option_get_int(Option,"NiceValue"));
+        }
+    }
+    
+}
+
+bool engine_active(engine_t *engine){
+    return pipex_active(engine->pipex);
+}
+
+bool engine_eof(engine_t *engine){
+    return pipex_eof(engine->pipex);
+}
+
+bool engine_get_non_blocking(engine_t * engine, char *string){
+    return pipex_readln_nb(engine->pipex,string);
+ }
+
+void engine_get(engine_t * engine, char *string){
+    pipex_readln(engine->pipex,string);
+}
+
+