#if HAVE_CONFIG_H
#include "clamav-config.h"
#endif
#ifdef BUILD_CLAMD
#include <stdio.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <string.h>
#include "others.h"
#include "cfgparser.h"
#include "output.h"
int notify(const char *cfgfile)
{
char buff[20];
struct sockaddr_un server;
struct sockaddr_in server2;
struct hostent *he;
struct cfgstruct *copt, *cpt;
int sockd, bread;
char *socktype;
if((copt = parsecfg(cfgfile, 1)) == NULL) {
mprintf("@Clamd was NOT notified: Can't find or parse configuration file %s\n", cfgfile);
return 1;
}
if(cfgopt(copt, "TCPSocket") && cfgopt(copt, "LocalSocket")) {
mprintf("@Clamd was NOT notified: Both socket types (TCP and local) declared in %s\n", cfgfile);
return 1;
} else if((cpt = cfgopt(copt, "LocalSocket"))) {
socktype = "UNIX";
server.sun_family = AF_UNIX;
strncpy(server.sun_path, cpt->strarg, sizeof(server.sun_path));
if((sockd = socket(AF_UNIX, SOCK_STREAM, 0)) < 0) {
mprintf("@Clamd was NOT notified: Can't create socket endpoint for %s\n", cpt->strarg);
perror("socket()");
return 1;
}
if(connect(sockd, (struct sockaddr *) &server, sizeof(struct sockaddr_un)) < 0) {
close(sockd);
mprintf("@Clamd was NOT notified: Can't connect to clamd through %s\n", cpt->strarg);
perror("connect()");
return 1;
}
} else if((cpt = cfgopt(copt, "TCPSocket"))) {
socktype = "TCP";
#ifdef PF_INET
if((sockd = socket(PF_INET, SOCK_STREAM, 0)) < 0) {
#else
if((sockd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
#endif
mprintf("@Clamd was NOT notified: Can't create TCP socket\n");
perror("socket()");
return 1;
}
server2.sin_family = AF_INET;
server2.sin_port = htons(cpt->numarg);
if ((cpt = cfgopt(copt, "TCPAddr"))) {
if ((he = gethostbyname(cpt->strarg)) == 0) {
perror("gethostbyname()");
mprintf("@Clamd was NOT notified: Can't resolve hostname '%s'\n", cpt->strarg);
return 1;
}
server2.sin_addr = *(struct in_addr *) he->h_addr_list[0];
} else
server2.sin_addr.s_addr = inet_addr("127.0.0.1");
if(connect(sockd, (struct sockaddr *) &server2, sizeof(struct sockaddr_in)) < 0) {
close(sockd);
mprintf("@Clamd was NOT notified: Can't connect to clamd on %s:%d\n",
inet_ntoa(server2.sin_addr), ntohs(server2.sin_port));
perror("connect()");
return 1;
}
} else {
mprintf("@Clamd was NOT notified: No socket specified in %s\n", cfgfile);
return 1;
}
if(write(sockd, "RELOAD", 6) < 0) {
mprintf("@Clamd was NOT notified: Could not write to %s socket\n", socktype);
perror("write()");
close(sockd);
return 1;
}
memset(buff, 0, sizeof(buff));
if((bread = read(sockd, buff, sizeof(buff))) > 0)
if(!strstr(buff, "RELOADING")) {
mprintf("@Clamd was NOT notified: Unknown answer from clamd: '%s'\n", buff);
close(sockd);
return 1;
}
close(sockd);
mprintf("Clamd successfully notified about the update.\n");
logg("Clamd successfully notified about the update.\n");
return 0;
}
#endif