source: mainline/uspace/srv/net/tl/tcp/state.c@ 6896409c

/*
 * Copyright (c) 2011 Jiri Svoboda
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/** @addtogroup tcp
 * @{
 */

/**
 * @file TCP entry points (close to those defined in the RFC)
 */

#include <fibril_synch.h>
#include <io/log.h>
#include <macros.h>
#include <mem.h>
#include "conn.h"
#include "state.h"
#include "tcp_type.h"
#include "tqueue.h"

/*
 * User calls
 */

/** OPEN user call
 *
 * @param lport         Local port
 * @param fsock         Foreign socket
 * @param acpass        Active/passive
 * @param conn          Connection
 *
 * XXX We should be able to call active open on an existing listening
 * connection.
 */
tcp_error_t tcp_uc_open(uint16_t lport, tcp_sock_t *fsock, acpass_t acpass,
    tcp_conn_t **conn)
{
        tcp_conn_t *nconn;
        tcp_sock_t lsock;

        log_msg(LVL_DEBUG, "tcp_uc_open(%" PRIu16 ", %p, %s, %p)",
            lport, fsock, acpass == ap_active ? "active" : "passive",
            conn);

        lsock.port = lport;
        lsock.addr.ipv4 = 0x7f000001;

        nconn = tcp_conn_new(&lsock, fsock);
        tcp_conn_add(nconn);

        if (acpass == ap_active) {
                /* Synchronize (initiate) connection */
                tcp_conn_sync(nconn);

                /* Wait for connection to be established or reset */
                log_msg(LVL_DEBUG, "tcp_uc_open: Wait for connection.");
                fibril_mutex_lock(&nconn->cstate_lock);
                while (nconn->cstate == st_syn_sent ||
                    nconn->cstate == st_syn_received) {
                        fibril_condvar_wait(&nconn->cstate_cv, &nconn->cstate_lock);
                }
                if (nconn->cstate != st_established) {
                        log_msg(LVL_DEBUG, "tcp_uc_open: Connection was reset.");
                        assert(nconn->cstate == st_closed);
                        fibril_mutex_unlock(&nconn->cstate_lock);
                        return TCP_ERESET;
                }
                fibril_mutex_unlock(&nconn->cstate_lock);
                log_msg(LVL_DEBUG, "tcp_uc_open: Connection was established.");
        }

        *conn = nconn;
        return TCP_EOK;
}

/** SEND user call */
tcp_error_t tcp_uc_send(tcp_conn_t *conn, void *data, size_t size,
    xflags_t flags)
{
        size_t buf_free;
        size_t xfer_size;

        log_msg(LVL_DEBUG, "tcp_uc_send()");

        if (conn->cstate == st_closed)
                return TCP_ENOTEXIST;

        if (conn->cstate == st_listen) {
                /* Change connection to active */
                tcp_conn_sync(conn);
        }

        if (conn->snd_buf_fin)
                return TCP_ECLOSING;

        while (size > 0) {
                buf_free = conn->snd_buf_size - conn->snd_buf_used;
                while (buf_free == 0)
                        tcp_tqueue_new_data(conn);

                xfer_size = min(size, buf_free);

                /* Copy data to buffer */
                memcpy(conn->snd_buf + conn->snd_buf_used, data, xfer_size);
                data += xfer_size;
                conn->snd_buf_used += xfer_size;
                size -= xfer_size;
        }

        tcp_tqueue_new_data(conn);

        return TCP_EOK;
}

/** RECEIVE user call */
tcp_error_t tcp_uc_receive(tcp_conn_t *conn, void *buf, size_t size,
    size_t *rcvd, xflags_t *xflags)
{
        size_t xfer_size;

        log_msg(LVL_DEBUG, "tcp_uc_receive()");

        if (conn->cstate == st_closed)
                return TCP_ENOTEXIST;

        fibril_mutex_lock(&conn->rcv_buf_lock);

        /* Wait for data to become available */
        while (conn->rcv_buf_used == 0 && !conn->rcv_buf_fin) {
                log_msg(LVL_DEBUG, "tcp_uc_receive() - wait for data");
                fibril_condvar_wait(&conn->rcv_buf_cv, &conn->rcv_buf_lock);
        }

        if (conn->rcv_buf_used == 0) {
                /* End of data, peer closed connection. */
                assert(conn->rcv_buf_fin);
                *rcvd = 0;
                *xflags = 0;
                return TCP_ECLOSING;
        }

        /* Copy data from receive buffer to user buffer */
        xfer_size = min(size, conn->rcv_buf_used);
        memcpy(buf, conn->rcv_buf, xfer_size);
        *rcvd = xfer_size;

        /* Remove data from receive buffer */
        memmove(conn->rcv_buf, conn->rcv_buf + xfer_size, conn->rcv_buf_used -
            xfer_size);
        conn->rcv_buf_used -= xfer_size;
        conn->rcv_wnd += xfer_size;

        fibril_mutex_unlock(&conn->rcv_buf_lock);

        /* TODO */
        *xflags = 0;

        /* Send new size of receive window */
        tcp_tqueue_ctrl_seg(conn, CTL_ACK);

        log_msg(LVL_DEBUG, "tcp_uc_receive() - returning %zu bytes",
            xfer_size);

        return TCP_EOK;
}

/** CLOSE user call */
tcp_error_t tcp_uc_close(tcp_conn_t *conn)
{
        log_msg(LVL_DEBUG, "tcp_uc_close()");

        if (conn->cstate == st_closed)
                return TCP_ENOTEXIST;

        if (conn->snd_buf_fin)
                return TCP_ECLOSING;

        conn->snd_buf_fin = true;
        tcp_tqueue_new_data(conn);

        return TCP_EOK;
}

/** ABORT user call */
void tcp_uc_abort(tcp_conn_t *conn)
{
        log_msg(LVL_DEBUG, "tcp_uc_abort()");
}

/** STATUS user call */
void tcp_uc_status(tcp_conn_t *conn, tcp_conn_status_t *cstatus)
{
        log_msg(LVL_DEBUG, "tcp_uc_status()");
}


/*
 * Arriving segments
 */

/** Segment arrived */
void tcp_as_segment_arrived(tcp_sockpair_t *sp, tcp_segment_t *seg)
{
        tcp_conn_t *conn;

        log_msg(LVL_DEBUG, "tcp_as_segment_arrived()");

        conn = tcp_conn_find(sp);
        if (conn != NULL && conn->cstate != st_closed) {
                tcp_conn_segment_arrived(conn, seg);
        } else {
                tcp_unexpected_segment(sp, seg);
        }
}

/*
 * Timeouts
 */

/** User timeout */
void tcp_to_user(void)
{
        log_msg(LVL_DEBUG, "tcp_to_user()");
}

/**
 * @}
 */