Method next

archive/ntreg/src/registry_value.rs:82–127

(&mut self)

impl Iterator for RegistryValues {
    type Item = RegistryValue;
    fn next(&mut self) -> Option<Self::Item> {
        if self.index >= self.key.sub_key_count {
            return None;
        }

        let max_value_len = self.key.max_value_name_len + 1;
        let max_data_len = self.key.max_data_len + 1;
        let buf_len = size_of::<KEY_VALUE_FULL_INFORMATION>() as u32 + max_value_len + max_data_len;
        let mut len: u32 = buf_len;
        let buf: Vec<u8> = vec![0; len as usize];
        let value_information: PKEY_VALUE_FULL_INFORMATION = buf.as_ptr() as PKEY_VALUE_FULL_INFORMATION;
        let status = unsafe {
            ntregapi::NtEnumerateValueKey(
                self.key.handle,
                self.index,
                ntregapi::KeyValueFullInformation,
                value_information as *mut c_void,
                len,
                &mut len
            )
        };

        if status == ERROR_NO_MORE_ITEMS {
            return None;
        }

        if !NT_SUCCESS(status) {
            return None;
        }

        let name_length = unsafe {
            ((*value_information).NameLength / size_of::<u16>() as u32) as usize
        };

        let reg_value_name = String::from_utf16_lossy( unsafe {
            std::slice::from_raw_parts((*value_information).Name.as_ptr(), name_length)
        });

        self.index += 1;

        Some(RegistryValue {
            key_path: self.key.path.clone(),
            name: if name_length > 0 { reg_value_name } else { "".to_string() },
            data: get_data_value(value_information),
        })
    }
}

impl fmt::Display for RegistryValueData {