summaryrefslogtreecommitdiff
path: root/digital/ucoolib/ucoolib/arch/host_stream.host.cc
blob: 6bf5f8b9cc401e229dff04e4aaa3fb9b872df2b4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
// ucoolib - Microcontroller object oriented library. {{{
//
// Copyright (C) 2012 Nicolas Schodet
//
// APBTeam:
//        Web: http://apbteam.org/
//      Email: team AT apbteam DOT org
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
//
// }}}
#include "host_stream.hh"

#include "ucoolib/common.hh"

#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>

namespace ucoo {

HostStream::HostStream ()
    : fdi_ (0), fdo_ (1)
{
}

HostStream::~HostStream ()
{
    if (fdi_ != -1 && fdi_ != 0)
        close (fdi_);
    if (fdo_ != -1 && fdo_ != 1)
        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