X-Git-Url: http://git.tdb.fi/?p=libs%2Fnet.git;a=blobdiff_plain;f=examples%2Fnetcat.cpp;fp=examples%2Fnetcat.cpp;h=2532959e56ec7e8b595c035d7334b14c797a8b95;hp=0000000000000000000000000000000000000000;hb=5d91f9edad6e8a11e21979a6aacb4ec6f2dc63ce;hpb=dbfb303f336037cf7ca4fca3d8bfe504c624b501

diff --git a/examples/netcat.cpp b/examples/netcat.cpp
new file mode 100644
index 0000000..2532959
--- /dev/null
+++ b/examples/netcat.cpp
@@ -0,0 +1,98 @@
+#include <msp/core/application.h>
+#include <msp/core/getopt.h>
+#include <msp/core/refptr.h>
+#include <msp/io/console.h>
+#include <msp/io/eventdispatcher.h>
+#include <msp/net/streamlistensocket.h>
+#include <msp/net/streamsocket.h>
+#include <msp/net/resolve.h>
+
+using namespace std;
+using namespace Msp;
+
+class NetCat: public RegisteredApplication<NetCat>
+{
+private:
+	bool ipv6;
+	bool listen;
+	Net::StreamListenSocket *server_sock;
+	Net::StreamSocket *sock;
+	IO::EventDispatcher event_disp;
+
+public:
+	NetCat(int, char **);
+
+private:
+	virtual void tick();
+	void net_data_available();
+	void console_data_available();
+};
+
+NetCat::NetCat(int argc, char **argv):
+	ipv6(false),
+	listen(false),
+	server_sock(0),
+	sock(0)
+{
+	GetOpt getopt;
+	getopt.add_option('6', "ipv6",   ipv6,   GetOpt::NO_ARG);
+	getopt.add_option('l', "listen", listen, GetOpt::NO_ARG);
+	getopt(argc, argv);
+
+	const vector<string> &args = getopt.get_args();
+	if(args.empty())
+		throw usage_error("host argument missing");
+
+	RefPtr<Net::SockAddr> addr = Net::resolve(args.front(), (ipv6 ? Net::INET6 : Net::INET));
+	if(!listen)
+	{
+		sock = new Net::StreamSocket(addr->get_family());
+		sock->connect(*addr);
+		event_disp.add(*sock);
+		sock->signal_data_available.connect(sigc::mem_fun(this, &NetCat::net_data_available));
+	}
+	else
+	{
+		server_sock = new Net::StreamListenSocket(addr->get_family());
+		server_sock->listen(*addr);
+		event_disp.add(*server_sock);
+		server_sock->signal_data_available.connect(sigc::mem_fun(this, &NetCat::net_data_available));
+	}
+
+	event_disp.add(IO::cin);
+	IO::cin.signal_data_available.connect(sigc::mem_fun(this, &NetCat::console_data_available));
+}
+
+void NetCat::tick()
+{
+	event_disp.tick();
+	if(server_sock && sock)
+	{
+		delete server_sock;
+		server_sock = 0;
+	}
+}
+
+void NetCat::net_data_available()
+{
+	if(server_sock)
+	{
+		sock = server_sock->accept();
+		event_disp.add(*sock);
+		sock->signal_data_available.connect(sigc::mem_fun(this, &NetCat::net_data_available));
+	}
+	else
+	{
+		char buf[1024];
+		unsigned len = sock->read(buf, sizeof(buf));
+		IO::cout.write(buf, len);
+	}
+}
+
+void NetCat::console_data_available()
+{
+	char buf[1024];
+	unsigned len = IO::cin.read(buf, sizeof(buf));
+	if(sock)
+		sock->write(buf, len);
+}