Low-overhead filter streams

diff --git a/src/include/gpxe/filter.h b/src/include/gpxe/filter.h
new file mode 100644 (file)
index 0000000..585bbe1
--- /dev/null
+++ b/src/include/gpxe/filter.h
@@ -0,0 +1,44 @@
+#ifndef _GPXE_FILTER_H
+#define _GPXE_FILTER_H
+
+/** @file
+ *
+ * Filter streams
+ */
+
+#include <gpxe/stream.h>
+
+/** A filter stream */
+struct filter_stream {
+       /** Upstream
+        *
+        * This is the end pointing towards the top-level application
+        * (e.g. HTTP).
+        */
+       struct stream_connection upstream;
+       /** Downstream
+        *
+        * This is the end pointing towards the bottom-level
+        * connection (e.g. TCP).
+        */
+       struct stream_application downstream;
+};
+
+extern void filter_connected ( struct stream_application *app );
+extern void filter_closed ( struct stream_application *app, int rc );
+extern void filter_senddata ( struct stream_application *app,
+                             void *data, size_t len );
+extern void filter_acked ( struct stream_application *app, size_t len );
+extern void filter_newdata ( struct stream_application *app,
+                            void *data, size_t len );
+
+extern int filter_bind ( struct stream_connection *conn,
+                        struct sockaddr *local );
+extern int filter_connect ( struct stream_connection *conn,
+                           struct sockaddr *peer );
+extern void filter_close ( struct stream_connection *conn );
+extern int filter_send ( struct stream_connection *conn,
+                        void *data, size_t len );
+extern int filter_kick ( struct stream_connection *conn );
+
+#endif /* _GPXE_FILTER_H */

diff --git a/src/net/filter.c b/src/net/filter.c
new file mode 100644 (file)
index 0000000..7d93c61
--- /dev/null
+++ b/src/net/filter.c
@@ -0,0 +1,163 @@
+/*
+ * Copyright (C) 2007 Michael Brown <mbrown@fensystems.co.uk>.
+ *
+ * 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 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., 675 Mass Ave, Cambridge, MA 02139, USA.
+ */
+
+/**
+ * @file
+ *
+ * Filter streams
+ */
+
+#include <stddef.h>
+#include <gpxe/stream.h>
+#include <gpxe/filter.h>
+
+/**
+ * Connection established
+ *
+ * @v app              Stream application
+ */
+void filter_connected ( struct stream_application *app ) {
+       struct filter_stream *filter = 
+               container_of ( app, struct filter_stream, downstream );
+
+       stream_connected ( &filter->upstream );
+}
+
+/**
+ * Connection closed
+ *
+ * @v app              Stream application
+ * @v rc               Error code, if any
+ */
+void filter_closed ( struct stream_application *app, int rc ) {
+       struct filter_stream *filter = 
+               container_of ( app, struct filter_stream, downstream );
+
+       stream_closed ( &filter->upstream, rc );
+}
+
+/**
+ * Transmit data
+ *
+ * @v app              Stream application
+ * @v buf              Temporary data buffer
+ * @v len              Length of temporary data buffer
+ */
+void filter_senddata ( struct stream_application *app,
+                      void *data, size_t len ) {
+       struct filter_stream *filter = 
+               container_of ( app, struct filter_stream, downstream );
+
+       stream_senddata ( &filter->upstream, data, len );
+}
+
+/**
+ * Transmitted data acknowledged
+ *
+ * @v app              Stream application
+ * @v len              Length of acknowledged data
+ */
+void filter_acked ( struct stream_application *app, size_t len ) {
+       struct filter_stream *filter = 
+               container_of ( app, struct filter_stream, downstream );
+
+       stream_acked ( &filter->upstream, len );
+}
+
+/**
+ * Receive new data
+ *
+ * @v app              Stream application
+ * @v data             Data
+ * @v len              Length of data
+ */
+void filter_newdata ( struct stream_application *app,
+                     void *data, size_t len ) {
+       struct filter_stream *filter = 
+               container_of ( app, struct filter_stream, downstream );
+
+       stream_newdata ( &filter->upstream, data, len );
+}
+
+/**
+ * Bind to local address
+ *
+ * @v conn             Stream connection
+ * @v local            Local address
+ * @ret rc             Return status code
+ */
+int filter_bind ( struct stream_connection *conn, struct sockaddr *local ) {
+       struct filter_stream *filter = 
+               container_of ( conn, struct filter_stream, upstream );
+
+       return stream_bind ( &filter->downstream, local );
+}
+
+/**
+ * Connect to remote address
+ *
+ * @v conn             Stream connection
+ * @v peer             Remote address
+ * @ret rc             Return status code
+ */
+int filter_connect ( struct stream_connection *conn, struct sockaddr *peer ) {
+       struct filter_stream *filter = 
+               container_of ( conn, struct filter_stream, upstream );
+
+       return stream_connect ( &filter->downstream, peer );
+}
+
+/**
+ * Close connection
+ *
+ * @v conn             Stream connection
+ */
+void filter_close ( struct stream_connection *conn ) {
+       struct filter_stream *filter = 
+               container_of ( conn, struct filter_stream, upstream );
+
+       stream_close ( &filter->downstream );
+}
+
+/**
+ * Send data via connection
+ *
+ * @v conn             Stream connection
+ * @v data             Data to send
+ * @v len              Length of data
+ * @ret rc             Return status code
+ */
+int filter_send ( struct stream_connection *conn, void *data, size_t len ) {
+       struct filter_stream *filter = 
+               container_of ( conn, struct filter_stream, upstream );
+
+       return stream_send ( &filter->downstream, data, len );
+}
+
+/**
+ * Notify connection that data is available to send
+ *
+ * @v conn             Stream connection
+ * @ret rc             Return status code
+ */
+int filter_kick ( struct stream_connection *conn ) {
+       struct filter_stream *filter = 
+               container_of ( conn, struct filter_stream, upstream );
+
+       return stream_kick ( &filter->downstream );
+}