Method set_value

archive/ntreg/src/registry_key.rs:309–393

Creates or sets a value for the current key. # Arguments `name` - A string containing the name of the value. # Remarks Returns a RegistryValue representing the new value. # Examples ``` # use ntreg::registry_key::*; # use ntreg::registry_value::*; let key = RegistryKey::new("HKCU").unwrap(); key.set_value("TestValue", &RegistryValueData::String("TestValueData".to_string())); key.delete(false);

(&self, name: &str, value: &RegistryValueData)

    /// key.delete(false);
    /// ```
    pub fn set_value(&self, name: &str, value: &RegistryValueData) -> Result<RegistryValue, NtStatusError> {
        let key = match open_key(&self.path.clone(), KEY_ALL_ACCESS) {
            Ok((key, _)) => key,
            Err(err) => return Err(err),
        };

        let mut data: *mut c_void = null_mut();
        let mut string_data: Vec<u16>;
        let mut binary_data: Vec<u8>;
        let mut len: usize = 0;
        match value {
            RegistryValueData::None => {}
            RegistryValueData::String(ref string) | RegistryValueData::ExpandString(ref string) | RegistryValueData::Link(ref string) => {
                string_data = string.encode_utf16().collect();
                string_data.push(0);
                data = string_data.as_mut_ptr() as *mut c_void;
                len = string_data.len() * size_of::<u16>();
            }
            RegistryValueData::MultiString(ref multi_string) => {
                string_data = Vec::new();
                for s in multi_string {
                    let mut s: Vec<u16> = s.encode_utf16().collect();
                    s.push(0);
                    string_data.extend(s)
                }
                string_data.push(0);
                data = string_data.as_mut_ptr() as *mut c_void;
                len = string_data.len() * size_of::<u16>();
            }
            RegistryValueData::Binary(ref binary) | RegistryValueData::FullResourceDescriptor(ref binary) |
                RegistryValueData::ResourceList(ref binary) | RegistryValueData::ResourceRequirementsList(ref binary) => {
                binary_data = binary.clone();
                data = binary_data.as_mut_ptr() as *mut c_void;
                len = binary_data.len();
            }
            RegistryValueData::DWord(ref value) => {
                let mut value = *value;
                data = &mut value as *mut u32 as *mut c_void;
                len = size_of::<u32>();
            }
            RegistryValueData::QWord(ref value) => {
                let mut value = *value;
                data = &mut value as *mut u64 as *mut c_void;
                len = size_of::<u64>();
            }
        };

        let data_type_value: u32 = match value {
            RegistryValueData::None => 0,
            RegistryValueData::String(_) => 1,
            RegistryValueData::ExpandString(_) => 2,
            RegistryValueData::Binary(_) => 3,
            RegistryValueData::DWord(_) => 4,
            RegistryValueData::Link(_) => 6,
            RegistryValueData::MultiString(_) => 7,
            RegistryValueData::ResourceList(_) => 8,
            RegistryValueData::FullResourceDescriptor(_) => 9,
            RegistryValueData::ResourceRequirementsList(_) => 10,