haiku/src/servers/net/Services.cpp
Axel Dörfler 16aceec16e Put the new server into its own session.
git-svn-id: file:///srv/svn/repos/haiku/haiku/trunk@19664 a95241bf-73f2-0310-859d-f6bbb57e9c96
2006-12-31 16:33:21 +00:00

534 lines
11 KiB
C++

/*
* Copyright 2006, Haiku, Inc. All Rights Reserved.
* Distributed under the terms of the MIT License.
*
* Authors:
* Axel Dörfler, axeld@pinc-software.de
*/
#include "Services.h"
#include "NetServer.h"
#include "Settings.h"
#include <Autolock.h>
#include <errno.h>
#include <netdb.h>
#include <netinet/in.h>
#include <new>
#include <sys/ioctl.h>
#include <sys/socket.h>
#include <vector>
using namespace std;
struct service_address {
struct service *owner;
int socket;
int family;
int type;
int protocol;
sockaddr address;
};
typedef vector<service_address> AddressList;
struct service {
std::string name;
std::string launch;
uid_t user;
gid_t group;
AddressList addresses;
~service();
bool operator!=(const struct service& other);
bool operator==(const struct service& other);
};
int
parse_type(const char* string)
{
if (!strcasecmp(string, "stream"))
return SOCK_STREAM;
return SOCK_DGRAM;
}
int
parse_protocol(const char* string)
{
struct protoent* proto = getprotobyname(string);
if (proto == NULL)
return IPPROTO_TCP;
return proto->p_proto;
}
int
type_for_protocol(int protocol)
{
// default determined by protocol
switch (protocol) {
case IPPROTO_TCP:
return SOCK_STREAM;
case IPPROTO_UDP:
default:
return SOCK_DGRAM;
}
}
// #pragma mark -
service::~service()
{
// close all open sockets
AddressList::const_iterator iterator = addresses.begin();
for (; iterator != addresses.end(); iterator++) {
const service_address& address = *iterator;
close(address.socket);
}
}
bool
service::operator!=(const struct service& other)
{
return !(*this == other);
}
bool
service::operator==(const struct service& other)
{
if (name != other.name
|| launch != other.launch)
return false;
// TODO: compare addresses!
return true;
}
// #pragma mark -
Services::Services(const BMessage& services)
:
fListener(-1),
fUpdate(0),
fMaxSocket(0)
{
_Update(services);
// setup pipe to communicate with the listener thread - as the listener
// blocks on select(), we need a mechanism to interrupt it
if (pipe(&fReadPipe) < 0) {
fReadPipe = -1;
return;
}
fcntl(fReadPipe, F_SETFD, FD_CLOEXEC);
fcntl(fWritePipe, F_SETFD, FD_CLOEXEC);
FD_ZERO(&fSet);
FD_SET(fReadPipe, &fSet);
fMinSocket = fWritePipe + 1;
fMaxSocket = fWritePipe + 1;
fListener = spawn_thread(_Listener, "services listener", B_NORMAL_PRIORITY, this);
if (fListener >= B_OK)
resume_thread(fListener);
}
Services::~Services()
{
wait_for_thread(fListener, NULL);
}
status_t
Services::InitCheck() const
{
return fListener >= B_OK ? B_OK : fListener;
}
void
Services::MessageReceived(BMessage* message)
{
switch (message->what) {
case kMsgUpdateServices:
_Update(*message);
break;
default:
BHandler::MessageReceived(message);
}
}
void
Services::_NotifyListener(bool quit)
{
write(fWritePipe, quit ? "q" : "u", 1);
}
void
Services::_UpdateMinMaxSocket(int socket)
{
if (socket >= fMaxSocket)
fMaxSocket = socket + 1;
if (socket < fMinSocket)
fMinSocket = socket;
}
status_t
Services::_StartService(struct service& service)
{
// create socket
bool failed = false;
AddressList::iterator iterator = service.addresses.begin();
for (; iterator != service.addresses.end(); iterator++) {
service_address& address = *iterator;
address.socket = socket(address.family, address.type, address.protocol);
if (address.socket < 0
|| bind(address.socket, &address.address, address.address.sa_len) < 0
|| fcntl(address.socket, F_SETFD, FD_CLOEXEC) < 0) {
failed = true;
break;
}
if (address.type == SOCK_STREAM && listen(address.socket, 50) < 0) {
failed = true;
break;
}
}
if (failed) {
// open sockets will be closed when the service is deleted
return errno;
}
// add service to maps and activate it
fNameMap[service.name] = &service;
iterator = service.addresses.begin();
for (; iterator != service.addresses.end(); iterator++) {
service_address& address = *iterator;
fSocketMap[address.socket] = &address;
_UpdateMinMaxSocket(address.socket);
FD_SET(address.socket, &fSet);
}
_NotifyListener();
printf("Starting service '%s'\n", service.name.c_str());
return B_OK;
}
status_t
Services::_StopService(struct service& service)
{
// remove service from maps
{
ServiceNameMap::iterator iterator = fNameMap.find(service.name);
if (iterator != fNameMap.end())
fNameMap.erase(iterator);
}
{
AddressList::const_iterator iterator = service.addresses.begin();
for (; iterator != service.addresses.end(); iterator++) {
const service_address& address = *iterator;
ServiceSocketMap::iterator socketIterator = fSocketMap.find(address.socket);
if (socketIterator != fSocketMap.end())
fSocketMap.erase(socketIterator);
close(address.socket);
FD_CLR(address.socket, &fSet);
}
}
delete &service;
return B_OK;
}
status_t
Services::_ToService(const BMessage& message, struct service*& service)
{
// get mandatory fields
const char* name;
const char* launch;
if (message.FindString("name", &name) != B_OK
|| message.FindString("launch", &launch) != B_OK)
return B_BAD_VALUE;
service = new (std::nothrow) ::service;
if (service == NULL)
return B_NO_MEMORY;
service->name = name;
service->launch = launch;
// TODO: user/group is currently ignored!
// Default family/port/protocol/type for all addresses
// we default to inet/tcp/port-from-service-name if nothing is specified
const char* string;
int32 serviceFamilyIndex;
int32 serviceFamily = -1;
if (message.FindString("family", &string) != B_OK)
string = "inet";
if (get_family_index(string, serviceFamilyIndex))
serviceFamily = family_at_index(serviceFamilyIndex);
int32 serviceProtocol;
if (message.FindString("protocol", &string) == B_OK)
serviceProtocol = parse_protocol(string);
else {
string = "tcp";
// we set 'string' here for an eventual call to getservbyname() below
serviceProtocol = IPPROTO_TCP;
}
int32 servicePort;
if (message.FindInt32("port", &servicePort) != B_OK) {
struct servent* servent = getservbyname(name, string);
if (servent != NULL)
servicePort = servent->s_port;
else
servicePort = -1;
}
int32 serviceType = -1;
if (message.FindString("type", &string) == B_OK) {
serviceType = parse_type(string);
} else {
serviceType = type_for_protocol(serviceProtocol);
}
BMessage address;
int32 i = 0;
for (; message.FindMessage("address", i, &address) == B_OK; i++) {
// TODO: dump problems in the settings to syslog
service_address serviceAddress;
if (address.FindString("family", &string) != B_OK)
continue;
int32 familyIndex;
if (!get_family_index(string, familyIndex))
continue;
serviceAddress.family = family_at_index(familyIndex);
if (address.FindString("protocol", &string) == B_OK)
serviceAddress.protocol = parse_protocol(string);
else
serviceAddress.protocol = serviceProtocol;
if (message.FindString("type", &string) == B_OK)
serviceAddress.type = parse_type(string);
else if (serviceAddress.protocol != serviceProtocol)
serviceAddress.type = type_for_protocol(serviceAddress.protocol);
else
serviceAddress.type = serviceType;
if (address.FindString("address", &string) == B_OK) {
if (!parse_address(familyIndex, string, serviceAddress.address))
continue;
} else
set_any_address(familyIndex, serviceAddress.address);
int32 port;
if (address.FindInt32("port", &port) != B_OK)
port = servicePort;
set_port(familyIndex, serviceAddress.address, port);
serviceAddress.socket = -1;
serviceAddress.owner = service;
service->addresses.push_back(serviceAddress);
}
if (i == 0 && (serviceFamily < 0 || servicePort < 0)) {
// no address specified
delete service;
return B_BAD_VALUE;
}
if (i == 0) {
// no address specified, but family/port were given; add empty address
service_address serviceAddress;
serviceAddress.family = serviceFamily;
serviceAddress.type = serviceType;
serviceAddress.protocol = serviceProtocol;
set_any_address(serviceFamilyIndex, serviceAddress.address);
set_port(serviceFamilyIndex, serviceAddress.address, servicePort);
serviceAddress.socket = -1;
serviceAddress.owner = service;
service->addresses.push_back(serviceAddress);
}
return B_OK;
}
void
Services::_Update(const BMessage& services)
{
BAutolock locker(fLock);
fUpdate++;
BMessage message;
for (int32 index = 0; services.FindMessage("service", index,
&message) == B_OK; index++) {
const char* name;
if (message.FindString("name", &name) != B_OK)
continue;
struct service* service;
if (_ToService(message, service) != B_OK)
continue;
ServiceNameMap::iterator iterator = fNameMap.find(name);
if (iterator == fNameMap.end()) {
// this service does not exist yet, start it
_StartService(*service);
} else {
// this service does already exist - check for any changes
if (service != iterator->second) {
_StopService(*iterator->second);
_StartService(*service);
}
}
}
}
status_t
Services::_LaunchService(struct service& service, int socket)
{
printf("LAUNCH: %s\n", service.launch.c_str());
if (fcntl(socket, F_SETFD, 0) < 0) {
// could not clear FD_CLOEXEC on socket
return errno;
}
pid_t child = fork();
if (child == 0) {
setsid();
// make sure we're in our own session, and don't accidently quit the net_server
// We're the child, replace standard input/output
dup2(socket, STDIN_FILENO);
dup2(socket, STDOUT_FILENO);
dup2(socket, STDERR_FILENO);
close(socket);
// build argument array
const char** args = (const char**)malloc(2 * sizeof(void *));
if (args == NULL)
exit(1);
args[0] = service.launch.c_str();
args[1] = NULL;
if (execv(service.launch.c_str(), (char* const*)args) < 0)
exit(1);
// we'll never trespass here
}
// TODO: make sure child started successfully...
return B_OK;
}
status_t
Services::_Listener()
{
while (true) {
fLock.Lock();
fd_set set = fSet;
fLock.Unlock();
if (select(fMaxSocket, &set, NULL, NULL, NULL) < 0) {
// sleep a bit before trying again
snooze(1000000LL);
}
printf("select returned!\n");
if (FD_ISSET(fReadPipe, &set)) {
char command;
if (read(fReadPipe, &command, 1) == 1 && command == 'q')
break;
}
BAutolock locker(fLock);
for (int i = fMinSocket; i < fMaxSocket; i++) {
if (!FD_ISSET(i, &set))
continue;
ServiceSocketMap::iterator iterator = fSocketMap.find(i);
if (iterator == fSocketMap.end())
continue;
struct service_address& address = *iterator->second;
int socket;
if (address.type == SOCK_STREAM) {
// accept incoming connection
int value = 1;
ioctl(i, FIONBIO, &value);
// make sure we don't wait for the connection
socket = accept(address.socket, NULL, NULL);
value = 0;
ioctl(i, FIONBIO, &value);
if (socket < 0)
continue;
} else
socket = address.socket;
// launch this service's handler
_LaunchService(*address.owner, socket);
}
}
return B_OK;
}
/*static*/ status_t
Services::_Listener(void* _self)
{
Services* self = (Services*)_self;
return self->_Listener();
}