Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/datenlord/async-rdma

Easy to use RDMA API in Rust async
https://github.com/datenlord/async-rdma

async rust

Last synced: 3 days ago
JSON representation

Easy to use RDMA API in Rust async

Awesome Lists containing this project

README

        

# async-rdma

A framework for writing RDMA applications with high-level abstraction and asynchronous APIs.

[![Join the chat at https://gitter.im/datenlord/async-rdma](https://badges.gitter.im/datenlord/async-rdma.svg)](https://gitter.im/datenlord/async-rdma?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge)
[![Crates.io][crates-badge]][crates-url]
[![Docs][docs-badge]][docs-url]
[![GPL licensed][gpl-badge]][gpl-url]
[![Build Status][actions-badge]][actions-url]

[crates-badge]: https://img.shields.io/crates/v/async-rdma.svg
[crates-url]: https://crates.io/crates/async-rdma
[docs-badge]: https://docs.rs/async-rdma/badge.svg
[docs-url]: https://docs.rs/async-rdma/
[gpl-badge]: https://img.shields.io/badge/license-GPLv3.0-blue.svg
[gpl-url]: https://github.com/datenlord/async-rdma/blob/master/LICENSE
[actions-badge]: https://github.com/datenlord/async-rdma/actions/workflows/ci.yml/badge.svg
[actions-url]: https://github.com/datenlord/async-rdma/actions

It provides a few major components:

* Tools for establishing connections with rdma endpoints such as `RdmaBuilder`.

* High-level APIs for data transmission between endpoints including `read`,
`write`, `send`, `receive`.

* High-level APIs for rdma memory region management including `alloc_local_mr`,
`request_remote_mr`, `send_mr`, `receive_local_mr`, `receive_remote_mr`.

* A framework including `agent` and `event_listener` working behind APIs for memory
region management and executing rdma requests such as `post_send` and `poll`.

The [ChangeLog] file contains a brief summary of changes for each release.

[ChangeLog]: https://github.com/datenlord/async-rdma/blob/master/ChangeLog.md

## Environment Setup

This section is for RDMA novices who want to try this library.

You can skip if your Machines have been configured with RDMA.

Next we will configure the RDMA environment in an Ubuntu20.04 VM.
If you are using another operating system distribution, please search and replace the relevant commands.

### 1. Check whether the current kernel supports RXE

Run the following command and if the CONFIG_RDMA_RXE = `y` or `m`, the current operating system supports RXE.
If not you need to search how to recompile the kernel to support RDMA.

```shell
cat /boot/config-$(uname -r) | grep RXE
```

### 2. Install Dependencies

```shell
sudo apt install -y libibverbs1 ibverbs-utils librdmacm1 libibumad3 ibverbs-providers rdma-core libibverbs-dev iproute2 perftest build-essential net-tools git librdmacm-dev rdmacm-utils cmake libprotobuf-dev protobuf-compiler clang curl
```

### 3. Configure RDMA netdev

(1) Load kernel driver

```shell
modprobe rdma_rxe
```

(2) User mode RDMA netdev configuration.

```shell
sudo rdma link add rxe_0 type rxe netdev ens33
```

`rxe_0` is the RDMA device name, and you can name it whatever you want. `ens33` is the name of the network device. The name of the network device may be different in each VM, and we can see it by running command "ifconfig".

(3) Check the RDMA device state

Run the following command and check if the state is `ACTIVE`.

```shell
rdma link
```

(4) Test it

Ib_send_bw is a program used to test the bandwidth of `RDMA SEND` operations.

Run the following command in a terminal.

```shell
ib_send_bw -d rxe_0
```

And run the following command in another terminal.

```shell
ib_send_bw -d rxe_0 localhost
```

### 4. Install Rust

```shell
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
source $HOME/.cargo/env
```

### 5. Try an example

```shell
git clone https://github.com/datenlord/async-rdma.git
cd async-rdma
cargo run --example rpc
```

if run rpc example failed, you can try run it with sudo permission.

```shell
cargo build --example rpc
sudo ./target/debug/examples/rpc
```

## Example

A simple example: client request a remote memory region and put data into this remote
memory region by rdma `write`.
And finally client `send_mr` to make server aware of this memory region.
Server `receive_local_mr`, and then get data from this mr.

```rust
use async_rdma::{LocalMrReadAccess, LocalMrWriteAccess, RdmaBuilder};
use portpicker::pick_unused_port;
use std::{
alloc::Layout,
io::{self, Write},
net::{Ipv4Addr, SocketAddrV4},
time::Duration,
};

async fn client(addr: SocketAddrV4) -> io::Result<()> {
let layout = Layout::new::<[u8; 8]>();
let rdma = RdmaBuilder::default().connect(addr).await?;
// alloc 8 bytes remote memory
let mut rmr = rdma.request_remote_mr(layout).await?;
// alloc 8 bytes local memory
let mut lmr = rdma.alloc_local_mr(layout)?;
// write data into lmr
let _num = lmr.as_mut_slice().write(&[1_u8; 8])?;
// write the second half of the data in lmr to the rmr
rdma.write(&lmr.get(4..8).unwrap(), &mut rmr.get_mut(4..8).unwrap())
.await?;
// send rmr's meta data to the remote end
rdma.send_remote_mr(rmr).await?;
Ok(())
}

#[tokio::main]
async fn server(addr: SocketAddrV4) -> io::Result<()> {
let rdma = RdmaBuilder::default().listen(addr).await?;
// receive mr's meta data from client
let lmr = rdma.receive_local_mr().await?;
let data = *lmr.as_slice();
println!("Data written by the client using RDMA WRITE: {:?}", data);
assert_eq!(data, [[0_u8; 4], [1_u8; 4]].concat());
Ok(())
}

#[tokio::main]
async fn main() {
let addr = SocketAddrV4::new(Ipv4Addr::new(127, 0, 0, 1), pick_unused_port().unwrap());
std::thread::spawn(move || server(addr));
tokio::time::sleep(Duration::new(1, 0)).await;
client(addr)
.await
.map_err(|err| println!("{}", err))
.unwrap();
}

```

## Getting Help

First, see if the answer to your question can be found in the found [API doc] or [Design doc]. If the answer is not here, please open an issue and describe your problem in detail.

[Design doc]: https://github.com/datenlord/async-rdma/tree/master/doc
[API doc]: https://docs.rs/async-rdma/0.2.0/async_rdma/

## Related Projects

* [`rdma-sys`]: Rust bindings for RDMA fundamental libraries: libibverbs-dev and librdmacm-dev.

[`rdma-sys`]: https://github.com/datenlord/rdma-sys