$search

dashel-posix.cpp

Go to the documentation of this file.

/*
        Dashel
        A cross-platform DAta Stream Helper Encapsulation Library
        Copyright (C) 2007 -- 2012:
                
                Stephane Magnenat <stephane at magnenat dot net>
                        (http://stephane.magnenat.net)
                Mobots group - Laboratory of Robotics Systems, EPFL, Lausanne
                        (http://mobots.epfl.ch)
                
                Sebastian Gerlach
                Kenzan Technologies
                        (http://www.kenzantech.com)
        
        All rights reserved.
        
        Redistribution and use in source and binary forms, with or without
        modification, are permitted provided that the following conditions are met:
                * Redistributions of source code must retain the above copyright
                  notice, this list of conditions and the following disclaimer.
                * Redistributions in binary form must reproduce the above copyright
                  notice, this list of conditions and the following disclaimer in the
                  documentation and/or other materials provided with the distribution.
                * Neither the names of "Mobots", "Laboratory of Robotics Systems", "EPFL",
                  "Kenzan Technologies" nor the names of the contributors may be used to
                  endorse or promote products derived from this software without specific
                  prior written permission.
        
        THIS SOFTWARE IS PROVIDED BY COPYRIGHT HOLDERS ``AS IS'' AND ANY
        EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
        WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
        DISCLAIMED. IN NO EVENT SHALL COPYRIGHT HOLDERS BE LIABLE FOR ANY
        DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
        (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
        LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
        ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
        (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
        SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/

#include <string.h>
#include <cassert>
#include <cstdlib>
#include <map>
#include <vector>
#include <valarray>
#include <algorithm>
#include <iostream>
#include <sstream>
#include <signal.h>
#include <errno.h>
#include <unistd.h>
#include <termios.h>
#include <fcntl.h>
#include <netdb.h>
#include <signal.h>
#include <sys/select.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <pthread.h>

#ifdef __APPLE__
#define MACOSX
#endif

#ifdef MACOSX
        #define USE_POLL_EMU
#endif

#ifdef MACOSX
        #include <CoreFoundation/CoreFoundation.h>
        #include <IOKit/IOKitLib.h>
        #include <IOKit/serial/IOSerialKeys.h>
#endif

#ifdef USE_LIBUDEV
extern "C" {
        #include <libudev.h>
}
#endif

#ifdef USE_HAL
        #include <hal/libhal.h>
#endif

#ifndef USE_POLL_EMU
        #include <poll.h>
#else
        #include "poll_emu.h"
#endif

#include "dashel-private.h"
#include "dashel-posix.h"



namespace Dashel
{
        using namespace std;
        
        // Exception
        
        void Stream::fail(DashelException::Source s, int se, const char* reason)
        {
                string sysMessage;
                failedFlag = true;
                
                if (se)
                        sysMessage = strerror(errno);
                        
                failReason = reason;
                failReason += " ";
                failReason += sysMessage;
                
                throw DashelException(s, se, failReason.c_str(), this);
        }
        
        // Serial port enumerator
        
        std::map<int, std::pair<std::string, std::string> > SerialPortEnumerator::getPorts()
        {
                std::map<int, std::pair<std::string, std::string> > ports;
                


#ifdef MACOSX
                // use IOKit to enumerates devices
                
                // get a matching dictionary to specify which IOService class we're interested in
                CFMutableDictionaryRef classesToMatch = IOServiceMatching(kIOSerialBSDServiceValue);
                if (classesToMatch == NULL)
                        throw DashelException(DashelException::EnumerationError, 0, "IOServiceMatching returned a NULL dictionary");
                
                // specify all types of serial devices
                CFDictionarySetValue(classesToMatch, CFSTR(kIOSerialBSDTypeKey), CFSTR(kIOSerialBSDAllTypes));
                
                // get an iterator to serial port services
                io_iterator_t matchingServices;
                kern_return_t kernResult = IOServiceGetMatchingServices(kIOMasterPortDefault, classesToMatch, &matchingServices);    
                if (KERN_SUCCESS != kernResult)
                        throw DashelException(DashelException::EnumerationError, kernResult, "IOServiceGetMatchingServices failed");
                        
                // iterate over services
                io_object_t modemService;
                int index = 0;
                while((modemService = IOIteratorNext(matchingServices)))
                {
                        // get path for device
                        CFTypeRef bsdPathAsCFString = IORegistryEntryCreateCFProperty(modemService, CFSTR(kIOCalloutDeviceKey), kCFAllocatorDefault, 0);
                        if (bsdPathAsCFString)
                        {
                                std::string path;
                                char cStr[255];
                                std::string name;
                                
                                bool res = CFStringGetCString((CFStringRef) bsdPathAsCFString, cStr, 255, kCFStringEncodingUTF8);
                                if(res)
                                        path = cStr;
                                else
                                        throw DashelException(DashelException::EnumerationError, 0, "CFStringGetCString failed");
                                
                                CFRelease(bsdPathAsCFString);
                                
                                CFTypeRef fn = IORegistryEntrySearchCFProperty(modemService, kIOServicePlane, CFSTR("USB Product Name"), kCFAllocatorDefault,
                                                                                kIORegistryIterateRecursively | kIORegistryIterateParents);
                                if(fn) {
                                        res = CFStringGetCString((CFStringRef) fn, cStr, 255, kCFStringEncodingUTF8);
                                        if(res) 
                                                name = cStr;
                                        else
                                                throw DashelException(DashelException::EnumerationError, 0, "CFStringGetString failed");

                                        CFRelease(fn);
                                } else 
                                        name = "Serial Port";
                                name = name + " (" + path + ")";
                                ports[index++] = std::make_pair<std::string, std::string>(path, name);
                        }
                        else
                                throw DashelException(DashelException::EnumerationError, 0, "IORegistryEntryCreateCFProperty returned a NULL path");
                        
                        // release service
                        IOObjectRelease(modemService);
                }

                IOObjectRelease(matchingServices);

                        
#elif defined(USE_LIBUDEV)

                struct udev *udev;
                struct udev_enumerate *enumerate;
                struct udev_list_entry *devices, *dev_list_entry;
                struct udev_device *dev;
                int index = 0;

                udev = udev_new();
                if(!udev)
                        throw DashelException(DashelException::EnumerationError, 0, "Cannot create udev context");

                enumerate = udev_enumerate_new(udev);
                udev_enumerate_add_match_subsystem(enumerate, "tty");
                udev_enumerate_scan_devices(enumerate);
                devices = udev_enumerate_get_list_entry(enumerate);

                udev_list_entry_foreach(dev_list_entry, devices) {
                        const char *sysfs_path;
                        struct udev_device *usb_dev;
                        const char * path;
                        struct stat st;
                        unsigned int maj,min;

                        /* Get sysfs path and create the udev device */
                        sysfs_path = udev_list_entry_get_name(dev_list_entry);
                        dev = udev_device_new_from_syspath(udev, sysfs_path);

                        // Some sanity check
                        path = udev_device_get_devnode(dev);
                        if(stat(path, &st)) 
                                throw DashelException(DashelException::EnumerationError, 0, "Cannot stat serial port");
                        
                        if(!S_ISCHR(st.st_mode))
                                throw DashelException(DashelException::EnumerationError, 0, "Serial port is not character device");

                        // Get the major/minor number
                        maj = major(st.st_rdev);
                        min = minor(st.st_rdev);

                        // Ignore all the non physical ports
                        if(!(maj == 2 || (maj == 4 && min < 64) || maj == 3 || maj == 5)) {
                                ostringstream oss;

                                // Check if usb, if yes get the device name
                                usb_dev = udev_device_get_parent_with_subsystem_devtype(dev,"usb","usb_device");
                                if(usb_dev)
                                        oss << udev_device_get_sysattr_value(usb_dev,"product");
                                else
                                        oss << "Serial Port";

                                oss << " (" << path << ")";

                                ports[index++] = std::make_pair<std::string, std::string>(path,oss.str());
                        }
                        udev_device_unref(dev);
                }
                
                udev_enumerate_unref(enumerate);

                udev_unref(udev);

#elif defined(USE_HAL)

                // use HAL to enumerates devices
                DBusConnection* dbusConnection = dbus_bus_get(DBUS_BUS_SYSTEM, 0);
                if (!dbusConnection)
                        throw DashelException(DashelException::EnumerationError, 0, "cannot connect to D-BUS.");
                
                LibHalContext* halContext = libhal_ctx_new();
                if (!halContext)
                        throw DashelException(DashelException::EnumerationError, 0, "cannot create HAL context: cannot create context");
                if (!libhal_ctx_set_dbus_connection(halContext, dbusConnection))
                        throw DashelException(DashelException::EnumerationError, 0, "cannot create HAL context: cannot connect to D-BUS");
                if (!libhal_ctx_init(halContext, 0))
                        throw DashelException(DashelException::EnumerationError, 0, "cannot create HAL context: cannot init context");
                
                int devicesCount;
                char** devices = libhal_find_device_by_capability(halContext, "serial", &devicesCount, 0);
                for (int i = 0; i < devicesCount; i++)
                {
                        char* devFileName = libhal_device_get_property_string(halContext, devices[i], "serial.device", 0);
                        char* info = libhal_device_get_property_string(halContext, devices[i], "info.product", 0);
                        int port = libhal_device_get_property_int(halContext, devices[i], "serial.port", 0);
                        
                        ostringstream oss;
                        oss << info << " " << port;
                        ports[devicesCount - i] = std::make_pair<std::string, std::string>(devFileName, oss.str());
                        
                        libhal_free_string(info);
                        libhal_free_string(devFileName);
                }
                
                libhal_free_string_array(devices);
                libhal_ctx_shutdown(halContext, 0);
                libhal_ctx_free(halContext);
#endif
                
                return ports;
        };
        
        // Asserted dynamic cast
        
        template<typename Derived, typename Base>
        inline Derived polymorphic_downcast(Base base)
        {
                Derived derived = dynamic_cast<Derived>(base);
                assert(derived);
                return derived;
        }
        
        // Streams
        
        #define RECV_BUFFER_SIZE        4096

        SelectableStream::SelectableStream(const string& protocolName) : 
                Stream(protocolName),
                fd(-1),
                writeOnly(false)
        {
                
        }
        
        SelectableStream::~SelectableStream()
        {
                // on POSIX, do not close stdin, stdout, nor stderr
                if (fd >= 3)
                        close(fd);
        }
        
        class DisconnectableStream: public SelectableStream
        {
        protected:
                friend class Hub;
                unsigned char recvBuffer[RECV_BUFFER_SIZE];     
                size_t recvBufferPos; 
                size_t recvBufferSize; 
        
        public:
                DisconnectableStream(const string& protocolName) :
                        Stream(protocolName),
                        SelectableStream(protocolName),
                        recvBufferPos(0),
                        recvBufferSize(0)
                {
                        
                }
                
                virtual bool isDataInRecvBuffer() const { return recvBufferPos != recvBufferSize; }
        };
        
        class SocketStream: public DisconnectableStream
        {
        protected:
                #ifndef TCP_CORK

                enum Consts
                {
                        SEND_BUFFER_SIZE_INITIAL = 256, 
                        SEND_BUFFER_SIZE_LIMIT = 65536 
                };
                
                ExpandableBuffer sendBuffer;
                #endif
                
        public:
                SocketStream(const string& targetName) :
                        Stream("tcp"),
                        DisconnectableStream("tcp")
                        #ifndef TCP_CORK
                        ,sendBuffer(SEND_BUFFER_SIZE_INITIAL)
                        #endif
                {
                        target.add("tcp:host;port;connectionPort=-1;sock=-1");
                        target.add(targetName.c_str());

                        fd = target.get<int>("sock");
                        if (fd < 0)
                        {
                                // create socket
                                fd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
                                if (fd < 0)
                                        throw DashelException(DashelException::ConnectionFailed, errno, "Cannot create socket.");
                                
                                IPV4Address remoteAddress(target.get("host"), target.get<int>("port"));
                                
                                // connect
                                sockaddr_in addr;
                                addr.sin_family = AF_INET;
                                addr.sin_port = htons(remoteAddress.port);
                                addr.sin_addr.s_addr = htonl(remoteAddress.address);
                                if (connect(fd, (struct sockaddr *)&addr, sizeof(addr)) != 0)
                                        throw DashelException(DashelException::ConnectionFailed, errno, "Cannot connect to remote host.");
                                
                                // overwrite target name with a canonical one
                                target.add(remoteAddress.format().c_str());
                                target.erase("connectionPort");
                        }
                        else
                        {
                                // remove file descriptor information from target name
                                target.erase("sock");
                        }
                        
                        // setup TCP Cork for delayed sending
                        #ifdef TCP_CORK
                        int flag = 1;
                        setsockopt(fd, IPPROTO_TCP, TCP_CORK, &flag , sizeof(flag));
                        #endif
                }
                
                virtual ~SocketStream()
                {
                        if (!failed())
                                flush();
                        
                        if (fd >= 0)
                                shutdown(fd, SHUT_RDWR);
                }
                
                virtual void write(const void *data, const size_t size)
                {
                        assert(fd >= 0);
                        
                        if (size == 0)
                                return;
                        
                        #ifdef TCP_CORK
                        send(data, size);
                        #else
                        if (size >= SEND_BUFFER_SIZE_LIMIT)
                        {
                                flush();
                                send(data, size);
                        }
                        else
                        {
                                sendBuffer.add(data, size);
                                if (sendBuffer.size() >= SEND_BUFFER_SIZE_LIMIT)
                                        flush();
                        }
                        #endif
                }
                
                void send(const void *data, size_t size)
                {
                        assert(fd >= 0);
                        
                        unsigned char *ptr = (unsigned char *)data;
                        size_t left = size;
                        
                        while (left)
                        {
#ifdef MACOSX
                                ssize_t len = ::send(fd, ptr, left, 0);
#else
                                ssize_t len = ::send(fd, ptr, left, MSG_NOSIGNAL);
#endif
                                
                                if (len < 0)
                                {
                                        fail(DashelException::IOError, errno, "Socket write I/O error.");
                                }
                                else if (len == 0)
                                {
                                        fail(DashelException::ConnectionLost, 0, "Connection lost.");
                                }
                                else
                                {
                                        ptr += len;
                                        left -= len;
                                }
                        }
                }
                
                virtual void flush()
                {
                        assert(fd >= 0);
                        
                        #ifdef TCP_CORK
                        int flag = 0;
                        setsockopt(fd, IPPROTO_TCP, TCP_CORK, &flag , sizeof(flag));
                        flag = 1;
                        setsockopt(fd, IPPROTO_TCP, TCP_CORK, &flag , sizeof(flag));
                        #else
                        send(sendBuffer.get(), sendBuffer.size());
                        sendBuffer.clear();
                        #endif
                }
                
                virtual void read(void *data, size_t size)
                {
                        assert(fd >= 0);
                        
                        if (size == 0)
                                return;
                        
                        unsigned char *ptr = (unsigned char *)data;
                        size_t left = size;
                        
                        if (isDataInRecvBuffer())
                        {
                                size_t toCopy = std::min(recvBufferSize - recvBufferPos, size);
                                memcpy(ptr, recvBuffer + recvBufferPos, toCopy);
                                recvBufferPos += toCopy;
                                ptr += toCopy;
                                left -= toCopy;
                        }
                        
                        while (left)
                        {
                                ssize_t len = recv(fd, ptr, left, 0);
                                
                                if (len < 0)
                                {
                                        fail(DashelException::IOError, errno, "Socket read I/O error.");
                                }
                                else if (len == 0)
                                {
                                        fail(DashelException::ConnectionLost, 0, "Connection lost.");
                                }
                                else
                                {
                                        ptr += len;
                                        left -= len;
                                }
                        }
                }
                
                virtual bool receiveDataAndCheckDisconnection()
                {
                        assert(recvBufferPos == recvBufferSize);
                        
                        ssize_t len = recv(fd, &recvBuffer, RECV_BUFFER_SIZE, 0);
                        if (len > 0)
                        {
                                recvBufferSize = len;
                                recvBufferPos = 0;
                                return false;
                        }
                        else
                        {
                                if (len < 0)
                                        fail(DashelException::IOError, errno, "Socket read I/O error.");
                                return true;
                        }
                }
        };
        

        class SocketServerStream : public SelectableStream
        {
        public:
                SocketServerStream(const std::string& targetName) :
                        Stream("tcpin"),
                        SelectableStream("tcpin")
                {
                        target.add("tcpin:port=5000;address=0.0.0.0");
                        target.add(targetName.c_str());
                        
                        IPV4Address bindAddress(target.get("address"), target.get<int>("port"));
                        
                        // create socket
                        fd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
                        if (fd < 0)
                                throw DashelException(DashelException::ConnectionFailed, errno, "Cannot create socket.");
                        
                        // reuse address
                        int flag = 1;
                        if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &flag, sizeof (flag)) < 0)
                                throw DashelException(DashelException::ConnectionFailed, errno, "Cannot set address reuse flag on socket, probably the port is already in use.");
                        
                        // bind
                        sockaddr_in addr;
                        addr.sin_family = AF_INET;
                        addr.sin_port = htons(bindAddress.port);
                        addr.sin_addr.s_addr = htonl(bindAddress.address);
                        if (bind(fd, (struct sockaddr *)&addr, sizeof(addr)) != 0)
                                throw DashelException(DashelException::ConnectionFailed, errno, "Cannot bind socket to port, probably the port is already in use.");
                        
                        // Listen on socket, backlog is sort of arbitrary.
                        if(listen(fd, 16) < 0)
                                throw DashelException(DashelException::ConnectionFailed, errno, "Cannot listen on socket.");
                }
                
                virtual void write(const void *data, const size_t size) { }
                virtual void flush() { }
                virtual void read(void *data, size_t size) { }
                virtual bool receiveDataAndCheckDisconnection() { return false; }
                virtual bool isDataInRecvBuffer() const { return false; }
        };
        
        class UDPSocketStream: public MemoryPacketStream, public SelectableStream
        {
        private:
                mutable bool selectWasCalled;
                
        public:
                UDPSocketStream(const string& targetName) :
                        Stream("udp"),
                        MemoryPacketStream("udp"),
                        SelectableStream("udp"),
                        selectWasCalled(false)
                {
                        target.add("udp:port=5000;address=0.0.0.0;sock=-1");
                        target.add(targetName.c_str());
                        
                        fd = target.get<int>("sock");
                        if (fd < 0)
                        {
                                // create socket
                                fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
                                if (fd < 0)
                                        throw DashelException(DashelException::ConnectionFailed, errno, "Cannot create socket.");
                                
                                IPV4Address bindAddress(target.get("address"), target.get<int>("port"));
                                
                                // bind
                                sockaddr_in addr;
                                addr.sin_family = AF_INET;
                                addr.sin_port = htons(bindAddress.port);
                                addr.sin_addr.s_addr = htonl(bindAddress.address);
                                if (bind(fd, (struct sockaddr *)&addr, sizeof(addr)) != 0)
                                        throw DashelException(DashelException::ConnectionFailed, errno, "Cannot bind socket to port, probably the port is already in use.");
                        }
                        else
                        {
                                // remove file descriptor information from target name
                                target.erase("sock");
                        }
                        
                        // enable broadcast
                        int broadcastPermission = 1;
                        setsockopt(fd, SOL_SOCKET, SO_BROADCAST, &broadcastPermission, sizeof(broadcastPermission));
                }
                
                virtual void send(const IPV4Address& dest)
                {
                        sockaddr_in addr;
                        addr.sin_family = AF_INET;
                        addr.sin_port = htons(dest.port);;
                        addr.sin_addr.s_addr = htonl(dest.address);
                        
                        if (sendto(fd, sendBuffer.get(), sendBuffer.size(), 0, (struct sockaddr *)&addr, sizeof(addr)) != sendBuffer.size())
                                fail(DashelException::IOError, errno, "UDP Socket write I/O error.");
                        
                        sendBuffer.clear();
                }
                
                virtual void receive(IPV4Address& source)
                {
                        unsigned char buf[4096];
                        sockaddr_in addr;
                        socklen_t addrLen = sizeof(addr);
                        ssize_t recvCount = recvfrom(fd, buf, 4096, 0, (struct sockaddr *)&addr, &addrLen);
                        if (recvCount <= 0)
                                fail(DashelException::ConnectionLost, errno, "UDP Socket read I/O error.");
                        
                        receptionBuffer.resize(recvCount);
                        std::copy(buf, buf+recvCount, receptionBuffer.begin());
                        
                        source = IPV4Address(ntohl(addr.sin_addr.s_addr), ntohs(addr.sin_port));
                }
                
                virtual bool receiveDataAndCheckDisconnection() { selectWasCalled = true; return false; }
                virtual bool isDataInRecvBuffer() const { bool ret = selectWasCalled; selectWasCalled = false; return ret; }
        };
        
        
        class FileDescriptorStream: public DisconnectableStream
        {
        public:
                FileDescriptorStream(const string& protocolName) :
                        Stream(protocolName),
                        DisconnectableStream(protocolName)
                { }
                
                virtual void write(const void *data, const size_t size)
                {
                        assert(fd >= 0);
                        
                        if (size == 0)
                                return;
                        
                        const char *ptr = (const char *)data;
                        size_t left = size;
                        
                        while (left)
                        {
                                ssize_t len = ::write(fd, ptr, left);
                                
                                if (len < 0)
                                {
                                        fail(DashelException::IOError, errno, "File write I/O error.");
                                }
                                else if (len == 0)
                                {
                                        fail(DashelException::ConnectionLost, 0, "File full.");
                                }
                                else
                                {
                                        ptr += len;
                                        left -= len;
                                }
                        }
                }
                
                virtual void flush()
                {
                        assert(fd >= 0);
                        
#ifdef MACOSX
                        if (fsync(fd) < 0)
#else
                        if (fdatasync(fd) < 0)
#endif
                        {
                                fail(DashelException::IOError, errno, "File flush error.");
                        }
                }
                
                virtual void read(void *data, size_t size)
                {
                        assert(fd >= 0);
                        
                        if (size == 0)
                                return;
                        
                        char *ptr = (char *)data;
                        size_t left = size;
                        
                        if (isDataInRecvBuffer())
                        {
                                size_t toCopy = std::min(recvBufferSize - recvBufferPos, size);
                                memcpy(ptr, recvBuffer + recvBufferPos, toCopy);
                                recvBufferPos += toCopy;
                                ptr += toCopy;
                                left -= toCopy;
                        }
                        
                        while (left)
                        {
                                ssize_t len = ::read(fd, ptr, left);
                                
                                if (len < 0)
                                {
                                        fail(DashelException::IOError, errno, "File read I/O error.");
                                }
                                else if (len == 0)
                                {
                                        fail(DashelException::ConnectionLost, 0, "Reached end of file.");
                                }
                                else
                                {
                                        ptr += len;
                                        left -= len;
                                }
                        }
                }
                
                virtual bool receiveDataAndCheckDisconnection()
                {
                        assert(recvBufferPos == recvBufferSize);
                        
                        ssize_t len = ::read(fd, &recvBuffer, RECV_BUFFER_SIZE);
                        if (len > 0)
                        {
                                recvBufferSize = len;
                                recvBufferPos = 0;
                                return false;
                        }
                        else
                        {
                                if (len < 0)
                                        fail(DashelException::IOError, errno, "File read I/O error.");
                                return true;
                        }
                }
        };
        
        class FileStream: public FileDescriptorStream
        {
        public:
                FileStream(const string& targetName) :
                        Stream("file"),
                        FileDescriptorStream("file")
                {
                        target.add("file:name;mode=read;fd=-1");
                        target.add(targetName.c_str());
                        fd = target.get<int>("fd");
                        if (fd < 0)
                        {
                                const std::string name = target.get("name");
                                const std::string mode = target.get("mode");
                                
                                // open file
                                if (mode == "read")
                                        fd = open(name.c_str(), O_RDONLY);
                                else if (mode == "write")
                                        fd = creat(name.c_str(), S_IRUSR|S_IWUSR|S_IRGRP|S_IWGRP), writeOnly = true;
                                else if (mode == "readwrite")
                                        fd = open(name.c_str(), O_RDWR);
                                else
                                        throw DashelException(DashelException::InvalidTarget, 0, "Invalid file mode.");
                                if (fd == -1)
                                {
                                        string errorMessage = "Cannot open file " + name + " for " + mode + ".";
                                        throw DashelException(DashelException::ConnectionFailed, errno, errorMessage.c_str());
                                }
                        }
                        else
                        {
                                // remove file descriptor information from target name
                                target.erase("fd");
                        }
                }
        };
        
        struct StdinStream: public FileStream
        {
                StdinStream(const string& targetName):
                        Stream("file"),
                        FileStream("file:name=/dev/stdin;mode=read;fd=0") {}
        };
        
        struct StdoutStream: public FileStream
        {
                StdoutStream(const string& targetName):
                        Stream("file"),
                        FileStream("file:name=/dev/stdout;mode=write;fd=1") {}
        };
        
        class SerialStream: public FileDescriptorStream
        {
        protected:
                struct termios oldtio;  
                
        public:
                SerialStream(const string& targetName) :
                        Stream("ser"),
                        FileDescriptorStream("ser")
                {
                        target.add("ser:port=1;baud=115200;stop=1;parity=none;fc=none;bits=8");
                        target.add(targetName.c_str());
                        string devFileName;
                        
                        if (target.isSet("device"))
                        {
                                target.addParam("device", NULL, true);
                                target.erase("port");
                                target.erase("name");
                                
                                devFileName = target.get("device");
                        }
                        else if (target.isSet("name"))
                        {
                                target.addParam("name", NULL, true);
                                target.erase("port");
                                target.erase("device");

                                // Enumerates the ports
                                std::string name = target.get("name");
                                std::map<int, std::pair<std::string, std:: string> > ports = SerialPortEnumerator::getPorts();

                                // Iterate on all ports to found one with "name" in its description
                                std::map<int, std::pair<std::string, std:: string> >::iterator it;
                                for (it = ports.begin(); it != ports.end(); it++)
                                {
                                        if (it->second.second.find(name) != std::string::npos)
                                        {
                                                devFileName = it->second.first;
                                                std::cout << "Found " << name << " on port " << devFileName << std::endl;
                                                break;
                                        }
                                }
                                if (devFileName.size() == 0)
                                        throw DashelException(DashelException::ConnectionFailed, 0, "The specified name could not be find among the serial ports.");
                        }
                        else // port
                        {
                                target.erase("device");
                                target.erase("name");
                                
                                std::map<int, std::pair<std::string, std::string> > ports = SerialPortEnumerator::getPorts();
                                std::map<int, std::pair<std::string, std::string> >::const_iterator it = ports.find(target.get<int>("port"));
                                if (it != ports.end())
                                {
                                        devFileName = it->first;
                                }
                                else
                                        throw DashelException(DashelException::ConnectionFailed, 0, "The specified serial port does not exists.");
                        }
                
                        fd = open(devFileName.c_str(), O_RDWR);
                        
                        if (fd == -1)
                                throw DashelException(DashelException::ConnectionFailed, 0, "Cannot open serial port.");
                        
                        struct termios newtio;
                        
                        // save old serial port state and clear new one
                        tcgetattr(fd, &oldtio);
                        memset(&newtio, 0, sizeof(newtio));
                        
                        newtio.c_cflag |= CLOCAL;                       // ignore modem control lines.
                        newtio.c_cflag |= CREAD;                        // enable receiver.
                        switch (target.get<int>("bits"))                // Set amount of bits per character
                        {
                                case 5: newtio.c_cflag |= CS5; break;
                                case 6: newtio.c_cflag |= CS6; break;
                                case 7: newtio.c_cflag |= CS7; break;
                                case 8: newtio.c_cflag |= CS8; break;
                                default: throw DashelException(DashelException::InvalidTarget, 0, "Invalid number of bits per character, must be 5, 6, 7, or 8.");
                        }
                        if (target.get("stop") == "2")
                                newtio.c_cflag |= CSTOPB;               // Set two stop bits, rather than one.
                        if (target.get("fc") == "hard")
                                newtio.c_cflag |= CRTSCTS;              // enable hardware flow control
                        if (target.get("parity") != "none")
                        {
                                newtio.c_cflag |= PARENB;               // enable parity generation on output and parity checking for input.
                                if (target.get("parity") == "odd")
                                        newtio.c_cflag |= PARODD;       // parity for input and output is odd.
                        }
                        
#ifdef MACOSX
                        if (cfsetspeed(&newtio,target.get<int>("baud")) != 0)
                                throw DashelException(DashelException::ConnectionFailed, errno, "Invalid baud rate.");
#else
                        switch (target.get<int>("baud"))
                        {
                                case 50: newtio.c_cflag |= B50; break;
                                case 75: newtio.c_cflag |= B75; break;
                                case 110: newtio.c_cflag |= B110; break;
                                case 134: newtio.c_cflag |= B134; break;
                                case 150: newtio.c_cflag |= B150; break;
                                case 200: newtio.c_cflag |= B200; break;
                                case 300: newtio.c_cflag |= B300; break;
                                case 600: newtio.c_cflag |= B600; break;
                                case 1200: newtio.c_cflag |= B1200; break;
                                case 1800: newtio.c_cflag |= B1800; break;
                                case 2400: newtio.c_cflag |= B2400; break;
                                case 4800: newtio.c_cflag |= B4800; break;
                                case 9600: newtio.c_cflag |= B9600; break;
                                case 19200: newtio.c_cflag |= B19200; break;
                                case 38400: newtio.c_cflag |= B38400; break;
                                case 57600: newtio.c_cflag |= B57600; break;
                                case 115200: newtio.c_cflag |= B115200; break;
                                case 230400: newtio.c_cflag |= B230400; break;
                                case 460800: newtio.c_cflag |= B460800; break;
                                case 500000: newtio.c_cflag |= B500000; break;
                                case 576000: newtio.c_cflag |= B576000; break;
                                case 921600: newtio.c_cflag |= B921600; break;
                                case 1000000: newtio.c_cflag |= B1000000; break;
                                case 1152000: newtio.c_cflag |= B1152000; break;
                                case 1500000: newtio.c_cflag |= B1500000; break;
                                case 2000000: newtio.c_cflag |= B2000000; break;
                                case 2500000: newtio.c_cflag |= B2500000; break;
                                case 3000000: newtio.c_cflag |= B3000000; break;
                                case 3500000: newtio.c_cflag |= B3500000; break;
                                case 4000000: newtio.c_cflag |= B4000000; break;
                                default: throw DashelException(DashelException::ConnectionFailed, 0, "Invalid baud rate.");
                        }
#endif
                        
                        newtio.c_iflag = IGNPAR;                        // ignore parity on input
                        
                        newtio.c_oflag = 0;
                        
                        newtio.c_lflag = 0;
                        
                        newtio.c_cc[VTIME] = 0;                         // block forever if no byte
                        newtio.c_cc[VMIN] = 1;                          // one byte is sufficient to return
                        
                        // set attributes
                        if ((tcflush(fd, TCIOFLUSH) < 0) || (tcsetattr(fd, TCSANOW, &newtio) < 0))
                                throw DashelException(DashelException::ConnectionFailed, 0, "Cannot setup serial port. The requested baud rate might not be supported.");
                }
                
                virtual ~SerialStream()
                {
                         tcsetattr(fd, TCSANOW, &oldtio);
                }
                
                virtual void flush()
                {
                }
        };
        
        
        /*
        We have decided to let the application choose what to do with signals.
        Hub::stop() being thread safe, it is ok to let the user decide.
        
        // Signal handler for SIGTERM
        
        typedef std::set<Hub*> HubsSet;
        typedef HubsSet::iterator HubsSetIterator;
        
        static HubsSet allHubs;
        
        void termHandler(int t)
        {
                for (HubsSetIterator it = allHubs.begin(); it != allHubs.end(); ++it)
                {
                        (*it)->stop();
                }
        }
        
        
        static class SigTermHandlerSetuper
        {
        public:
                SigTermHandlerSetuper()
                {
                        struct sigaction new_act, old_act;
                        new_act.sa_handler = termHandler;
                        sigemptyset(&new_act.sa_mask);

                        new_act.sa_flags = 0;

                        sigaction(SIGTERM, &new_act, &old_act);
                        sigaction(SIGINT, &new_act, &old_act);
                }
        } staticSigTermHandlerSetuper;
        */
        
        // Hub
        
        Hub::Hub(const bool resolveIncomingNames):
                resolveIncomingNames(resolveIncomingNames)
        {
                int *terminationPipes = new int[2];
                if (pipe(terminationPipes) != 0)
                        abort();
                hTerminate = terminationPipes;
                
                streamsLock = new pthread_mutex_t;
                
                pthread_mutex_init((pthread_mutex_t*)streamsLock, NULL);

                // commented because we let the users manage the signal themselves
                //allHubs.insert(this);
        }
        
        Hub::~Hub()
        {
                // commented because we let the users manage the signal themselves
                //allHubs.erase(this);
                
                int *terminationPipes = (int*)hTerminate;
                close(terminationPipes[0]);
                close(terminationPipes[1]);
                delete[] terminationPipes;
                
                for (StreamsSet::iterator it = streams.begin(); it != streams.end(); ++it)
                        delete *it;
                
                pthread_mutex_destroy((pthread_mutex_t*)streamsLock);
                
                delete (pthread_mutex_t*) streamsLock;
        }
        
        Stream* Hub::connect(const std::string &target)
        {
                std::string proto, params;
                size_t c = target.find_first_of(':');
                if (c == std::string::npos)
                        throw DashelException(DashelException::InvalidTarget, 0, "No protocol specified in target.");
                proto = target.substr(0, c);
                params = target.substr(c+1);
                
                SelectableStream *s(dynamic_cast<SelectableStream*>(streamTypeRegistry.create(proto, target, *this)));
                if(!s)
                {
                        std::string r = "Invalid protocol in target: ";
                        r += proto;
                        r += ", known protocol are: ";
                        r += streamTypeRegistry.list();
                        throw DashelException(DashelException::InvalidTarget, 0, r.c_str());
                }
                
                /* The caller must have the stream lock held */
                
                streams.insert(s);
                if (proto != "tcpin")
                {
                        dataStreams.insert(s);
                        connectionCreated(s);
                }
                
                return s;
        }
        
        void Hub::run()
        {
                while (step(-1));
        }
        
        bool Hub::step(const int timeout)
        {
                bool firstPoll = true;
                bool wasActivity = false;
                bool runInterrupted = false;
                
                pthread_mutex_lock((pthread_mutex_t*)streamsLock);
                
                do
                {
                        wasActivity = false;
                        size_t streamsCount = streams.size();
                        valarray<struct pollfd> pollFdsArray(streamsCount+1);
                        valarray<SelectableStream*> streamsArray(streamsCount);
                        
                        // add streams
                        size_t i = 0;
                        for (StreamsSet::iterator it = streams.begin(); it != streams.end(); ++it)
                        {
                                SelectableStream* stream = polymorphic_downcast<SelectableStream*>(*it);
                                
                                streamsArray[i] = stream;
                                pollFdsArray[i].fd = stream->fd;
                                pollFdsArray[i].events = 0;
                                if ((!stream->failed()) && (!stream->writeOnly))
                                        pollFdsArray[i].events |= POLLIN;
                                
                                i++;
                        }
                        // add pipe
                        int *terminationPipes = (int*)hTerminate;
                        pollFdsArray[i].fd = terminationPipes[0];
                        pollFdsArray[i].events = POLLIN;
                        
                        // do poll and check for error
                        int thisPollTimeout = firstPoll ? timeout : 0;
                        firstPoll = false;
                        
                        pthread_mutex_unlock((pthread_mutex_t*)streamsLock);
                        
                        #ifndef USE_POLL_EMU
                        int ret = poll(&pollFdsArray[0], pollFdsArray.size(), thisPollTimeout);
                        #else
                        int ret = poll_emu(&pollFdsArray[0], pollFdsArray.size(), thisPollTimeout);
                        #endif
                        if (ret < 0)
                                throw DashelException(DashelException::SyncError, errno, "Error during poll.");
                        
                        pthread_mutex_lock((pthread_mutex_t*)streamsLock);
                        
                        // check streams for errors
                        for (i = 0; i < streamsCount; i++)
                        {
                                SelectableStream* stream = streamsArray[i];
                                
                                // make sure we do not try to handle removed streams
                                if (streams.find(stream) == streams.end())
                                        continue;
                                
                                assert((pollFdsArray[i].revents & POLLNVAL) == 0);
                                
                                if (pollFdsArray[i].revents & POLLERR)
                                {
                                        //std::cerr << "POLLERR" << std::endl;
                                        wasActivity = true;
                                        
                                        try
                                        {
                                                stream->fail(DashelException::SyncError, 0, "Error on stream during poll.");
                                        }
                                        catch (DashelException e)
                                        {
                                                assert(e.stream);
                                        }
                                        
                                        try
                                        {
                                                connectionClosed(stream, true);
                                        }
                                        catch (DashelException e)
                                        {
                                                assert(e.stream);
                                        }
                                        
                                        closeStream(stream);
                                }
                                else if (pollFdsArray[i].revents & POLLHUP)
                                {
                                        //std::cerr << "POLLHUP" << std::endl;
                                        wasActivity = true;
                                        
                                        try
                                        {
                                                connectionClosed(stream, false);
                                        }
                                        catch (DashelException e)
                                        {
                                                assert(e.stream);
                                        }
                                        
                                        closeStream(stream);
                                }
                                else if (pollFdsArray[i].revents & POLLIN)
                                {
                                        //std::cerr << "POLLIN" << std::endl;
                                        wasActivity = true;
                                        
                                        // test if listen stream
                                        SocketServerStream* serverStream = dynamic_cast<SocketServerStream*>(stream);
                                        
                                        if (serverStream)
                                        {       
                                                // accept connection
                                                struct sockaddr_in targetAddr;
                                                socklen_t l = sizeof (targetAddr);
                                                int targetFD = accept (stream->fd, (struct sockaddr *)&targetAddr, &l);
                                                if (targetFD < 0)
                                                {
                                                        pthread_mutex_unlock((pthread_mutex_t*)streamsLock);
                                                        throw DashelException(DashelException::SyncError, errno, "Cannot accept new stream.");
                                                }
                                                
                                                // create a target stream using the new file descriptor from accept
                                                ostringstream targetName;
                                                targetName << IPV4Address(ntohl(targetAddr.sin_addr.s_addr), ntohs(targetAddr.sin_port)).format(resolveIncomingNames);
                                                targetName << ";connectionPort=";
                                                targetName << atoi(serverStream->getTargetParameter("port").c_str());
                                                targetName << ";sock=";
                                                targetName << targetFD;
                                                connect(targetName.str());
                                        }
                                        else
                                        {
                                                bool streamClosed = false;
                                                try
                                                {
                                                        if (stream->receiveDataAndCheckDisconnection())
                                                        {
                                                                //std::cerr << "connection closed" << std::endl;
                                                                connectionClosed(stream, false);
                                                                streamClosed = true;
                                                        }
                                                        else
                                                        {
                                                                // read all data available on this socket
                                                                while (stream->isDataInRecvBuffer())
                                                                        incomingData(stream);
                                                                //std::cerr << "incoming data" << std::endl;
                                                        }
                                                }
                                                catch (DashelException e)
                                                {
                                                        //std::cerr << "exception on POLLIN" << std::endl;
                                                        assert(e.stream);
                                                }
                                                
                                                if (streamClosed)
                                                        closeStream(stream);
                                        }
                                }
                        }
                        // check pipe for termination
                        if (pollFdsArray[i].revents)
                        {
                                char c;
                                const ssize_t ret = read(pollFdsArray[i].fd, &c, 1);
                                if (ret != 1)
                                        abort(); // poll did notify us that there was something to read, but we did not read anything, this is a bug
                                runInterrupted = true;
                        }
                        
                        // collect and remove all failed streams
                        vector<Stream*> failedStreams;
                        for (StreamsSet::iterator it = streams.begin(); it != streams.end();++it)
                                if ((*it)->failed())
                                        failedStreams.push_back(*it);
                        
                        for (size_t i = 0; i < failedStreams.size(); i++)
                        {
                                Stream* stream = failedStreams[i];
                                if (streams.find(stream) == streams.end())
                                        continue;
                                if (stream->failed())
                                {
                                        try
                                        {
                                                connectionClosed(stream, true);
                                        }
                                        catch (DashelException e)
                                        {
                                                assert(e.stream);
                                        }
                                        closeStream(stream);
                                }
                        }
                }
                while (wasActivity && !runInterrupted);
                
                pthread_mutex_unlock((pthread_mutex_t*)streamsLock);
                
                return !runInterrupted;
        }
        
        void Hub::lock()
        {
                pthread_mutex_lock((pthread_mutex_t*)streamsLock);
        }
        
        void Hub::unlock()
        {
                pthread_mutex_unlock((pthread_mutex_t*)streamsLock);
        }
        
        void Hub::stop()
        {
                int *terminationPipes = (int*)hTerminate;
                char c = 0;
                const ssize_t ret = write(terminationPipes[1], &c, 1);
                if (ret != 1)
                        throw DashelException(DashelException::IOError, ret, "Cannot write to termination pipe.");
        }
        
        StreamTypeRegistry::StreamTypeRegistry()
        {
                reg("file", &createInstance<FileStream>);
                reg("stdin", &createInstance<StdinStream>);
                reg("stdout", &createInstance<StdoutStream>);
                reg("ser", &createInstance<SerialStream>);
                reg("tcpin", &createInstance<SocketServerStream>);
                reg("tcp", &createInstance<SocketStream>);
                reg("udp", &createInstance<UDPSocketStream>);
        }
        
        StreamTypeRegistry __attribute__((init_priority(1000))) streamTypeRegistry;
}

---

dashel
Author(s): Stéphane Magnenat
autogenerated on Sat Mar 2 12:31:28 2013