Explorar o código

Improve socket init and close in tests

Stéphane Raimbault %!s(int64=12) %!d(string=hai) anos
pai
achega
9ba562e558
Modificáronse 3 ficheiros con 12 adicións e 6 borrados
  1. 4 2
      tests/bandwidth-server-many-up.c
  2. 4 2
      tests/random-test-server.c
  3. 4 2
      tests/unit-test-server.c

+ 4 - 2
tests/bandwidth-server-many-up.c

@@ -35,12 +35,14 @@
 #define NB_CONNECTION    5
 #define NB_CONNECTION    5
 
 
 modbus_t *ctx = NULL;
 modbus_t *ctx = NULL;
-int server_socket;
+int server_socket = -1;
 modbus_mapping_t *mb_mapping;
 modbus_mapping_t *mb_mapping;
 
 
 static void close_sigint(int dummy)
 static void close_sigint(int dummy)
 {
 {
-    close(server_socket);
+    if (server_socket != -1) {
+        close(server_socket);
+    }
     modbus_free(ctx);
     modbus_free(ctx);
     modbus_mapping_free(mb_mapping);
     modbus_mapping_free(mb_mapping);
 
 

+ 4 - 2
tests/random-test-server.c

@@ -26,7 +26,7 @@
 
 
 int main(void)
 int main(void)
 {
 {
-    int socket;
+    int socket = -1;
     modbus_t *ctx;
     modbus_t *ctx;
     modbus_mapping_t *mb_mapping;
     modbus_mapping_t *mb_mapping;
 
 
@@ -60,7 +60,9 @@ int main(void)
 
 
     printf("Quit the loop: %s\n", modbus_strerror(errno));
     printf("Quit the loop: %s\n", modbus_strerror(errno));
 
 
-    close(socket);
+    if (socket != -1) {
+        close(socket);
+    }
     modbus_mapping_free(mb_mapping);
     modbus_mapping_free(mb_mapping);
     modbus_close(ctx);
     modbus_close(ctx);
     modbus_free(ctx);
     modbus_free(ctx);

+ 4 - 2
tests/unit-test-server.c

@@ -32,7 +32,7 @@ enum {
 
 
 int main(int argc, char*argv[])
 int main(int argc, char*argv[])
 {
 {
-    int socket;
+    int socket = -1;
     modbus_t *ctx;
     modbus_t *ctx;
     modbus_mapping_t *mb_mapping;
     modbus_mapping_t *mb_mapping;
     int rc;
     int rc;
@@ -187,7 +187,9 @@ int main(int argc, char*argv[])
     printf("Quit the loop: %s\n", modbus_strerror(errno));
     printf("Quit the loop: %s\n", modbus_strerror(errno));
 
 
     if (use_backend == TCP) {
     if (use_backend == TCP) {
-        close(socket);
+        if (socket != -1) {
+            close(socket);
+        }
     }
     }
     modbus_mapping_free(mb_mapping);
     modbus_mapping_free(mb_mapping);
     free(query);
     free(query);