iframe-proxy | Sunbelt Computer Software

History

103 lines (93 loc) · 3.27 KB

Raw

100

101

102

103

// spell-checker:disable

pub(crate) use grp::module_def;

#[pymodule]

mod grp {

use crate::vm::{

PyObjectRef, PyResult, VirtualMachine,

builtins::{PyIntRef, PyListRef, PyUtf8StrRef},

convert::{IntoPyException, ToPyObject},

exceptions,

types::PyStructSequence,

};

use core::ptr::NonNull;

use nix::unistd;

#[pystruct_sequence_data]

struct GroupData {

gr_name: String,

gr_passwd: String,

gr_gid: u32,

gr_mem: PyListRef,

}

#[pyattr]

#[pystruct_sequence(name = "struct_group", module = "grp", data = "GroupData")]

struct PyGroup;

#[pyclass(with(PyStructSequence))]

impl PyGroup {}

impl GroupData {

fn from_unistd_group(group: unistd::Group, vm: &VirtualMachine) -> Self {

let cstr_lossy = |s: alloc::ffi::CString| {

s.into_string()

.unwrap_or_else(|e| e.into_cstring().to_string_lossy().into_owned())

};

GroupData {

gr_name: group.name,

gr_passwd: cstr_lossy(group.passwd),

gr_gid: group.gid.as_raw(),

gr_mem: vm

.ctx

.new_list(group.mem.iter().map(|s| s.to_pyobject(vm)).collect()),

}

#[pyfunction]

fn getgrgid(gid: PyIntRef, vm: &VirtualMachine) -> PyResult<GroupData> {

let gr_gid = gid.as_bigint();

let gid = libc::gid_t::try_from(gr_gid)

.map(unistd::Gid::from_raw)

.ok();

let group = gid

.map(unistd::Group::from_gid)

.transpose()

.map_err(|err| err.into_pyexception(vm))?

.flatten();

let group = group.ok_or_else(|| {

vm.new_key_error(

vm.ctx

.new_str(format!("getgrgid: group id {gr_gid} not found"))

.into(),

)

})?;

Ok(GroupData::from_unistd_group(group, vm))

}

#[pyfunction]

fn getgrnam(name: PyUtf8StrRef, vm: &VirtualMachine) -> PyResult<GroupData> {

let gr_name = name.as_str();

if gr_name.contains('\0') {

return Err(exceptions::cstring_error(vm));

}

let group = unistd::Group::from_name(gr_name).map_err(|err| err.into_pyexception(vm))?;

let group = group.ok_or_else(|| {

vm.new_key_error(

vm.ctx

.new_str(format!("getgrnam: group name {gr_name} not found"))

.into(),

)

})?;

Ok(GroupData::from_unistd_group(group, vm))

}

#[pyfunction]

fn getgrall(vm: &VirtualMachine) -> PyResult<Vec<PyObjectRef>> {

// setgrent, getgrent, etc are not thread safe. Could use fgetgrent_r, but this is easier

static GETGRALL: parking_lot::Mutex<()> = parking_lot::Mutex::new(());

let _guard = GETGRALL.lock();

let mut list = Vec::new();

unsafe { libc::setgrent() };

while let Some(ptr) = NonNull::new(unsafe { libc::getgrent() }) {

let group = unistd::Group::from(unsafe { ptr.as_ref() });

let group = GroupData::from_unistd_group(group, vm).to_pyobject(vm);

list.push(group);

}

unsafe { libc::endgrent() };

Ok(list)

}

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

grp.rs

Latest commit

History

grp.rs

File metadata and controls