Fix types

trailofbits · Oct 1, 2024 · c347055 · c347055
1 parent 1dfc598
commit c347055
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/rust/src/lib.rs b/rust/src/lib.rs
@@ -57,6 +57,7 @@ impl TimeStampReq {
 
     #[getter]
     fn message_imprint(&self) -> PyResult<PyMessageImprint> {
+    // fn message_imprint<'p>(&self, py: pyo3::Python<'p>) -> PyResult<PyMessageImprint> {
         Ok(PyMessageImprint {
             contents: OwnedMessageImprint::try_new(Arc::clone(&self.raw), |v| {
                 Ok::<_, ()>(v.borrow_dependent().message_imprint.clone())
@@ -80,6 +81,7 @@ impl TimeStampReq {
 self_cell::self_cell!(
     struct OwnedMessageImprint {
         owner: Arc<OwnedTimeStamReq>,
+        // owner: pyo3::Py<pyo3::types::PyBytes>,
         #[covariant]
         dependent: RawMessageImprint,
     }

diff --git a/rust/src/util.rs b/rust/src/util.rs
@@ -64,8 +64,8 @@ pub(crate) fn datetime_to_py_utc<'p>(
     py: pyo3::Python<'p>,
     dt: &asn1::DateTime,
 ) -> pyo3::PyResult<pyo3::Bound<'p, pyo3::PyAny>> {
-    let timezone = types::DATETIME_TIMEZONE_UTC.get(py)?;
-    types::DATETIME_DATETIME.get(py)?.call1((
+    let timezone = DATETIME_TIMEZONE_UTC.get(py)?;
+    DATETIME_DATETIME.get(py)?.call1((
         dt.year(),
         dt.month(),
         dt.day(),