123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352 |
- /*
- * 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
- *
- * Stream API
- */
-
- #include <stdint.h>
- #include <string.h>
- #include <errno.h>
- #include <assert.h>
- #include <gpxe/stream.h>
-
- /**
- * Associate application with connection
- *
- * @v app Stream application
- * @v conn Stream connection
- */
- void stream_associate ( struct stream_application *app,
- struct stream_connection *conn ) {
-
- DBGC ( app, "Stream %p associating with connection %p\n", app, conn );
-
- assert ( conn->app == NULL );
- assert ( app->conn == NULL );
- conn->app = app;
- app->conn = conn;
- }
-
- /**
- * Disassociate application from connection
- *
- * @v app Stream application
- * @v conn Stream connection
- */
- static void stream_disassociate ( struct stream_application *app,
- struct stream_connection *conn ) {
-
- DBGC ( app, "Stream %p disassociating from connection %p\n",
- app, conn );
-
- assert ( conn->app == app );
- assert ( app->conn == conn );
- conn->app = NULL;
- app->conn = NULL;
- }
-
- /**
- * Connection established
- *
- * @v conn Stream connection
- */
- void stream_connected ( struct stream_connection *conn ) {
- struct stream_application *app = conn->app;
-
- DBGC ( app, "Stream %p connected\n", app );
-
- /* Check connection actually exists */
- if ( ! app ) {
- DBGC ( conn, "Stream connection %p has no application\n",
- conn );
- return;
- }
-
- /* Hand off to application */
- if ( app->op->connected )
- app->op->connected ( app );
- }
-
- /**
- * Connection closed
- *
- * @v conn Stream connection
- * @v rc Error code, if any
- */
- void stream_closed ( struct stream_connection *conn, int rc ) {
- struct stream_application *app = conn->app;
-
- /* Check connection actually exists */
- if ( ! app ) {
- /* Not an error; don't display a debug message */
- return;
- }
-
- DBGC ( app, "Stream %p closed (%s)\n", app, strerror ( rc ) );
-
- /* Disassociate application from connection */
- stream_disassociate ( app, conn );
-
- /* Hand off to application */
- if ( app->op->closed )
- app->op->closed ( app, rc );
- }
-
- /**
- * Transmit data
- *
- * @v conn Stream connection
- * @v data Temporary data buffer
- * @v len Length of temporary data buffer
- */
- void stream_senddata ( struct stream_connection *conn,
- void *data, size_t len ) {
- struct stream_application *app = conn->app;
-
- DBGC2 ( app, "Stream %p sending data\n", app );
-
- /* Check connection actually exists */
- if ( ! app ) {
- DBGC ( conn, "Stream connection %p has no application\n",
- conn );
- return;
- }
-
- /* Hand off to application */
- if ( app->op->senddata )
- app->op->senddata ( app, data, len );
- }
-
- /**
- * Transmitted data acknowledged
- *
- * @v conn Stream connection
- * @v len Length of acknowledged data
- *
- * @c len must not exceed the outstanding amount of unacknowledged
- * data.
- */
- void stream_acked ( struct stream_connection *conn, size_t len ) {
- struct stream_application *app = conn->app;
-
- DBGC2 ( app, "Stream %p had %zd bytes acknowledged\n", app, len );
-
- /* Check connection actually exists */
- if ( ! app ) {
- DBGC ( conn, "Stream connection %p has no application\n",
- conn );
- return;
- }
-
- /* Ignore zero-length blocks */
- if ( len == 0 )
- return;
-
- /* Hand off to application */
- if ( app->op->acked )
- app->op->acked ( app, len );
- }
-
- /**
- * Receive new data
- *
- * @v conn Stream connection
- * @v data Data
- * @v len Length of data
- */
- void stream_newdata ( struct stream_connection *conn,
- void *data, size_t len ) {
- struct stream_application *app = conn->app;
-
- DBGC2 ( app, "Stream %p received %zd bytes\n", app, len );
-
- /* Check connection actually exists */
- if ( ! app ) {
- DBGC ( conn, "Stream connection %p has no application\n",
- conn );
- return;
- }
-
- /* Ignore zero-length blocks */
- if ( len == 0 )
- return;
-
- /* Hand off to application */
- if ( app->op->newdata )
- app->op->newdata ( app, data, len );
- }
-
- /**
- * Bind to local address
- *
- * @v app Stream application
- * @v local Local address
- * @ret rc Return status code
- */
- int stream_bind ( struct stream_application *app, struct sockaddr *local ) {
- struct stream_connection *conn = app->conn;
- int rc;
-
- DBGC2 ( app, "Stream %p binding\n", app );
-
- /* Check connection actually exists */
- if ( ! conn ) {
- DBGC ( app, "Stream %p has no connection\n", app );
- return -ENOTCONN;
- }
-
- /* Hand off to connection */
- if ( ! conn->op->bind )
- return -ENOTSUP;
- if ( ( rc = conn->op->bind ( conn, local ) ) != 0 ) {
- DBGC ( app, "Stream %p failed to bind: %s\n",
- app, strerror ( rc ) );
- return rc;
- }
-
- return 0;
- }
-
- /**
- * Connect to remote address
- *
- * @v app Stream application
- * @v peer Remote address
- * @ret rc Return status code
- */
- int stream_connect ( struct stream_application *app, struct sockaddr *peer ) {
- struct stream_connection *conn = app->conn;
- int rc;
-
- DBGC2 ( app, "Stream %p connecting\n", app );
-
- /* Check connection actually exists */
- if ( ! conn ) {
- DBGC ( app, "Stream %p has no connection\n", app );
- return -ENOTCONN;
- }
-
- /* Hand off to connection */
- if ( ! conn->op->connect )
- return -ENOTSUP;
- if ( ( rc = conn->op->connect ( conn, peer ) ) != 0 ) {
- DBGC ( app, "Stream %p failed to connect: %s\n",
- app, strerror ( rc ) );
- return rc;
- }
-
- return 0;
- }
-
- /**
- * Close connection
- *
- * @v app Stream application
- */
- void stream_close ( struct stream_application *app ) {
- struct stream_connection *conn = app->conn;
-
- DBGC2 ( app, "Stream %p closing\n", app );
-
- /* Check connection actually exists */
- if ( ! conn ) {
- /* Not an error; don't display a debug message */
- return;
- }
-
- /* Disassociate application from connection */
- stream_disassociate ( app, conn );
-
- /* Hand off to connection */
- if ( ! conn->op->close )
- return;
- conn->op->close ( conn );
- }
-
- /**
- * Send data via connection
- *
- * @v app Stream application
- * @v data Data to send
- * @v len Length of data
- * @ret rc Return status code
- *
- * This method should be called only in the context of an
- * application's senddata() method.
- */
- int stream_send ( struct stream_application *app,
- const void *data, size_t len ) {
- struct stream_connection *conn = app->conn;
- int rc;
-
- DBGC2 ( app, "Stream %p sending %zd bytes\n", app, len );
-
- /* Check connection actually exists */
- if ( ! conn ) {
- DBGC ( app, "Stream %p has no connection\n", app );
- return -ENOTCONN;
- }
-
- /* Ignore zero-length blocks */
- if ( len == 0 )
- return 0;
-
- /* Hand off to connection */
- if ( ! conn->op->send )
- return -ENOTSUP;
- if ( ( rc = conn->op->send ( conn, data, len ) ) != 0 ) {
- DBGC ( app, "Stream %p failed to send %zd bytes: %s\n",
- app, len, strerror ( rc ) );
- return rc;
- }
-
- return 0;
- }
-
- /**
- * Notify connection that data is available to send
- *
- * @v app Stream application
- * @ret rc Return status code
- */
- int stream_kick ( struct stream_application *app ) {
- struct stream_connection *conn = app->conn;
- int rc;
-
- DBGC2 ( app, "Stream %p kicking connection\n", app );
-
- /* Check connection actually exists */
- if ( ! conn ) {
- DBGC ( app, "Stream %p has no connection\n", app );
- return -ENOTCONN;
- }
-
- /* Hand off to connection */
- if ( ! conn->op->send )
- return -ENOTSUP;
- if ( ( rc = conn->op->kick ( conn ) ) != 0 ) {
- DBGC ( app, "Stream %p failed to kick connection: %s\n",
- app, strerror ( rc ) );
- return rc;
- }
-
- return 0;
- }
|