summaryrefslogtreecommitdiff
path: root/ucoo/arch/host/host_stream.host.cc
diff options
context:
space:
mode:
Diffstat (limited to 'ucoo/arch/host/host_stream.host.cc')
-rw-r--r--ucoo/arch/host/host_stream.host.cc146
1 files changed, 146 insertions, 0 deletions
diff --git a/ucoo/arch/host/host_stream.host.cc b/ucoo/arch/host/host_stream.host.cc
new file mode 100644
index 0000000..d971862
--- /dev/null
+++ b/ucoo/arch/host/host_stream.host.cc
@@ -0,0 +1,146 @@
+// ucoolib - Microcontroller object oriented library. {{{
+//
+// Copyright (C) 2012 Nicolas Schodet
+//
+// Permission is hereby granted, free of charge, to any person obtaining a
+// copy of this software and associated documentation files (the "Software"),
+// to deal in the Software without restriction, including without limitation
+// the rights to use, copy, modify, merge, publish, distribute, sublicense,
+// and/or sell copies of the Software, and to permit persons to whom the
+// Software is furnished to do so, subject to the following conditions:
+//
+// The above copyright notice and this permission notice shall be included in
+// all copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+// DEALINGS IN THE SOFTWARE.
+//
+// }}}
+#include "host_stream.hh"
+
+#include "ucoo/common.hh"
+
+#include <stdlib.h>
+#include <sys/time.h>
+#include <sys/types.h>
+#include <unistd.h>
+#include <pty.h>
+#include <fcntl.h>
+#include <errno.h>
+
+namespace ucoo {
+
+/// Setup non canonical mode.
+static void
+setup_raw (int fd)
+{
+ struct termios tc;
+ tcgetattr (fd, &tc);
+ tc.c_iflag &= ~(IGNPAR | PARMRK | ISTRIP | IGNBRK | BRKINT | IGNCR |
+ ICRNL | INLCR | IXON | IXOFF | IXANY | IMAXBEL);
+ tc.c_iflag |= INPCK;
+ tc.c_oflag &= ~(OPOST);
+ tc.c_cflag &= ~(HUPCL | CSTOPB | PARENB | PARODD | CSIZE);
+ tc.c_cflag |= CS8 | CLOCAL | CREAD;
+ tc.c_lflag &= ~(ICANON | ECHO | ISIG | IEXTEN | NOFLSH | TOSTOP);
+ tc.c_cc[VTIME] = 0;
+ tc.c_cc[VMIN] = 1;
+ tcflush (fd, TCIFLUSH);
+ tcsetattr (fd, TCSANOW, &tc);
+}
+
+HostStream::HostStream ()
+ : fdi_ (0), fdo_ (1)
+{
+}
+
+HostStream::HostStream (const char *name)
+ : fdi_ (-1), fdo_ (-1)
+{
+ int fd, slave_fd, r;
+ // Open and unlock pt.
+ if (openpty (&fd, &slave_fd, 0, 0, 0) == -1
+ || grantpt (fd) == -1
+ || unlockpt (fd) == -1)
+ halt_perror ();
+ // Make a link to the slave pts.
+ unlink (name);
+ const char *slave_name = ptsname (fd);
+ assert (slave_name);
+ r = symlink (slave_name, name);
+ assert_perror (r != -1);
+ // Make slave raw.
+ setup_raw (slave_fd);
+ // Use as both in and out.
+ fdi_ = fdo_ = fd;
+ // slave_fd is left open.
+}
+
+HostStream::~HostStream ()
+{
+ if (fdi_ != -1 && fdi_ != 0)
+ close (fdi_);
+ if (fdo_ != -1 && fdo_ != 1 && fdo_ != fdi_)
+ close (fdo_);
+}
+
+void
+HostStream::block (bool block)
+{
+ Stream::block (block);
+ // Do not care to use non blocking on output for host. However, if
+ // fdi_ == fdo_, this will change input and output.
+ int r = fcntl (fdi_, F_SETFL, block ? 0 : O_NONBLOCK);
+ assert_perror (r != -1);
+}
+
+int
+HostStream::read (char *buf, int count)
+{
+ int r = ::read (fdi_, buf, count);
+ if (r == 0)
+ return -2;
+ if (r == -1 && errno == EAGAIN)
+ return 0;
+ assert_perror (r != -1);
+ return r;
+}
+
+int
+HostStream::write (const char *buf, int count)
+{
+ int writen = 0;
+ while (writen < count)
+ {
+ int r = ::write (fdo_, buf + writen, count - writen);
+ if (r == -1 && errno == EAGAIN)
+ break;
+ assert_perror (r != -1);
+ writen += r;
+ }
+ return writen;
+}
+
+int
+HostStream::poll ()
+{
+ fd_set fds;
+ struct timeval tv;
+ int r;
+ // Use select to poll.
+ FD_ZERO (&fds);
+ FD_SET (fdi_, &fds);
+ tv.tv_sec = 0;
+ tv.tv_usec = 0;
+ r = select (FD_SETSIZE, &fds, 0, 0, &tv);
+ // Check result.
+ assert_perror (r != -1);
+ return r;
+}
+
+} // namespace ucoo