diff options
Diffstat (limited to 'src/kiro-client.c')
| -rw-r--r-- | src/kiro-client.c | 32 |
1 files changed, 30 insertions, 2 deletions
diff --git a/src/kiro-client.c b/src/kiro-client.c index 92fd6b1..b9c33c5 100644 --- a/src/kiro-client.c +++ b/src/kiro-client.c @@ -424,7 +424,7 @@ fail: int -kiro_client_sync (KiroClient *self) +kiro_client_sync_partial (KiroClient *self, gulong remote_offset, gulong size, gulong local_offset) { g_return_val_if_fail (self != NULL, -1); KiroClientPrivate *priv = KIRO_CLIENT_GET_PRIVATE (self); @@ -436,8 +436,29 @@ kiro_client_sync (KiroClient *self) struct kiro_connection_context *ctx = (struct kiro_connection_context *)priv->conn->context; + if (remote_offset > ctx->peer_mr.length) { + g_warning ("kiro_client_sync_partial: remote_offset too large! Won't sync."); + return -1; + } + + gulong read_size = ctx->peer_mr.length; + if (size > 0) + read_size = size; + else if (remote_offset > 0) + read_size -= remote_offset; //read to the end of the memory, starting at offset + + if ((remote_offset + read_size) > ctx->peer_mr.length) { + g_warning ("kiro_client_sync_partial: remote_offset + read_size would exceed remote memory boundary! Won't sync."); + return -1; + } + + if ((local_offset + read_size) > ctx->rdma_mr->size) { + g_warning ("kiro_client_sync_partial: local_offset + read_size would exceed local memory boundary! Won't sync."); + return -1; + } + G_LOCK (sync_lock); - if (rdma_post_read (priv->conn, priv->conn, ctx->rdma_mr->mem, ctx->peer_mr.length, ctx->rdma_mr->mr, 0, (uint64_t)ctx->peer_mr.addr, ctx->peer_mr.rkey)) { + if (rdma_post_read (priv->conn, priv->conn, ctx->rdma_mr->mem + local_offset, read_size, ctx->rdma_mr->mr, 0, (uint64_t)ctx->peer_mr.addr + remote_offset, ctx->peer_mr.rkey)) { g_critical ("Failed to RDMA_READ from server: %s", strerror (errno)); goto fail; } @@ -470,6 +491,13 @@ fail: } +int +kiro_client_sync (KiroClient *self) +{ + return kiro_client_sync_partial (self, 0, 0, 0); +} + + gboolean ping_timeout (gpointer data) { |