WebSVN – heating – Diff – /trunk/main.cpp

 #include <sys/ioctl.h>
 #include <sys/param.h>
 #include <pwd.h>
 #include <grp.h>
 #include "config.h"
+#include "heating.h"
 struct SOCKETS
+{
 	int sockfd;
 	int newfd;
 void sig_handler(int sig);
 void changeToUser(const char *usr, const char *grp);
 void sig_child ();
 void *pthr_parser(void *pV_data);
-void *pthr_Probe(void * pV_data);
+void *pthr_Heat(void * pV_data);
 void *processCommands(void *pV_data);
 int parseCommand(int s1, char *buf);
+static pthread_t pthr_pars, pthr_heat, pthr_process;
+static pthread_mutex_t fastmutex_proc = PTHREAD_MUTEX_INITIALIZER;
+heating *heat;
 struct SOCKETS soc;				// The network sockets to be passed to a thread
 int main(int argc, char **argv)
+{
+	heat = nullptr;
 	// First read config file and check if it was ok
 	config *cfg = new config();
 	if (!cfg->is_initialized())
+	{
 		std::cerr << "Error reading config file! Make sure there is one and that it is readable by root!" << std::endl;
-		exit (0);
+		delete (cfg);
+		return 1;
+	}
 	CONFIGURE configs = cfg->getConfig();
 	// Daemonize and run in background
 	daemon_start(0);
 	delete (cfg);
 	// Now start our Thread
 	if (pthread_create(&pthr_pars, NULL, pthr_parser, (void *)0) != 0)
+	{
 		syslog (LOG_DAEMON,"Create of thread \"pthr_parser\" failed!");
-		return 1;
+		return 2;
+	}
 	while (1)
 		sleep(3600);
 	return 0;
+}
+/*
+ * Detach application from console and make it a daemon.
+ */
+void daemon_start (int ignsigcld)
+{
+	int childpid, fd;
+	char hv0[128];
+	if (getpid () == 1)
+		goto out;
+	#ifdef SIGTTOU
+	signal (SIGTTOU, SIG_IGN);
+	#endif
+	#ifdef SIGTTIN
+	signal (SIGTTIN, SIG_IGN);
+	#endif
+	#ifdef SIGTSTP
+	signal (SIGTSTP, SIG_IGN);
+	#endif
+	if ((childpid = fork ()) < 0)
+		fprintf (stderr, "Can't fork this child\n");
+	else if (childpid > 0)
+		exit (0);
+	if (setpgrp () == -1)
+		fprintf (stderr, "Can't change process group\n");
+	signal (SIGHUP, SIG_IGN);
+	if ((childpid = fork ()) < 0)
+		syslog (LOG_DAEMON, "Can't fork second child");
+	else if (childpid > 0)
+		exit (0);            /* first child */
+	/* second child */
+out:
+	for (fd = 0; fd < NOFILE; fd++)
+		close (fd);
+	errno = 0;
+	chdir ("/");
+	umask (0);
+	if (ignsigcld)
+		signal (SIGCLD, SIG_IGN);
+	// Define a signal handler for terminate signals
+	if (signal(SIGTERM, sig_handler) == SIG_ERR)
+		syslog(LOG_WARNING,"Can't catch signal SIGTERM!");
+	// Create PID file
+	if ((fd = open(Configure.pidfile, O_RDWR | O_TRUNC | O_CREAT, S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH)) == -1)
+	{
+		syslog(LOG_WARNING,"Can't create PID file %s: %s",Configure.pidfile, strerror(errno));
+		return;
+	}
+	sprintf(hv0, "%d", getpid());
+	write(fd, hv0, strlen(hv0));
+	close(fd);
+}
+void sig_child ()
+{
+#if defined(BSD) && !defined(sinix) && !defined(Linux)
+	int pid;
+	union wait status;
+	while ((pid = wait3 (&status, WNOHANG, (struct rusage *)0)) > 0)
+		;
+#endif
+}
+/*
+ * This is the signal handler who disconnects from network and terminates
+ * this daemon
+ */
+void sig_handler(int sig)
+{
+	if (sig == SIGTERM || sig == SIGKILL)
+	{
+		syslog(LOG_INFO, "Terminating program! Killed by signal %d", sig);
+		if (heat != nullptr)
+		{
+			heat->stopRun();
+			sleep(2);
+		}
+		exit(0);
+	}
+}
+void changeToUser(const char *usr, const char *grp)
+{
+gid_t gr_gid;
+	if (usr && strlen(usr))
+	{
+		/* get uid */
+		struct passwd *userpwd;
+		struct group *usergrp;
+		if ((userpwd = getpwnam(usr)) == NULL)
+		{
+			syslog(LOG_DAEMON,"no such user: %s", usr);
+			exit(EXIT_FAILURE);
+		}
+		if (!grp || !strlen(grp) || (usergrp = getgrnam(grp)) == NULL)
+		{
+			if (grp && strlen(grp))
+				syslog(LOG_WARNING,"no such group: %s", grp);
+			gr_gid = userpwd->pw_gid;
+		}
+		else
+			gr_gid = usergrp->gr_gid;
+		if (setgid(gr_gid) == -1)
+		{
+			syslog(LOG_DAEMON,"cannot setgid of user %s: %s", usr, strerror(errno));
+			//	      exit(EXIT_FAILURE);
+		}
+#ifdef _BSD_SOURCE
+		/* init suplementary groups
+		 * (must be done before we change our uid)
+		 */
+		if (initgroups(usr, gr_gid) == -1)
+			syslog(LOG_DAEMON,"Cannot init suplementary groups of user %s: %s", usr, strerror(errno));
+#endif
+		/* set uid */
+		if (setuid(userpwd->pw_uid) == -1)
+		{
+			syslog(LOG_DAEMON,"Cannot change to uid of user %s: %s\n", usr, strerror(errno));
+			//	      exit(EXIT_FAILURE);
+		}
+		if(userpwd->pw_dir)
+			setenv("HOME", userpwd->pw_dir, 1);
+	}
+}
+void *pthr_Heat(void * pV_data)
+{
+	pthread_mutex_lock (&fastmutex_proc);
+	if (heat != nullptr && !heat->statusRun())
+		heat->run();
+	pthread_mutex_unlock(&fastmutex_proc);
+}
+void *pthr_parser(void *pV_data)
+{
+char ch, str[INET_ADDRSTRLEN];
+// socket structure
+struct sockaddr_in client1, server1;
+struct servent *serviceInfo;
+int s1, s;
+socklen_t length;
+	// socket server
+	if (Configure.port > 0)
+		server1.sin_port = htons(Configure.port);
+	else if ((serviceInfo = getservbyname ("heating", "tcp")))
+		server1.sin_port = serviceInfo->s_port;
+	else
+	{
+		syslog(LOG_DAEMON,"Error: No TCP port defined!");
+		exit(EXIT_FAILURE);
+	}
+	server1.sin_family = AF_INET;
+	server1.sin_addr.s_addr = INADDR_ANY;
+	s = socket(AF_INET,SOCK_STREAM,0);
+	if (s < 0)
+	{
+		syslog (LOG_DAEMON, "Error in socket: %s",strerror(errno));
+		exit(EXIT_FAILURE);
+	}
+	if (bind (s, (struct sockaddr *)&server1, sizeof (server1)) < 0)
+	{
+		syslog (LOG_DAEMON, "Error in bind: %s", strerror(errno));
+		exit(EXIT_FAILURE);
+	}
+	if (listen (s, 5) < 0)
+	{
+		syslog (LOG_DAEMON, "Error in listen: %s", strerror(errno));
+		exit(EXIT_FAILURE);
+	}
+	length = sizeof(client1);
+	if (Configure.debug)
+		syslog (LOG_DEBUG, "Server ready: %d",s);
+	// Initialize and start the probe
+	if (heat == nullptr)
+		heat = new heating();
+	if (pthread_create(&pthr_heat, NULL, pthr_Heat, (void *)0) != 0)
+	{
+		syslog (LOG_DAEMON,"Create of thread \"pthr_heat\" failed!");
+		close (s);
+		exit(EXIT_FAILURE);
+	}
+	if (Configure.debug)
+		syslog(LOG_DEBUG, "Heating was initialized and is polling ...");
+	while (heat->statusRun())
+	{
+		if ((s1 = accept (s, (struct sockaddr *)&client1, &length)) < 0)
+		{
+			syslog (LOG_DAEMON, "Error in accept: %d: %s", s1, strerror(errno));
+			continue;
+		}
+		inet_ntop(AF_INET, &(client1.sin_addr), str, INET_ADDRSTRLEN);
+		if (Configure.debug)
+			syslog (LOG_DEBUG, "Connected to client %s", str);
+		soc.ip[0] = 0;
+		soc.sockfd = s;
+		soc.newfd = s1;
+		// This thread will parse the commands comming to this daemon.
+		if (pthread_create(&pthr_process, NULL, processCommands, (void *)&soc) != 0)
+		{
+			syslog (LOG_DAEMON,"Create of thread \"processCommands\" failed!");
+			close(s1);
+			soc.newfd = 0;
+		}
+	}
+	close (s);
+	soc.sockfd = 0;
+	delete heat;
+	heat = nullptr;
+	return NULL;
+}
+void *processCommands(void *pV_data)
+{
+char ch, buf[128];
+int i, s1, s;
+bool initialized = false;
+struct SOCKETS *socket;
+	socket = (struct SOCKETS *)pV_data;
+	s1 = socket->newfd;
+	s = socket->sockfd;
+	heat->setHandle(s1);
+	if (Configure.debug)
+		syslog(LOG_DEBUG, "Starting to process commands ...");
+	memset(&buf[0], 0, sizeof(buf));
+	i = 0;
+	write(s1, "HEATING:READY;", 11);
+	while (heat->statusRun() && read(s1,&ch,1) > 0)
+	{
+		if (i < (int)(sizeof(buf) - 1))
+		{
+			buf[i] = ch;
+			if (ch < 0x0a || ch == 0x0b || ch == 0x0c)
+				continue;
+			if (ch == ';' || ch == 0x0d || ch == 0x0a)
+			{
+				int pstat;
+				buf[i] = 0;
+				if (!strncmp(buf, "quit", 4))
+					break;
+				if ((pstat = parseCommand(s1, buf)) == 0)
+				{
+					char hv0[128];
+					sprintf(&hv0[0],"INVALID COMMAND:%s;",buf);
+					write(s1,hv0,strlen(hv0));
+				}
+				else if (pstat == 2)
+					write(s1, "NAK;", 4);
+				else
+					write(s1,"OK;",3);
+				memset(&buf[0], 0, sizeof(buf));
+				i = 0;
+				continue;
+			}
+		}
+		i++;
+	}
+	heat->removeHandle(s1);
+	close(s1);
+	socket->newfd = 0;
+}
+int parseCommand(int s1, char *buf)
+{
+std::string line(buf);
+char hv0[64];
+std::string vname, vcontent;
+helper help;
+	if (line.find(":") != std::string::npos)
+	{
+		vname = line.substr(0, line.find(":"));
+		vcontent = line.substr(line.find(":") + 1);
+		if (Configure.debug)
+			syslog(LOG_DEBUG, "Command: %s, Content: %s", vname.c_str(), vcontent.c_str());
+	}
+	else
+	{
+		vname = line;
+		vcontent = "";
+		if (Configure.debug)
+			syslog(LOG_DEBUG, "Command only: %s", vname.c_str());
+	}
+	// INCREMENT:<room>;
+	if (vname.compare("INCREMENT") == 0 && vcontent.length() > 0)
+	{
+		int room = atoi(vcontent.c_str());
+		heat->incSoll(room);
+		return 1;
+	}
+	// DECREMENT:<room>;
+	if (vname.compare("DECREMENT") == 0 && vcontent.length() > 0)
+	{
+		int room = atoi(vcontent.c_str());
+		heat->decSoll(room);
+		return 1;
+	}
+	// GET CONFIG;
+	if (vname.compare("GET CONFIG") == 0)
+	{
+		heat->getConfig(s1);
+		return 1;
+	}
+	// POWER:<ON|OFF>;
+	if (vname.compare("POWER") == 0)
+	{
+		if (vcontent.compare("ON") == 0)
+			heat->switchOnOff(true);
+		else
+			heat->switchOnOff(false);
+		return 1;
+	}
+	// TODO: SET SOLL:<room>:<soll>; --> Solltemperatur direkt setzen
+	// TODO: SET NIGHT:<room>:<soll>;
+	// TODO: SET MINIMAL:<room>:<soll>;
+	// TODO: NIGHT:<soll>;
+	// TODO: MINIMAL:<soll>;
+	return 0;
+}

Subversion Repositories heating

(root)/trunk/main.cpp – Rev 3 → 5