450 lines
13 KiB
Rust
450 lines
13 KiB
Rust
// Copyright 2012 The Rust Project Developers. See the COPYRIGHT
|
|
// file at the top-level directory of this distribution and at
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
// option. This file may not be copied, modified, or distributed
|
|
// except according to those terms.
|
|
|
|
/*!
|
|
|
|
Runtime type reflection
|
|
|
|
*/
|
|
|
|
#[allow(missing_doc)];
|
|
|
|
use unstable::intrinsics::{Disr, Opaque, TyDesc, TyVisitor};
|
|
use mem;
|
|
use unstable::raw;
|
|
|
|
/**
|
|
* Trait for visitor that wishes to reflect on data. To use this, create a
|
|
* struct that encapsulates the set of pointers you wish to walk through a
|
|
* data structure, and implement both `MovePtr` for it as well as `TyVisitor`;
|
|
* then build a MovePtrAdaptor wrapped around your struct.
|
|
*/
|
|
pub trait MovePtr {
|
|
fn move_ptr(&mut self, adjustment: |*u8| -> *u8);
|
|
fn push_ptr(&mut self);
|
|
fn pop_ptr(&mut self);
|
|
}
|
|
|
|
/// Helper function for alignment calculation.
|
|
#[inline]
|
|
pub fn align(size: uint, align: uint) -> uint {
|
|
((size + align) - 1u) & !(align - 1u)
|
|
}
|
|
|
|
/// Adaptor to wrap around visitors implementing MovePtr.
|
|
pub struct MovePtrAdaptor<V> {
|
|
priv inner: V
|
|
}
|
|
pub fn MovePtrAdaptor<V:TyVisitor + MovePtr>(v: V) -> MovePtrAdaptor<V> {
|
|
MovePtrAdaptor { inner: v }
|
|
}
|
|
|
|
impl<V:TyVisitor + MovePtr> MovePtrAdaptor<V> {
|
|
#[inline]
|
|
pub fn bump(&mut self, sz: uint) {
|
|
self.inner.move_ptr(|p| ((p as uint) + sz) as *u8)
|
|
}
|
|
|
|
#[inline]
|
|
pub fn align(&mut self, a: uint) {
|
|
self.inner.move_ptr(|p| align(p as uint, a) as *u8)
|
|
}
|
|
|
|
#[inline]
|
|
pub fn align_to<T>(&mut self) {
|
|
self.align(mem::min_align_of::<T>());
|
|
}
|
|
|
|
#[inline]
|
|
pub fn bump_past<T>(&mut self) {
|
|
self.bump(mem::size_of::<T>());
|
|
}
|
|
|
|
pub fn unwrap(self) -> V { self.inner }
|
|
}
|
|
|
|
/// Abstract type-directed pointer-movement using the MovePtr trait
|
|
impl<V:TyVisitor + MovePtr> TyVisitor for MovePtrAdaptor<V> {
|
|
fn visit_bot(&mut self) -> bool {
|
|
self.align_to::<()>();
|
|
if ! self.inner.visit_bot() { return false; }
|
|
self.bump_past::<()>();
|
|
true
|
|
}
|
|
|
|
fn visit_nil(&mut self) -> bool {
|
|
self.align_to::<()>();
|
|
if ! self.inner.visit_nil() { return false; }
|
|
self.bump_past::<()>();
|
|
true
|
|
}
|
|
|
|
fn visit_bool(&mut self) -> bool {
|
|
self.align_to::<bool>();
|
|
if ! self.inner.visit_bool() { return false; }
|
|
self.bump_past::<bool>();
|
|
true
|
|
}
|
|
|
|
fn visit_int(&mut self) -> bool {
|
|
self.align_to::<int>();
|
|
if ! self.inner.visit_int() { return false; }
|
|
self.bump_past::<int>();
|
|
true
|
|
}
|
|
|
|
fn visit_i8(&mut self) -> bool {
|
|
self.align_to::<i8>();
|
|
if ! self.inner.visit_i8() { return false; }
|
|
self.bump_past::<i8>();
|
|
true
|
|
}
|
|
|
|
fn visit_i16(&mut self) -> bool {
|
|
self.align_to::<i16>();
|
|
if ! self.inner.visit_i16() { return false; }
|
|
self.bump_past::<i16>();
|
|
true
|
|
}
|
|
|
|
fn visit_i32(&mut self) -> bool {
|
|
self.align_to::<i32>();
|
|
if ! self.inner.visit_i32() { return false; }
|
|
self.bump_past::<i32>();
|
|
true
|
|
}
|
|
|
|
fn visit_i64(&mut self) -> bool {
|
|
self.align_to::<i64>();
|
|
if ! self.inner.visit_i64() { return false; }
|
|
self.bump_past::<i64>();
|
|
true
|
|
}
|
|
|
|
fn visit_uint(&mut self) -> bool {
|
|
self.align_to::<uint>();
|
|
if ! self.inner.visit_uint() { return false; }
|
|
self.bump_past::<uint>();
|
|
true
|
|
}
|
|
|
|
fn visit_u8(&mut self) -> bool {
|
|
self.align_to::<u8>();
|
|
if ! self.inner.visit_u8() { return false; }
|
|
self.bump_past::<u8>();
|
|
true
|
|
}
|
|
|
|
fn visit_u16(&mut self) -> bool {
|
|
self.align_to::<u16>();
|
|
if ! self.inner.visit_u16() { return false; }
|
|
self.bump_past::<u16>();
|
|
true
|
|
}
|
|
|
|
fn visit_u32(&mut self) -> bool {
|
|
self.align_to::<u32>();
|
|
if ! self.inner.visit_u32() { return false; }
|
|
self.bump_past::<u32>();
|
|
true
|
|
}
|
|
|
|
fn visit_u64(&mut self) -> bool {
|
|
self.align_to::<u64>();
|
|
if ! self.inner.visit_u64() { return false; }
|
|
self.bump_past::<u64>();
|
|
true
|
|
}
|
|
|
|
fn visit_f32(&mut self) -> bool {
|
|
self.align_to::<f32>();
|
|
if ! self.inner.visit_f32() { return false; }
|
|
self.bump_past::<f32>();
|
|
true
|
|
}
|
|
|
|
fn visit_f64(&mut self) -> bool {
|
|
self.align_to::<f64>();
|
|
if ! self.inner.visit_f64() { return false; }
|
|
self.bump_past::<f64>();
|
|
true
|
|
}
|
|
|
|
fn visit_char(&mut self) -> bool {
|
|
self.align_to::<char>();
|
|
if ! self.inner.visit_char() { return false; }
|
|
self.bump_past::<char>();
|
|
true
|
|
}
|
|
|
|
fn visit_estr_box(&mut self) -> bool {
|
|
true
|
|
}
|
|
|
|
fn visit_estr_uniq(&mut self) -> bool {
|
|
self.align_to::<~str>();
|
|
if ! self.inner.visit_estr_uniq() { return false; }
|
|
self.bump_past::<~str>();
|
|
true
|
|
}
|
|
|
|
fn visit_estr_slice(&mut self) -> bool {
|
|
self.align_to::<&'static str>();
|
|
if ! self.inner.visit_estr_slice() { return false; }
|
|
self.bump_past::<&'static str>();
|
|
true
|
|
}
|
|
|
|
fn visit_estr_fixed(&mut self, n: uint,
|
|
sz: uint,
|
|
align: uint) -> bool {
|
|
self.align(align);
|
|
if ! self.inner.visit_estr_fixed(n, sz, align) { return false; }
|
|
self.bump(sz);
|
|
true
|
|
}
|
|
|
|
fn visit_box(&mut self, mtbl: uint, inner: *TyDesc) -> bool {
|
|
self.align_to::<@u8>();
|
|
if ! self.inner.visit_box(mtbl, inner) { return false; }
|
|
self.bump_past::<@u8>();
|
|
true
|
|
}
|
|
|
|
fn visit_uniq(&mut self, mtbl: uint, inner: *TyDesc) -> bool {
|
|
self.align_to::<~u8>();
|
|
if ! self.inner.visit_uniq(mtbl, inner) { return false; }
|
|
self.bump_past::<~u8>();
|
|
true
|
|
}
|
|
|
|
fn visit_ptr(&mut self, mtbl: uint, inner: *TyDesc) -> bool {
|
|
self.align_to::<*u8>();
|
|
if ! self.inner.visit_ptr(mtbl, inner) { return false; }
|
|
self.bump_past::<*u8>();
|
|
true
|
|
}
|
|
|
|
fn visit_rptr(&mut self, mtbl: uint, inner: *TyDesc) -> bool {
|
|
self.align_to::<&'static u8>();
|
|
if ! self.inner.visit_rptr(mtbl, inner) { return false; }
|
|
self.bump_past::<&'static u8>();
|
|
true
|
|
}
|
|
|
|
fn visit_unboxed_vec(&mut self, mtbl: uint, inner: *TyDesc) -> bool {
|
|
self.align_to::<raw::Vec<()>>();
|
|
if ! self.inner.visit_vec(mtbl, inner) { return false; }
|
|
true
|
|
}
|
|
|
|
fn visit_vec(&mut self, mtbl: uint, inner: *TyDesc) -> bool {
|
|
self.align_to::<~[u8]>();
|
|
if ! self.inner.visit_vec(mtbl, inner) { return false; }
|
|
self.bump_past::<~[u8]>();
|
|
true
|
|
}
|
|
|
|
fn visit_evec_box(&mut self, _mtbl: uint, _inner: *TyDesc) -> bool {
|
|
true
|
|
}
|
|
|
|
fn visit_evec_uniq(&mut self, mtbl: uint, inner: *TyDesc) -> bool {
|
|
self.align_to::<~[u8]>();
|
|
if ! self.inner.visit_evec_uniq(mtbl, inner) { return false; }
|
|
self.bump_past::<~[u8]>();
|
|
true
|
|
}
|
|
|
|
fn visit_evec_slice(&mut self, mtbl: uint, inner: *TyDesc) -> bool {
|
|
self.align_to::<&'static [u8]>();
|
|
if ! self.inner.visit_evec_slice(mtbl, inner) { return false; }
|
|
self.bump_past::<&'static [u8]>();
|
|
true
|
|
}
|
|
|
|
fn visit_evec_fixed(&mut self, n: uint, sz: uint, align: uint,
|
|
mtbl: uint, inner: *TyDesc) -> bool {
|
|
self.align(align);
|
|
if ! self.inner.visit_evec_fixed(n, sz, align, mtbl, inner) {
|
|
return false;
|
|
}
|
|
self.bump(sz);
|
|
true
|
|
}
|
|
|
|
fn visit_enter_rec(&mut self, n_fields: uint, sz: uint, align: uint) -> bool {
|
|
self.align(align);
|
|
if ! self.inner.visit_enter_rec(n_fields, sz, align) { return false; }
|
|
true
|
|
}
|
|
|
|
fn visit_rec_field(&mut self, i: uint, name: &str,
|
|
mtbl: uint, inner: *TyDesc) -> bool {
|
|
unsafe { self.align((*inner).align); }
|
|
if ! self.inner.visit_rec_field(i, name, mtbl, inner) {
|
|
return false;
|
|
}
|
|
unsafe { self.bump((*inner).size); }
|
|
true
|
|
}
|
|
|
|
fn visit_leave_rec(&mut self, n_fields: uint, sz: uint, align: uint) -> bool {
|
|
if ! self.inner.visit_leave_rec(n_fields, sz, align) { return false; }
|
|
true
|
|
}
|
|
|
|
fn visit_enter_class(&mut self, name: &str, named_fields: bool, n_fields: uint, sz: uint,
|
|
align: uint) -> bool {
|
|
self.align(align);
|
|
if ! self.inner.visit_enter_class(name, named_fields, n_fields, sz, align) {
|
|
return false;
|
|
}
|
|
true
|
|
}
|
|
|
|
fn visit_class_field(&mut self, i: uint, name: &str, named: bool, mtbl: uint,
|
|
inner: *TyDesc) -> bool {
|
|
unsafe { self.align((*inner).align); }
|
|
if ! self.inner.visit_class_field(i, name, named, mtbl, inner) {
|
|
return false;
|
|
}
|
|
unsafe { self.bump((*inner).size); }
|
|
true
|
|
}
|
|
|
|
fn visit_leave_class(&mut self, name: &str, named_fields: bool, n_fields: uint, sz: uint,
|
|
align: uint) -> bool {
|
|
if ! self.inner.visit_leave_class(name, named_fields, n_fields, sz, align) {
|
|
return false;
|
|
}
|
|
true
|
|
}
|
|
|
|
fn visit_enter_tup(&mut self, n_fields: uint, sz: uint, align: uint) -> bool {
|
|
self.align(align);
|
|
if ! self.inner.visit_enter_tup(n_fields, sz, align) { return false; }
|
|
true
|
|
}
|
|
|
|
fn visit_tup_field(&mut self, i: uint, inner: *TyDesc) -> bool {
|
|
unsafe { self.align((*inner).align); }
|
|
if ! self.inner.visit_tup_field(i, inner) { return false; }
|
|
unsafe { self.bump((*inner).size); }
|
|
true
|
|
}
|
|
|
|
fn visit_leave_tup(&mut self, n_fields: uint, sz: uint, align: uint) -> bool {
|
|
if ! self.inner.visit_leave_tup(n_fields, sz, align) { return false; }
|
|
true
|
|
}
|
|
|
|
fn visit_enter_fn(&mut self, purity: uint, proto: uint,
|
|
n_inputs: uint, retstyle: uint) -> bool {
|
|
if ! self.inner.visit_enter_fn(purity, proto, n_inputs, retstyle) {
|
|
return false
|
|
}
|
|
true
|
|
}
|
|
|
|
fn visit_fn_input(&mut self, i: uint, mode: uint, inner: *TyDesc) -> bool {
|
|
if ! self.inner.visit_fn_input(i, mode, inner) { return false; }
|
|
true
|
|
}
|
|
|
|
fn visit_fn_output(&mut self, retstyle: uint, variadic: bool, inner: *TyDesc) -> bool {
|
|
if ! self.inner.visit_fn_output(retstyle, variadic, inner) { return false; }
|
|
true
|
|
}
|
|
|
|
fn visit_leave_fn(&mut self, purity: uint, proto: uint,
|
|
n_inputs: uint, retstyle: uint) -> bool {
|
|
if ! self.inner.visit_leave_fn(purity, proto, n_inputs, retstyle) {
|
|
return false;
|
|
}
|
|
true
|
|
}
|
|
|
|
fn visit_enter_enum(&mut self, n_variants: uint,
|
|
get_disr: extern unsafe fn(ptr: *Opaque) -> Disr,
|
|
sz: uint, align: uint)
|
|
-> bool {
|
|
self.align(align);
|
|
if ! self.inner.visit_enter_enum(n_variants, get_disr, sz, align) {
|
|
return false;
|
|
}
|
|
true
|
|
}
|
|
|
|
fn visit_enter_enum_variant(&mut self, variant: uint,
|
|
disr_val: Disr,
|
|
n_fields: uint,
|
|
name: &str) -> bool {
|
|
if ! self.inner.visit_enter_enum_variant(variant, disr_val,
|
|
n_fields, name) {
|
|
return false;
|
|
}
|
|
true
|
|
}
|
|
|
|
fn visit_enum_variant_field(&mut self, i: uint, offset: uint, inner: *TyDesc) -> bool {
|
|
self.inner.push_ptr();
|
|
self.bump(offset);
|
|
if ! self.inner.visit_enum_variant_field(i, offset, inner) { return false; }
|
|
self.inner.pop_ptr();
|
|
true
|
|
}
|
|
|
|
fn visit_leave_enum_variant(&mut self, variant: uint,
|
|
disr_val: Disr,
|
|
n_fields: uint,
|
|
name: &str) -> bool {
|
|
if ! self.inner.visit_leave_enum_variant(variant, disr_val,
|
|
n_fields, name) {
|
|
return false;
|
|
}
|
|
true
|
|
}
|
|
|
|
fn visit_leave_enum(&mut self, n_variants: uint,
|
|
get_disr: extern unsafe fn(ptr: *Opaque) -> Disr,
|
|
sz: uint, align: uint) -> bool {
|
|
if ! self.inner.visit_leave_enum(n_variants, get_disr, sz, align) {
|
|
return false;
|
|
}
|
|
self.bump(sz);
|
|
true
|
|
}
|
|
|
|
fn visit_trait(&mut self, name: &str) -> bool {
|
|
self.align_to::<~TyVisitor>();
|
|
if ! self.inner.visit_trait(name) { return false; }
|
|
self.bump_past::<~TyVisitor>();
|
|
true
|
|
}
|
|
|
|
fn visit_param(&mut self, i: uint) -> bool {
|
|
if ! self.inner.visit_param(i) { return false; }
|
|
true
|
|
}
|
|
|
|
fn visit_self(&mut self) -> bool {
|
|
self.align_to::<&'static u8>();
|
|
if ! self.inner.visit_self() { return false; }
|
|
self.align_to::<&'static u8>();
|
|
true
|
|
}
|
|
|
|
fn visit_type(&mut self) -> bool {
|
|
if ! self.inner.visit_type() { return false; }
|
|
true
|
|
}
|
|
}
|