fix: update package metadata and repository links

Author: ZR233

Cargo.toml

@@ -1,19 +1,21 @@
 [package]
-name = "dma-api"
-version = "0.3.0"
-edition = "2021"
 authors = ["周睿 <zrufo747@outlook.com>"]
-repository = "https://github.com/ZR233/dma-api"
-license = "MIT"
+categories = ["embedded", "no-std"]
 description = "Trait for DMA alloc and some collections"
+edition = "2021"
 keywords = ["os", "dma"]
-categories = ["embedded", "no-std"]
+license = "MIT"
+name = "dma-api"
+repository = "https://github.com/drivercraft/dma-api"
+version = "0.3.1"
 
 [features]
 alloc = []
 
 [dependencies]
 
+[target.'cfg(target_arch = "aarch64")'.dependencies]
+aarch64-cpu-ext = "0.1"
 
 [[test]]
 name = "test"

README.md

@@ -1,6 +1,6 @@
 # DMA API
 
-[![Rust](https://github.com/ZR233/dma-api/actions/workflows/rust.yml/badge.svg?branch=main)](https://github.com/ZR233/dma-api/actions/workflows/rust.yml)
+[![Rust](https://github.com/drivercraft/dma-api/actions/workflows/rust.yml/badge.svg?branch=main)](https://github.com/drivercraft/dma-api/actions/workflows/rust.yml)
 
 ## Example
 

src/dma/alloc/mod.rs

@@ -1,4 +1,3 @@
-#[cfg(feature = "alloc")]
 use alloc::vec::Vec;
 use core::{
     alloc::Layout,
@@ -34,7 +33,6 @@ impl<T> DCommon<T> {
         }
     }
 
-    #[cfg(feature = "alloc")]
     pub fn from_vec(mut value: Vec<T>, direction: Direction) -> Self {
         unsafe {
             let layout = Layout::from_size_align_unchecked(

src/dma/mod.rs

@@ -1,8 +1,9 @@
 use crate::{flush, invalidate, Direction};
 use core::ptr::NonNull;
 
-pub mod slice;
+#[cfg(feature = "alloc")]
 pub mod alloc;
+pub mod slice;
 
 impl Direction {
     pub fn preper_read(self, ptr: NonNull<u8>, size: usize) {

src/lib.rs

@@ -1,17 +1,19 @@
 #![cfg_attr(not(test), no_std)]
 #![doc = include_str!("../README.md")]
 
-// #[cfg(feature = "alloc")]
+#[cfg(feature = "alloc")]
 extern crate alloc;
 
-use core::{alloc::Layout, ptr::NonNull};
+use core::ptr::NonNull;
 
 mod dma;
 
+#[cfg(feature = "alloc")]
 pub use dma::alloc::{r#box::DBox, vec::DVec};
 pub use dma::slice::{DSlice, DSliceMut};
 
 #[derive(Debug, Clone, Copy, PartialEq, Eq)]
+#[repr(C)]
 pub enum Direction {
     ToDevice,
     FromDevice,
@@ -33,14 +35,10 @@ pub trait Impl {
     /// # Safety
     ///
     /// layout must have non-zero size. Attempting to allocate for a zero-sized layout may result in undefined behavior.
+    #[cfg(feature = "alloc")]
     #[allow(unused_variables)]
-    unsafe fn alloc(layout: Layout) -> *mut u8 {
-        #[cfg(feature = "alloc")]
-        unsafe {
-            alloc::alloc::alloc(layout)
-        }
-        #[cfg(not(feature = "alloc"))]
-        core::ptr::null_mut()
+    unsafe fn alloc(layout: core::alloc::Layout) -> *mut u8 {
+        unsafe { alloc::alloc::alloc(layout) }
     }
 
     /// Deallocates the block of memory at the given `ptr` pointer with the given `layout`.
@@ -55,12 +53,9 @@ pub trait Impl {
     ///   memory.
     ///
     /// Otherwise undefined behavior can result.
-    #[allow(unused_variables)]
-    unsafe fn dealloc(ptr: *mut u8, layout: Layout) {
-        #[cfg(feature = "alloc")]
-        unsafe {
-            alloc::alloc::dealloc(ptr, layout)
-        }
+    #[cfg(feature = "alloc")]
+    unsafe fn dealloc(ptr: *mut u8, layout: core::alloc::Layout) {
+        unsafe { alloc::alloc::dealloc(ptr, layout) }
     }
 }
 
@@ -69,8 +64,10 @@ extern "Rust" {
     fn __dma_api_unmap(addr: NonNull<u8>, size: usize);
     fn __dma_api_flush(addr: NonNull<u8>, size: usize);
     fn __dma_api_invalidate(addr: NonNull<u8>, size: usize);
-    fn __dma_api_alloc(layout: Layout) -> *mut u8;
-    fn __dma_api_dealloc(ptr: *mut u8, layout: Layout);
+    #[cfg(feature = "alloc")]
+    fn __dma_api_alloc(layout: core::alloc::Layout) -> *mut u8;
+    #[cfg(feature = "alloc")]
+    fn __dma_api_dealloc(ptr: *mut u8, layout: core::alloc::Layout);
 }
 
 fn map(addr: NonNull<u8>, size: usize, direction: Direction) -> u64 {
@@ -88,17 +85,17 @@ fn flush(addr: NonNull<u8>, size: usize) {
 fn invalidate(addr: NonNull<u8>, size: usize) {
     unsafe { __dma_api_invalidate(addr, size) }
 }
-
-unsafe fn alloc(layout: Layout) -> *mut u8 {
+#[cfg(feature = "alloc")]
+unsafe fn alloc(layout: core::alloc::Layout) -> *mut u8 {
     unsafe { __dma_api_alloc(layout) }
 }
-
-unsafe fn dealloc(ptr: *mut u8, layout: Layout) {
+#[cfg(feature = "alloc")]
+unsafe fn dealloc(ptr: *mut u8, layout: core::alloc::Layout) {
     unsafe { __dma_api_dealloc(ptr, layout) }
 }
 
 #[macro_export]
-macro_rules! set_impl {
+macro_rules! __set_impl_base {
     ($t: ty) => {
         #[no_mangle]
         fn __dma_api_map(
@@ -120,6 +117,22 @@ macro_rules! set_impl {
         fn __dma_api_invalidate(addr: core::ptr::NonNull<u8>, size: usize) {
             <$t as $crate::Impl>::invalidate(addr, size)
         }
+    };
+}
+
+#[cfg(not(feature = "alloc"))]
+#[macro_export]
+macro_rules! set_impl {
+    ($t: ty) => {
+        $crate::__set_impl_base!($t);
+    };
+}
+
+#[cfg(feature = "alloc")]
+#[macro_export]
+macro_rules! set_impl {
+    ($t: ty) => {
+        $crate::__set_impl_base!($t);
         #[no_mangle]
         fn __dma_api_alloc(layout: core::alloc::Layout) -> *mut u8 {
             unsafe { <$t as $crate::Impl>::alloc(layout) }