add immutable flag to heaptype, check immutable flag in SetAttr

lijm1358 · lijm1358 · commit 076e8bcc1bea · 2022-06-26T15:37:32.000+09:00
diff --git a/vm/src/builtins/type.rs b/vm/src/builtins/type.rs
@@ -104,6 +104,10 @@ impl PyType {
 
         *slots.name.get_mut() = Some(String::from(name));
 
+        if !slots.flags.has_feature(PyTypeFlags::HEAPTYPE) {
+            slots.flags |= PyTypeFlags::IMMUTABLETYPE
+        }
+
         let new_type = PyRef::new_ref(
             PyType {
                 base: Some(base),
@@ -707,6 +711,14 @@ impl SetAttr for PyType {
         value: Option<PyObjectRef>,
         vm: &VirtualMachine,
     ) -> PyResult<()> {
+        if zelf.slots.flags.has_feature(PyTypeFlags::IMMUTABLETYPE) {
+            return Err(vm.new_type_error(format!(
+                "cannot set '{}' attribute of immutable type '{}'",
+                attr_name,
+                zelf.name()
+            )));
+        }
+
         // TODO: pass PyRefExact instead of &str
         let attr_name = vm.ctx.intern_str(attr_name.as_str());
         if let Some(attr) = zelf.get_class_attr(attr_name) {
