This commit is contained in:
Bert Belder 2019-10-08 01:56:41 +02:00
commit fdfa52f4bf
No known key found for this signature in database
GPG Key ID: 7A77887B2E2ED461
6 changed files with 267 additions and 0 deletions

2
.gitignore vendored Normal file
View File

@ -0,0 +1,2 @@
/target
**/*.rs.bk

3
.vscode/settings.json vendored Normal file
View File

@ -0,0 +1,3 @@
{
"rust.clippy_preference": "on"
}

6
Cargo.lock generated Normal file
View File

@ -0,0 +1,6 @@
# This file is automatically @generated by Cargo.
# It is not intended for manual editing.
[[package]]
name = "test4"
version = "0.1.0"

9
Cargo.toml Normal file
View File

@ -0,0 +1,9 @@
[package]
name = "test4"
version = "0.1.0"
authors = ["Bert Belder <bertbelder@gmail.com>"]
edition = "2018"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[dependencies]

1
_git2_a11280 Symbolic link
View File

@ -0,0 +1 @@
testing

246
src/main.rs Normal file
View File

@ -0,0 +1,246 @@
mod channel {
use super::util;
extern "C" {
// Call a method/destructor; virtual methods use C++ dynamic dispatch.
fn Channel__DTOR(this: &mut Channel) -> ();
fn Channel__a(this: &mut Channel) -> ();
fn Channel__b(this: &Channel) -> i32;
// Call a method of a specific class implementation, bypassing dynamic
// dispatch. C++ equivalent: `my_channel.Channel::a()`.
fn Channel__Channel__a(this: &mut Channel) -> ();
// Constructs a special class derived from Channel that forwards all
// virtual method invocations to rust. It is assumed that this subclass
// has the same size and memory layout as the class it's deriving from.
fn Channel__OVERRIDE__CTOR(this: &mut std::mem::MaybeUninit<Channel>) -> ();
}
#[repr(C)]
pub struct Channel {
_cxx_vtable: *const [usize; 0],
}
#[allow(dead_code)]
impl Channel {
pub fn a(&mut self) {
unsafe { Channel__a(self) }
}
pub fn b(&self) -> i32 {
unsafe { Channel__b(self) }
}
}
pub struct ChannelDefaults;
impl ChannelDefaults {
pub fn a(this: &mut Channel) {
unsafe { Channel__Channel__a(this) }
}
}
pub trait ChannelOverrides {
fn base(&self) -> &Override;
fn base_mut(&mut self) -> &mut Override;
fn a(&mut self) {
ChannelDefaults::a(self.base_mut())
}
fn b(&self) -> i32;
}
pub struct Override {
cxx_channel: Channel,
base_offset: usize,
rust_vtable: util::RustVTable<&'static dyn ChannelOverrides>,
}
#[no_mangle]
unsafe extern "C" fn Channel__OVERRIDE__a__DISPATCH(this: &mut Channel) {
Override::dispatch_mut(this).a()
}
#[no_mangle]
unsafe extern "C" fn Channel__OVERRIDE__b__DISPATCH(this: &Channel) -> i32 {
Override::dispatch(this).b()
}
impl Drop for Channel {
fn drop(&mut self) {
unsafe { Channel__DTOR(self) }
}
}
impl Override {
fn construct_cxx_channel() -> Channel {
unsafe {
let mut buf = std::mem::MaybeUninit::<Channel>::uninit();
Channel__OVERRIDE__CTOR(&mut buf);
buf.assume_init()
}
}
fn get_base_offset<T>() -> usize
where
T: ChannelOverrides,
{
let buf = std::mem::MaybeUninit::<T>::uninit();
let top_ptr: *const T = buf.as_ptr();
let self_ptr: *const Self = unsafe { (*top_ptr).base() };
util::FieldOffset::from_ptrs(top_ptr, self_ptr).offset()
}
fn get_rust_vtable<T>() -> util::RustVTable<&'static dyn ChannelOverrides>
where
T: ChannelOverrides,
{
let buf = std::mem::MaybeUninit::<T>::uninit();
let embedder_ptr = buf.as_ptr();
let trait_object: *const dyn ChannelOverrides = embedder_ptr;
let (data_ptr, vtable): (*const T, util::RustVTable<_>) =
unsafe { std::mem::transmute(trait_object) };
assert_eq!(data_ptr, embedder_ptr);
vtable
}
pub fn new<T>() -> Self
where
T: ChannelOverrides,
{
Self {
cxx_channel: Self::construct_cxx_channel(),
base_offset: Self::get_base_offset::<T>(),
rust_vtable: Self::get_rust_vtable::<T>(),
}
}
fn channel_offset() -> util::FieldOffset<Self, Channel> {
let buf = std::mem::MaybeUninit::<Self>::uninit();
util::FieldOffset::from_ptrs(buf.as_ptr(), unsafe { &(*buf.as_ptr()).cxx_channel })
}
fn embedder_offset(&self) -> util::FieldOffset<util::Opaque, Self> {
util::FieldOffset::<util::Opaque, Self>::from_offset(self.base_offset)
}
unsafe fn dispatch(channel: &Channel) -> &dyn ChannelOverrides {
let this = Self::channel_offset().to_outer(channel);
let embedder = this.embedder_offset().to_outer(this);
std::mem::transmute((embedder, this.rust_vtable))
}
unsafe fn dispatch_mut(channel: &mut Channel) -> &mut dyn ChannelOverrides {
let this = Self::channel_offset().to_outer_mut(channel);
let vtable = this.rust_vtable;
let embedder = this.embedder_offset().to_outer_mut(this);
std::mem::transmute((embedder, vtable))
}
}
impl std::ops::Deref for Override {
type Target = Channel;
fn deref(&self) -> &Channel {
&self.cxx_channel
}
}
impl std::ops::DerefMut for Override {
fn deref_mut(&mut self) -> &mut Channel {
&mut self.cxx_channel
}
}
}
mod trying {
use super::channel::*;
#[allow(dead_code)]
pub struct Session {
a: i32,
b: String,
c: Override,
}
impl ChannelOverrides for Session {
fn base(&self) -> &Override {
&self.c
}
fn base_mut(&mut self) -> &mut Override {
&mut self.c
}
fn a(&mut self) {
println!("Override a!");
}
fn b(&self) -> i32 {
println!("Override b!");
42
}
}
impl Session {
pub fn new() -> Self {
Self {
a: 1,
b: "abc".to_owned(),
c: Override::new::<Self>(),
}
}
}
}
mod util {
use std::marker::PhantomData;
use std::mem::size_of;
pub type Opaque = [usize; 0];
#[repr(transparent)]
#[derive(Copy, Clone, Debug)]
pub struct RustVTable<DynT>(pub *const Opaque, pub PhantomData<DynT>);
#[derive(Copy, Clone, Debug)]
#[repr(transparent)]
pub struct FieldOffset<O, I>(isize, PhantomData<(O, I)>);
impl<O, I> FieldOffset<O, I> {
pub fn from_ptrs(o_ptr: *const O, i_ptr: *const I) -> Self {
let o_addr = o_ptr as usize;
let i_addr = i_ptr as usize;
assert!(i_addr >= o_addr);
assert!((i_addr + size_of::<I>()) <= (o_addr + size_of::<O>()));
let offset = (o_addr - i_addr) as isize;
assert!(offset > 0);
Self(offset, PhantomData)
}
pub fn from_offset(offset: usize) -> Self {
assert!((offset as isize) > 0);
Self(offset as isize, PhantomData)
}
pub fn offset(self) -> usize {
self.0 as usize
}
fn shift<PI, PO>(ptr: *const PI, delta: isize) -> *mut PO {
(ptr as isize + delta) as *mut PO
}
pub unsafe fn to_outer<'a>(&self, inner: &'a I) -> &'a O {
Self::shift::<I, O>(inner, -self.0).as_ref().unwrap()
}
#[allow(dead_code)]
pub unsafe fn to_outer_mut<'a>(&self, inner: &'a mut I) -> &'a mut O {
Self::shift::<I, O>(inner, -self.0).as_mut().unwrap()
}
}
impl<O, M, I> std::ops::Add<FieldOffset<M, I>> for FieldOffset<O, M> {
type Output = FieldOffset<O, I>;
fn add(self, that: FieldOffset<M, I>) -> Self::Output {
FieldOffset::<O, I>::from_offset(self.offset() + that.offset())
}
}
}
fn main() {
trying::Session::new();
}