libs/midi++2/ipmidi_port.cc

   1 /*
   2     Copyright (C) 2012 Paul Davis
   3
   4     Using code from Rui Nuno Capela's qmidinet as inspiration.
   5
   6     This program is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     This program is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with this program; if not, write to the Free Software
  18     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  19
  20     $Id: port.cc 12065 2012-04-23 16:23:48Z paul $
  21 */
  22 #include <iostream>
  23 #include <cstdio>
  24 #include <fcntl.h>
  25 #include <errno.h>
  26 #include <stdlib.h>
  27 #include <string.h>
  28 #ifdef   COMPILER_MSVC
  29 #undef   O_NONBLOCK
  30 #define  O_NONBLOCK 0
  31 #endif
  32 #if defined(PLATFORM_WINDOWS)
  33 #include <winsock2.h>
  34 #else
  35 #include <netdb.h>
  36 #endif
  37
  38 #if defined(PLATFORM_WINDOWS)
  39 static WSADATA g_wsaData;
  40 typedef int socklen_t;
  41 #else
  42 #include <unistd.h>
  43 #include <sys/ioctl.h>
  44 inline void closesocket(int s) { ::close(s); }
  45 #endif
  46
  47 #include "pbd/xml++.h"
  48 #include "pbd/error.h"
  49 #include "pbd/failed_constructor.h"
  50 #include "pbd/convert.h"
  51 #include "pbd/compose.h"
  52
  53 #include "midi++/types.h"
  54 #include "midi++/ipmidi_port.h"
  55 #include "midi++/channel.h"
  56
  57 using namespace MIDI;
  58 using namespace std;
  59 using namespace PBD;
  60
  61 IPMIDIPort::IPMIDIPort (int base_port, const string& iface)
  62         : Port (string_compose ("IPmidi@%1", base_port), Port::Flags (Port::IsInput|Port::IsOutput))
  63         , sockin (-1)
  64         , sockout (-1)
  65 {
  66         if (!open_sockets (base_port, iface)) {
  67                 throw (failed_constructor ());
  68         }
  69 }
  70
  71 IPMIDIPort::IPMIDIPort (const XMLNode& node)
  72         : Port (node)
  73 {
  74         /* base class does not class set_state() */
  75         set_state (node);
  76 }
  77
  78 IPMIDIPort::~IPMIDIPort ()
  79 {
  80         close_sockets ();
  81 }
  82
  83 int
  84 IPMIDIPort::selectable () const
  85 {
  86         return sockin;
  87 }
  88
  89 XMLNode&
  90 IPMIDIPort::get_state () const
  91 {
  92         return Port::get_state ();
  93 }
  94
  95 void
  96 IPMIDIPort::set_state (const XMLNode& node)
  97 {
  98         Port::set_state (node);
  99 }
 100
 101 void
 102 IPMIDIPort::close_sockets ()
 103 {
 104         if (sockin >= 0) {
 105                 ::closesocket (sockin);
 106                 sockin = -1;
 107         }
 108
 109         if (sockout >= 0) {
 110                 ::closesocket (sockout);
 111                 sockout = -1;
 112         }
 113 }
 114
 115 static bool
 116 get_address (int sock, struct in_addr *inaddr, const string& ifname )
 117 {
 118         // Get interface address from supplied name.
 119
 120 #if !defined(PLATFORM_WINDOWS)
 121         struct ifreq ifr;
 122         ::strncpy(ifr.ifr_name, ifname.c_str(), sizeof(ifr.ifr_name));
 123
 124         if (::ioctl(sock, SIOCGIFFLAGS, (char *) &ifr)) {
 125                 ::perror("ioctl(SIOCGIFFLAGS)");
 126                 return false;
 127         }
 128
 129         if ((ifr.ifr_flags & IFF_UP) == 0) {
 130                 error << string_compose ("interface %1 is down", ifname) << endmsg;
 131                 return false;
 132         }
 133
 134         if (::ioctl(sock, SIOCGIFADDR, (char *) &ifr)) {
 135                 ::perror("ioctl(SIOCGIFADDR)");
 136                 return false;
 137         }
 138
 139         struct sockaddr_in sa;
 140         ::memcpy(&sa, &ifr.ifr_addr, sizeof(struct sockaddr_in));
 141         inaddr->s_addr = sa.sin_addr.s_addr;
 142
 143         return true;
 144
 145 #else
 146
 147         return false;
 148
 149 #endif  // !PLATFORM_WINDOWS'
 150 }
 151
 152 bool
 153 IPMIDIPort::open_sockets (int base_port, const string& ifname)
 154 {
 155 #if !defined(PLATFORM_WINDOWS)
 156         int protonum = 0;
 157         struct protoent *proto = ::getprotobyname("IP");
 158
 159         if (proto) {
 160                 protonum = proto->p_proto;
 161         }
 162
 163         sockin = ::socket (PF_INET, SOCK_DGRAM, protonum);
 164         if (sockin < 0) {
 165                 ::perror("socket(in)");
 166                 return false;
 167         }
 168
 169         struct sockaddr_in addrin;
 170         ::memset(&addrin, 0, sizeof(addrin));
 171         addrin.sin_family = AF_INET;
 172         addrin.sin_addr.s_addr = htonl(INADDR_ANY);
 173         addrin.sin_port = htons(base_port);
 174
 175         if (::bind(sockin, (struct sockaddr *) (&addrin), sizeof(addrin)) < 0) {
 176                 ::perror("bind");
 177                 return false;
 178         }
 179
 180         // Will Hall, 2007
 181         // INADDR_ANY will bind to default interface,
 182         // specify alternate interface nameon which to bind...
 183         struct in_addr if_addr_in;
 184         if (!ifname.empty()) {
 185                 if (!get_address(sockin, &if_addr_in, ifname)) {
 186                         error << string_compose ("socket(in): could not find interface address for %1", ifname) << endmsg;
 187                         return false;
 188                 }
 189                 if (::setsockopt(sockin, IPPROTO_IP, IP_MULTICAST_IF,
 190                                  (char *) &if_addr_in, sizeof(if_addr_in))) {
 191                         ::perror("setsockopt(IP_MULTICAST_IF)");
 192                         return false;
 193                 }
 194         } else {
 195                 if_addr_in.s_addr = htonl (INADDR_ANY);
 196         }
 197
 198         struct ip_mreq mreq;
 199         mreq.imr_multiaddr.s_addr = ::inet_addr("225.0.0.37");
 200         mreq.imr_interface.s_addr = if_addr_in.s_addr;
 201         if(::setsockopt (sockin, IPPROTO_IP, IP_ADD_MEMBERSHIP, (char *) &mreq, sizeof(mreq)) < 0) {
 202                 ::perror("setsockopt(IP_ADD_MEMBERSHIP)");
 203                 fprintf(stderr, "socket(in): your kernel is probably missing multicast support.\n");
 204                 return false;
 205         }
 206
 207         // Output socket...
 208
 209         sockout = ::socket (AF_INET, SOCK_DGRAM, protonum);
 210
 211         if (sockout < 0) {
 212                 ::perror("socket(out)");
 213                 return false;
 214         }
 215
 216         // Will Hall, Oct 2007
 217         if (!ifname.empty()) {
 218                 struct in_addr if_addr_out;
 219                 if (!get_address(sockout, &if_addr_out, ifname)) {
 220                         error << string_compose ("socket(out): could not find interface address for %1", ifname) << endmsg;
 221                         return false;
 222                 }
 223                 if (::setsockopt(sockout, IPPROTO_IP, IP_MULTICAST_IF, (char *) &if_addr_out, sizeof(if_addr_out))) {
 224                         ::perror("setsockopt(IP_MULTICAST_IF)");
 225                         return false;
 226                 }
 227         }
 228
 229         ::memset(&addrout, 0, sizeof(struct sockaddr_in));
 230         addrout.sin_family = AF_INET;
 231         addrout.sin_addr.s_addr = ::inet_addr("225.0.0.37");
 232         addrout.sin_port = htons (base_port);
 233
 234         // Turn off loopback...
 235         int loop = 0;
 236         if (::setsockopt(sockout, IPPROTO_IP, IP_MULTICAST_LOOP, (char *) &loop, sizeof (loop)) < 0) {
 237                 ::perror("setsockopt(IP_MULTICAST_LOOP)");
 238                 return false;
 239         }
 240
 241         if (fcntl (sockin, F_SETFL, O_NONBLOCK)) {
 242                 error << "cannot set non-blocking mode for IP MIDI input socket (" << ::strerror (errno) << ')' << endmsg;
 243                 return false;
 244         }
 245
 246         if (fcntl (sockout, F_SETFL, O_NONBLOCK)) {
 247                 error << "cannot set non-blocking mode for IP MIDI output socket (" << ::strerror (errno) << ')' << endmsg;
 248                 return false;
 249         }
 250
 251         return true;
 252 #else
 253         return false;
 254 #endif  // !PLATFORM_WINDOWS'
 255 }
 256
 257 int
 258 IPMIDIPort::write (const MIDI::byte* msg, size_t msglen, timestamp_t /* ignored */) {
 259
 260         if (sockout) {
 261                 Glib::Threads::Mutex::Lock lm (write_lock);
 262                 if (::sendto (sockout, (const char *) msg, msglen, 0, (struct sockaddr *) &addrout, sizeof(struct sockaddr_in)) < 0) {
 263                         ::perror("sendto");
 264                         return -1;
 265                 }
 266                 return msglen;
 267         }
 268         return 0;
 269 }
 270
 271 int
 272 IPMIDIPort::read (MIDI::byte* /*buf*/, size_t /*bufsize*/)
 273 {
 274         /* nothing to do here - all handled by parse() */
 275         return 0;
 276 }
 277
 278 void
 279 IPMIDIPort::parse (framecnt_t timestamp)
 280 {
 281         /* input was detected on the socket, so go get it and hand it to the
 282          * parser. This will emit appropriate signals that will be handled
 283          * by anyone who cares.
 284          */
 285
 286         unsigned char buf[1024];
 287         struct sockaddr_in sender;
 288         socklen_t slen = sizeof(sender);
 289         int r = ::recvfrom (sockin, (char *) buf, sizeof(buf), 0, (struct sockaddr *) &sender, &slen);
 290
 291         if (r >= 0) {
 292
 293                 _parser->set_timestamp (timestamp);
 294
 295                 for (int i = 0; i < r; ++i) {
 296                         _parser->scanner (buf[i]);
 297                 }
 298         } else {
 299                 ::perror ("failed to recv from socket");
 300         }
 301 }
 302