X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=lib%2Fvconn-stream.c;h=d707e063c302364bd33305e8946b9c9881ee90e9;hb=d58c5bc2d854ce601338d0785c8086d619a16135;hp=034d1d5cee505be0488eb252f39c035d9224d2c8;hpb=f39dc942afd5fe241903aada30850a1d96122c8c;p=openvswitch diff --git a/lib/vconn-stream.c b/lib/vconn-stream.c index 034d1d5c..d707e063 100644 --- a/lib/vconn-stream.c +++ b/lib/vconn-stream.c @@ -1,5 +1,5 @@ /* - * Copyright (c) 2008, 2009, 2010 Nicira Networks. + * Copyright (c) 2008, 2009, 2010, 2011 Nicira, Inc. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -32,9 +32,9 @@ #include "util.h" #include "vconn-provider.h" #include "vconn.h" - #include "vlog.h" -#define THIS_MODULE VLM_vconn_stream + +VLOG_DEFINE_THIS_MODULE(vconn_stream); /* Active stream socket vconn. */ @@ -44,6 +44,7 @@ struct vconn_stream struct stream *stream; struct ofpbuf *rxbuf; struct ofpbuf *txbuf; + int n_packets; }; static struct vconn_class stream_vconn_class; @@ -63,6 +64,7 @@ vconn_stream_new(struct stream *stream, int connect_status) s->stream = stream; s->txbuf = NULL; s->rxbuf = NULL; + s->n_packets = 0; s->vconn.remote_ip = stream_get_remote_ip(stream); s->vconn.remote_port = stream_get_remote_port(stream); s->vconn.local_ip = stream_get_local_ip(stream); @@ -76,20 +78,23 @@ vconn_stream_new(struct stream *stream, int connect_status) * Returns 0 if successful, otherwise a positive errno value. */ static int vconn_stream_open(const char *name, char *suffix OVS_UNUSED, - struct vconn **vconnp) + struct vconn **vconnp, uint8_t dscp) { struct stream *stream; int error; error = stream_open_with_default_ports(name, OFP_TCP_PORT, OFP_SSL_PORT, - &stream); - - if (error && error != EAGAIN) { - return error; + &stream, dscp); + if (!error) { + error = stream_connect(stream); + if (!error || error == EAGAIN) { + *vconnp = vconn_stream_new(stream, error); + return 0; + } } - *vconnp = vconn_stream_new(stream, error); - return 0; + stream_close(stream); + return error; } static struct vconn_stream * @@ -102,6 +107,12 @@ static void vconn_stream_close(struct vconn *vconn) { struct vconn_stream *s = vconn_stream_cast(vconn); + + if ((vconn->error == EPROTO || s->n_packets < 1) && s->rxbuf) { + stream_report_content(s->rxbuf->data, s->rxbuf->size, STREAM_OPENFLOW, + THIS_MODULE, vconn_get_name(vconn)); + } + stream_close(s->stream); vconn_stream_clear_txbuf(s); ofpbuf_delete(s->rxbuf); @@ -116,63 +127,67 @@ vconn_stream_connect(struct vconn *vconn) } static int -vconn_stream_recv(struct vconn *vconn, struct ofpbuf **bufferp) +vconn_stream_recv__(struct vconn_stream *s, int rx_len) { - struct vconn_stream *s = vconn_stream_cast(vconn); - struct ofpbuf *rx; - size_t want_bytes; - ssize_t retval; - - if (s->rxbuf == NULL) { - s->rxbuf = ofpbuf_new(1564); - } - rx = s->rxbuf; + struct ofpbuf *rx = s->rxbuf; + int want_bytes, retval; -again: - if (sizeof(struct ofp_header) > rx->size) { - want_bytes = sizeof(struct ofp_header) - rx->size; - } else { - struct ofp_header *oh = rx->data; - size_t length = ntohs(oh->length); - if (length < sizeof(struct ofp_header)) { - VLOG_ERR_RL(&rl, "received too-short ofp_header (%zu bytes)", - length); - return EPROTO; - } - want_bytes = length - rx->size; - if (!want_bytes) { - *bufferp = rx; - s->rxbuf = NULL; - return 0; - } - } + want_bytes = rx_len - rx->size; ofpbuf_prealloc_tailroom(rx, want_bytes); - retval = stream_recv(s->stream, ofpbuf_tail(rx), want_bytes); if (retval > 0) { rx->size += retval; - if (retval == want_bytes) { - if (rx->size > sizeof(struct ofp_header)) { - *bufferp = rx; - s->rxbuf = NULL; - return 0; - } else { - goto again; - } - } - return EAGAIN; + return retval == want_bytes ? 0 : EAGAIN; } else if (retval == 0) { if (rx->size) { VLOG_ERR_RL(&rl, "connection dropped mid-packet"); return EPROTO; - } else { - return EOF; } + return EOF; } else { return -retval; } } +static int +vconn_stream_recv(struct vconn *vconn, struct ofpbuf **bufferp) +{ + struct vconn_stream *s = vconn_stream_cast(vconn); + const struct ofp_header *oh; + int rx_len; + + /* Allocate new receive buffer if we don't have one. */ + if (s->rxbuf == NULL) { + s->rxbuf = ofpbuf_new(1564); + } + + /* Read ofp_header. */ + if (s->rxbuf->size < sizeof(struct ofp_header)) { + int retval = vconn_stream_recv__(s, sizeof(struct ofp_header)); + if (retval) { + return retval; + } + } + + /* Read payload. */ + oh = s->rxbuf->data; + rx_len = ntohs(oh->length); + if (rx_len < sizeof(struct ofp_header)) { + VLOG_ERR_RL(&rl, "received too-short ofp_header (%d bytes)", rx_len); + return EPROTO; + } else if (s->rxbuf->size < rx_len) { + int retval = vconn_stream_recv__(s, rx_len); + if (retval) { + return retval; + } + } + + s->n_packets++; + *bufferp = s->rxbuf; + s->rxbuf = NULL; + return 0; +} + static void vconn_stream_clear_txbuf(struct vconn_stream *s) { @@ -212,6 +227,7 @@ vconn_stream_run(struct vconn *vconn) struct vconn_stream *s = vconn_stream_cast(vconn); ssize_t retval; + stream_run(s->stream); if (!s->txbuf) { return; } @@ -237,6 +253,7 @@ vconn_stream_run_wait(struct vconn *vconn) { struct vconn_stream *s = vconn_stream_cast(vconn); + stream_run_wait(s->stream); if (s->txbuf) { stream_send_wait(s->stream); } @@ -294,14 +311,14 @@ pvconn_pstream_cast(struct pvconn *pvconn) * implementation never fails.) */ static int pvconn_pstream_listen(const char *name, char *suffix OVS_UNUSED, - struct pvconn **pvconnp) + struct pvconn **pvconnp, uint8_t dscp) { struct pvconn_pstream *ps; struct pstream *pstream; int error; error = pstream_open_with_default_ports(name, OFP_TCP_PORT, OFP_SSL_PORT, - &pstream); + &pstream, dscp); if (error) { return error; } @@ -350,9 +367,9 @@ pvconn_pstream_wait(struct pvconn *pvconn) /* Stream-based vconns and pvconns. */ -#define DEFINE_VCONN_STREAM_CLASS(NAME) \ - struct vconn_class NAME##_vconn_class = { \ - #NAME, \ +#define STREAM_INIT(NAME) \ + { \ + NAME, \ vconn_stream_open, \ vconn_stream_close, \ vconn_stream_connect, \ @@ -361,27 +378,27 @@ pvconn_pstream_wait(struct pvconn *pvconn) vconn_stream_run, \ vconn_stream_run_wait, \ vconn_stream_wait, \ - }; + } -#define DEFINE_PVCONN_STREAM_CLASS(NAME) \ - struct pvconn_class NAME##_pvconn_class = { \ - #NAME, \ +#define PSTREAM_INIT(NAME) \ + { \ + NAME, \ pvconn_pstream_listen, \ pvconn_pstream_close, \ pvconn_pstream_accept, \ pvconn_pstream_wait \ - }; + } -static DEFINE_VCONN_STREAM_CLASS(stream); -static DEFINE_PVCONN_STREAM_CLASS(pstream); +static struct vconn_class stream_vconn_class = STREAM_INIT("stream"); +static struct pvconn_class pstream_pvconn_class = PSTREAM_INIT("pstream"); -DEFINE_VCONN_STREAM_CLASS(tcp); -DEFINE_PVCONN_STREAM_CLASS(ptcp); +struct vconn_class tcp_vconn_class = STREAM_INIT("tcp"); +struct pvconn_class ptcp_pvconn_class = PSTREAM_INIT("ptcp"); -DEFINE_VCONN_STREAM_CLASS(unix); -DEFINE_PVCONN_STREAM_CLASS(punix); +struct vconn_class unix_vconn_class = STREAM_INIT("unix"); +struct pvconn_class punix_pvconn_class = PSTREAM_INIT("punix"); #ifdef HAVE_OPENSSL -DEFINE_VCONN_STREAM_CLASS(ssl); -DEFINE_PVCONN_STREAM_CLASS(pssl); +struct vconn_class ssl_vconn_class = STREAM_INIT("ssl"); +struct pvconn_class pssl_pvconn_class = PSTREAM_INIT("pssl"); #endif