/* This file is part of Ingen. Copyright 2007-2015 David Robillard Ingen is free software: you can redistribute it and/or modify it under the terms of the GNU Affero General Public License as published by the Free Software Foundation, either version 3 of the License, or any later version. Ingen is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for details. You should have received a copy of the GNU Affero General Public License along with Ingen. If not, see . */ #include #include #include #include #include "ingen/Configuration.hpp" #include "ingen/Module.hpp" #include "ingen/World.hpp" #include "raul/Socket.hpp" #include "../server/Engine.hpp" #include "../server/EventWriter.hpp" #include "SocketListener.hpp" #include "SocketServer.hpp" namespace Ingen { namespace Server { void SocketListener::ingen_listen(Engine* engine, Raul::Socket* unix_sock, Raul::Socket* net_sock) { Ingen::World* world = engine->world(); const std::string unix_path(world->conf().option("socket").ptr()); // Bind UNIX socket const Raul::URI unix_uri(unix_scheme + unix_path); if (!unix_sock->bind(unix_uri) || !unix_sock->listen()) { world->log().error("Failed to create UNIX socket\n"); unix_sock->close(); } else { world->log().info(fmt("Listening on socket %1%\n") % unix_uri); } // Bind TCP socket const int port = world->conf().option("engine-port").get(); std::ostringstream ss; ss << "tcp://localhost:"; ss << port; if (!net_sock->bind(Raul::URI(ss.str())) || !net_sock->listen()) { world->log().error("Failed to create TCP socket\n"); net_sock->close(); } else { world->log().info(fmt("Listening on TCP port %1%\n") % port); } if (unix_sock->fd() == -1 && net_sock->fd() == -1) { return; // No sockets to listen to, exit thread } struct pollfd pfds[2]; int nfds = 0; if (unix_sock->fd() != -1) { pfds[nfds].fd = unix_sock->fd(); pfds[nfds].events = POLLIN; pfds[nfds].revents = 0; ++nfds; } if (net_sock->fd() != -1) { pfds[nfds].fd = net_sock->fd(); pfds[nfds].events = POLLIN; pfds[nfds].revents = 0; ++nfds; } while (true) { // Wait for input to arrive at a socket const int ret = poll(pfds, nfds, -1); if (ret == -1) { world->log().error(fmt("Poll error: %1%\n") % strerror(errno)); break; } else if ((pfds[0].revents & POLLHUP) || pfds[1].revents & POLLHUP) { break; } else if (ret == 0) { world->log().error("Poll returned with no data\n"); continue; } if (pfds[0].revents & POLLIN) { SPtr conn = unix_sock->accept(); if (conn) { new SocketServer(*world, *engine, conn); } } if (pfds[1].revents & POLLIN) { SPtr conn = net_sock->accept(); if (conn) { new SocketServer(*world, *engine, conn); } } } } } // namespace Server } // namespace Ingen