ssize_t sock_comm_flush(struct sock_pe_entry *pe_entry)
{
ssize_t ret1, ret2 = 0;
size_t endlen, len, xfer_len;
len = ofi_rbused(&pe_entry->comm_buf);
endlen = pe_entry->comm_buf.size -
(pe_entry->comm_buf.rcnt & pe_entry->comm_buf.size_mask);
xfer_len = MIN(len, endlen);
ret1 = sock_comm_send_socket(pe_entry->conn, (char*)pe_entry->comm_buf.buf +
(pe_entry->comm_buf.rcnt & pe_entry->comm_buf.size_mask),
xfer_len);
if (ret1 > 0)
pe_entry->comm_buf.rcnt += ret1;
if (ret1 == xfer_len && xfer_len < len) {
ret2 = sock_comm_send_socket(pe_entry->conn, (char*)pe_entry->comm_buf.buf +
(pe_entry->comm_buf.rcnt & pe_entry->comm_buf.size_mask),
len - xfer_len);
if (ret2 > 0)
pe_entry->comm_buf.rcnt += ret2;
else
ret2 = 0;
}
return (ret1 > 0) ? ret1 + ret2 : 0;
}
ssize_t sock_comm_flush(struct sock_conn *conn)
{
ssize_t ret1, ret2 = 0;
size_t endlen, len, xfer_len;
len = rbused(&conn->outbuf);
endlen = conn->outbuf.size - (conn->outbuf.rcnt & conn->outbuf.size_mask);
xfer_len = MIN(len, endlen);
ret1 = sock_comm_send_socket(conn, conn->outbuf.buf +
(conn->outbuf.rcnt & conn->outbuf.size_mask),
xfer_len);
if (ret1 > 0)
conn->outbuf.rcnt += ret1;
if (ret1 == xfer_len && xfer_len < len) {
ret2 = sock_comm_send_socket(conn, conn->outbuf.buf +
(conn->outbuf.rcnt & conn->outbuf.size_mask),
len - xfer_len);
if (ret2 > 0)
conn->outbuf.rcnt += ret2;
else
ret2 = 0;
}
return (ret1 > 0) ? ret1 + ret2 : 0;
}
ssize_t sock_comm_send(struct sock_pe_entry *pe_entry,
const void *buf, size_t len)
{
ssize_t ret, used;
if (len > pe_entry->cache_sz) {
used = ofi_rbused(&pe_entry->comm_buf);
if (used == sock_comm_flush(pe_entry)) {
return sock_comm_send_socket(pe_entry->conn, buf, len);
} else {
return 0;
}
}
if (ofi_rbavail(&pe_entry->comm_buf) < len) {
ret = sock_comm_flush(pe_entry);
if (ret <= 0)
return 0;
}
ret = MIN(ofi_rbavail(&pe_entry->comm_buf), len);
ofi_rbwrite(&pe_entry->comm_buf, buf, ret);
ofi_rbcommit(&pe_entry->comm_buf);
SOCK_LOG_DBG("buffered %lu\n", ret);
return ret;
}
ssize_t sock_comm_send(struct sock_conn *conn, const void *buf, size_t len)
{
ssize_t ret, used;
if (len >= SOCK_COMM_THRESHOLD) {
used = rbused(&conn->outbuf);
if (used == sock_comm_flush(conn)) {
return sock_comm_send_socket(conn, buf, len);
} else {
return 0;
}
}
if (rbavail(&conn->outbuf) < len) {
ret = sock_comm_flush(conn);
if (ret <= 0)
return 0;
}
ret = MIN(rbavail(&conn->outbuf), len);
rbwrite(&conn->outbuf, buf, ret);
rbcommit(&conn->outbuf);
SOCK_LOG_INFO("buffered %lu\n", ret);
return ret;
}
