Resolve rust-lang#962 - implement --no-copy with tests

Author: hallfox

src/ir/analysis/derive_copy.rs

@@ -143,6 +143,10 @@ impl<'ctx> MonotoneFramework for CannotDeriveCopy<'ctx> {
             }
         };
 
+        if self.ctx.no_copy_by_name(&item) {
+            return self.insert(id);
+        }
+
         if item.is_opaque(self.ctx, &()) {
             let layout_can_derive = ty.layout(self.ctx).map_or(true, |l| {
                 l.opaque().can_trivially_derive_copy()

src/ir/context.rs

@@ -2450,8 +2450,16 @@ impl BindgenContext {
     /// Check if `--no-partialeq` flag is enabled for this item.
     pub fn no_partialeq_by_name(&self, item: &Item) -> bool {
         let name = item.canonical_path(self)[1..].join("::");
+
         self.options().no_partialeq_types.matches(&name)
     }
+
+    /// Check if `--no-copy` flag is enabled for this item.
+    pub fn no_copy_by_name(&self, item: &Item) -> bool {
+        let name = item.canonical_path(self)[1..].join("::");
+
+        self.options().no_copy_types.matches(&name)
+    }
 }
 
 /// A builder struct for configuring item resolution options.

src/lib.rs

@@ -522,6 +522,20 @@ impl Builder {
             })
             .count();
 
+        self.options
+            .no_copy_types
+            .get_items()
+            .iter()
+            .map(|item| {
+                output_vector.push("--no-copy".into());
+                output_vector.push(
+                    item.trim_left_matches("^")
+                        .trim_right_matches("$")
+                        .into(),
+                );
+            })
+            .count();
+
         output_vector
     }
 
@@ -1158,6 +1172,13 @@ impl Builder {
         self.options.no_partialeq_types.insert(arg);
         self
     }
+
+    /// Don't derive `Copy` for a given type. Regular
+    /// expressions are supported.
+    pub fn no_copy(mut self, arg: String) -> Self {
+        self.options.no_copy_types.insert(arg);
+        self
+    }
 }
 
 /// Configuration options for generated bindings.
@@ -1345,6 +1366,9 @@ struct BindgenOptions {
 
     /// The set of types that we should not derive `PartialEq` for.
     no_partialeq_types: RegexSet,
+
+    /// The set of types that we should not derive `Copy` for.
+    no_copy_types: RegexSet,
 }
 
 /// TODO(emilio): This is sort of a lie (see the error message that results from
@@ -1363,6 +1387,7 @@ impl BindgenOptions {
         self.constified_enum_modules.build();
         self.rustified_enums.build();
         self.no_partialeq_types.build();
+        self.no_copy_types.build();
     }
 
     /// Update rust target version
@@ -1434,6 +1459,7 @@ impl Default for BindgenOptions {
             rustfmt_bindings: true,
             rustfmt_configuration_file: None,
             no_partialeq_types: Default::default(),
+            no_copy_types: Default::default(),
         }
     }
 }

src/options.rs

@@ -285,6 +285,13 @@ where
                 .takes_value(true)
                 .multiple(true)
                 .number_of_values(1),
+            Arg::with_name("no-copy")
+                .long("no-copy")
+                .help("Avoid deriving Copy for types matching <regex>.")
+                .value_name("regex")
+                .takes_value(true)
+                .multiple(true)
+                .number_of_values(1),
         ]) // .args()
         .get_matches_from(args);
 
@@ -579,6 +586,12 @@ where
         }
     }
 
+    if let Some(no_partialeq) = matches.values_of("no-copy") {
+        for regex in no_partialeq {
+            builder = builder.no_copy(String::from(regex));
+        }
+    }
+
     let verbose = matches.is_present("verbose");
 
     Ok((builder, output, verbose))

tests/headers/no-copy-opaque.hpp

@@ -0,0 +1,5 @@
+// bindgen-flags: --with-derive-copy --opaque-type "NoCopy" --no-copy "NoCopy"
+
+class NoCopy {
+  int i;
+};

tests/headers/no-copy-whitelisted.h

@@ -0,0 +1,5 @@
+// bindgen-flags: --with-derive-copy --whitelist-type "NoCopy" --no-copy "NoCopy"
+
+class NoCopy {
+  int i;
+};

tests/headers/whitelisted-item-references-no-copy.hpp

@@ -0,0 +1,7 @@
+// bindgen-flags: --with-derive-copy --whitelist-type "WhitelistMe" --no-copy "NoCopy"
+
+struct NoCopy {};
+
+class WhitelistMe {
+  NoCopy a;
+};