X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Funixsock.c;h=45ed9c699b783fa9e9e10a4dfd482a594e59d1a4;hb=4fb5d4b471e3f6a24d7cd36272e48e57bcb2c68f;hp=63c3ae958d2e5aece8e857f738ad474196961b00;hpb=f59c94bac0b78d0ce4d53d39c84a88d0b6e5390a;p=collectd.git diff --git a/src/unixsock.c b/src/unixsock.c index 63c3ae95..45ed9c69 100644 --- a/src/unixsock.c +++ b/src/unixsock.c @@ -547,7 +547,7 @@ static int us_handle_listval (FILE *fh, char **fields, int fields_num) static void *us_handle_client (void *arg) { int fd; - FILE *fh; + FILE *fhin, *fhout; char buffer[1024]; char *fields[128]; int fields_num; @@ -558,8 +558,8 @@ static void *us_handle_client (void *arg) DEBUG ("Reading from fd #%i", fd); - fh = fdopen (fd, "r+"); - if (fh == NULL) + fhin = fdopen (fd, "r"); + if (fhin == NULL) { char errbuf[1024]; ERROR ("unixsock plugin: fdopen failed: %s", @@ -568,7 +568,17 @@ static void *us_handle_client (void *arg) pthread_exit ((void *) 1); } - while (fgets (buffer, sizeof (buffer), fh) != NULL) + fhout = fdopen (fd, "w"); + if (fhout == NULL) + { + char errbuf[1024]; + ERROR ("unixsock plugin: fdopen failed: %s", + sstrerror (errno, errbuf, sizeof (errbuf))); + fclose (fhin); /* this closes fd as well */ + pthread_exit ((void *) 1); + } + + while (fgets (buffer, sizeof (buffer), fhin) != NULL) { int len; @@ -593,31 +603,33 @@ static void *us_handle_client (void *arg) if (strcasecmp (fields[0], "getval") == 0) { - us_handle_getval (fh, fields, fields_num); + us_handle_getval (fhout, fields, fields_num); } else if (strcasecmp (fields[0], "putval") == 0) { - handle_putval (fh, fields, fields_num); + handle_putval (fhout, fields, fields_num); } else if (strcasecmp (fields[0], "listval") == 0) { - us_handle_listval (fh, fields, fields_num); + us_handle_listval (fhout, fields, fields_num); } else if (strcasecmp (fields[0], "putnotif") == 0) { - handle_putnotif (fh, fields, fields_num); + handle_putnotif (fhout, fields, fields_num); } else { - fprintf (fh, "-1 Unknown command: %s\n", fields[0]); - fflush (fh); + fprintf (fhout, "-1 Unknown command: %s\n", fields[0]); + fflush (fhout); } } /* while (fgets) */ DEBUG ("Exiting.."); - close (fd); + fclose (fhin); + fclose (fhout); pthread_exit ((void *) 0); + return ((void *) 0); } /* void *us_handle_client */ static void *us_server_thread (void *arg) @@ -747,7 +759,6 @@ static int us_shutdown (void) } plugin_unregister_init ("unixsock"); - plugin_unregister_write ("unixsock"); plugin_unregister_shutdown ("unixsock"); return (0);