wvconstream.cc

/*
 * Worldvisions Weaver Software:
 *   Copyright (C) 1997-2002 Net Integration Technologies, Inc.
 * 
 * Declarations for wvcon, wvin, wvout, and wverr global streams.
 */
#include "wvfdstream.h"
#include "wvmoniker.h"
#include "wvlinkerhack.h"
 
WV_LINK(WvConStream);
 
// just like WvFDStream, but doesn't close the fd
class _WvConStream : public WvFDStream
{
public:
    bool isopen;
    
    _WvConStream(int _rfd, int _wfd, WvStringParm name = WvString::null);
    virtual ~_WvConStream();
    virtual void close();
    virtual bool isok() const;
};
 
 
_WvConStream::_WvConStream(int _rfd, int _wfd,
        WvStringParm name)
    : WvFDStream(_rfd, _wfd)
{
    isopen = true;
    if (!name.isnull())
        set_wsname(name);
}
 
 
_WvConStream::~_WvConStream()
{
    close();
}
 
 
void _WvConStream::close()
{
    isopen = false;
    setfd(-1); // prevent WvFdStream from closing our fds
    WvFDStream::close();
}
 
 
bool _WvConStream::isok() const
{
    return isopen;
}
 
 
// console streams
#ifdef _WIN32
 
#include "streams.h"
SocketFromFDMaker _zero(_dup(0), fd2socket_fwd, false);
SocketFromFDMaker _one(1, socket2fd_fwd, true);
SocketFromFDMaker _two(2, socket2fd_fwd, true);
 
static _WvConStream _wvcon(_zero.GetSocket(), _one.GetSocket(), "wvcon");
static _WvConStream _wvin(_zero.GetSocket(), -1, "wvin");
static _WvConStream _wvout(-1, _one.GetSocket(), "wvout");
static _WvConStream _wverr(-1, _two.GetSocket(), "wverr");
 
#else // _WIN32
 
static _WvConStream _wvcon(0, 1, "wvcon");
static _WvConStream _wvin(0, -1, "wvin");
static _WvConStream _wvout(-1, 1, "wvout");
static _WvConStream _wverr(-1, 2, "wverr");
 
#endif // !_WIN32
 
WvStream *wvcon = &_wvcon;
WvStream *wvin = &_wvin;
WvStream *wvout = &_wvout;
WvStream *wverr = &_wverr;
 
 
static IWvStream *create_stdin(WvStringParm s, IObject*)
{
    wvin->addRef();
    return wvin;
}
static IWvStream *create_stdout(WvStringParm s, IObject*)
{
    wvout->addRef();
    return wvout;
}
static IWvStream *create_stderr(WvStringParm s, IObject*)
{
    wverr->addRef();
    return wverr;
}
static IWvStream *create_stdio(WvStringParm s, IObject*)
{
    wvcon->addRef();
    return wvcon;
}
 
static WvMoniker<IWvStream> reg0("stdin",  create_stdin);
static WvMoniker<IWvStream> reg1("stdout", create_stdout);
static WvMoniker<IWvStream> reg2("stderr", create_stderr);
static WvMoniker<IWvStream> reg3("stdio",  create_stdio);