support no_std for tls_codec_derive

RustCrypto · Jul 2, 2023 · fb4b494 · fb4b494
1 parent 9c745bd
commit fb4b494
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/tls_codec/Cargo.toml b/tls_codec/Cargo.toml
@@ -31,7 +31,7 @@ arbitrary = [ "std", "dep:arbitrary" ]
 derive = [ "std", "tls_codec_derive" ]
 serde = [ "std", "dep:serde" ]
 mls = [] # In MLS variable length vectors are limited compared to QUIC.
-std = []
+std = [ "tls_codec_derive?/std" ]
 
 [[bench]]
 name = "tls_vec"

diff --git a/tls_codec/derive/Cargo.toml b/tls_codec/derive/Cargo.toml
@@ -21,3 +21,7 @@ proc-macro2 = "1.0"
 [dev-dependencies]
 tls_codec = { path = "../" }
 trybuild = "1"
+
+[features]
+default = [ "std" ]
+std = []
diff --git a/tls_codec/derive/src/lib.rs b/tls_codec/derive/src/lib.rs
@@ -698,7 +698,7 @@ fn impl_tls_size(parsed_ast: TlsStruct) -> TokenStream2 {
                         let field_len = match self {
                             #(#field_arms)*
                         };
-                        std::mem::size_of::<#repr>() + field_len
+                        core::mem::size_of::<#repr>() + field_len
                     }
                 }
 
@@ -740,6 +740,7 @@ fn impl_serialize(parsed_ast: TlsStruct, svariant: SerializeVariant) -> TokenStr
                 SerializeVariant::Write => {
                     quote! {
                         impl #impl_generics tls_codec::Serialize for #ident #ty_generics #where_clause {
+                            #[cfg(feature = "std")]
                             fn tls_serialize<W: std::io::Write>(&self, writer: &mut W) -> core::result::Result<usize, tls_codec::Error> {
                                 let mut written = 0usize;
                                 #(
@@ -760,6 +761,7 @@ fn impl_serialize(parsed_ast: TlsStruct, svariant: SerializeVariant) -> TokenStr
                         }
 
                         impl #impl_generics tls_codec::Serialize for &#ident #ty_generics #where_clause {
+                            #[cfg(feature = "std")]
                             fn tls_serialize<W: std::io::Write>(&self, writer: &mut W) -> core::result::Result<usize, tls_codec::Error> {
                                 tls_codec::Serialize::tls_serialize(*self, writer)
                             }
@@ -850,6 +852,7 @@ fn impl_serialize(parsed_ast: TlsStruct, svariant: SerializeVariant) -> TokenStr
                 SerializeVariant::Write => {
                     quote! {
                         impl #impl_generics tls_codec::Serialize for #ident #ty_generics #where_clause {
+                            #[cfg(feature = "std")]
                             fn tls_serialize<W: std::io::Write>(&self, writer: &mut W) -> core::result::Result<usize, tls_codec::Error> {
                                 #discriminant_constants
                                 match self {
@@ -859,6 +862,7 @@ fn impl_serialize(parsed_ast: TlsStruct, svariant: SerializeVariant) -> TokenStr
                         }
 
                         impl #impl_generics tls_codec::Serialize for &#ident #ty_generics #where_clause {
+                            #[cfg(feature = "std")]
                             fn tls_serialize<W: std::io::Write>(&self, writer: &mut W) -> core::result::Result<usize, tls_codec::Error> {
                                 tls_codec::Serialize::tls_serialize(*self, writer)
                             }
@@ -909,6 +913,7 @@ fn impl_deserialize(parsed_ast: TlsStruct) -> TokenStream2 {
             let (impl_generics, ty_generics, where_clause) = generics.split_for_impl();
             quote! {
                 impl #impl_generics tls_codec::Deserialize for #ident #ty_generics #where_clause {
+                    #[cfg(feature = "std")]
                     fn tls_deserialize<R: std::io::Read>(bytes: &mut R) -> core::result::Result<Self, tls_codec::Error> {
                         Ok(Self {
                             #(#members: #prefixes::tls_deserialize(bytes)?,)*
@@ -948,6 +953,7 @@ fn impl_deserialize(parsed_ast: TlsStruct) -> TokenStream2 {
             quote! {
                 impl #impl_generics tls_codec::Deserialize for #ident #ty_generics #where_clause {
                     #[allow(non_upper_case_globals)]
+                    #[cfg(feature = "std")]
                     fn tls_deserialize<R: std::io::Read>(bytes: &mut R) -> core::result::Result<Self, tls_codec::Error> {
                         #discriminant_constants
                         let discriminant = <#repr as tls_codec::Deserialize>::tls_deserialize(bytes)?;