forked from BigfootDev/flatbuffers
442949bc11
* Updated comments and fixed a fundemental type error. * bump rust flatbuffers semver * Initial commit with verifier, need to clean up * Verifier tested. Needs clean up and refactoring. * Display for InvalidFlatbuffer and better errors for strings * SimpleToVerify, some refactoring * Combined VerifierType TableAccessorFuncBody into FollowType * scrub todos * Update Rust get_root functions. There are 6 variants, with verifier options, default verifier options and no verification "fast". * Rename root fns * inline * Update to use thiserror * fix for bad compiler * improve error formatting * Replace multiply with saturating_multiply * saturating adds too * Add docs disclaiming experimental verification system Co-authored-by: Casper Neo <cneo@google.com>
70 lines
1.9 KiB
Rust
70 lines
1.9 KiB
Rust
/*
|
|
* Copyright 2018 Google Inc. All rights reserved.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
use crate::follow::Follow;
|
|
use crate::primitives::*;
|
|
use crate::vtable::VTable;
|
|
|
|
#[derive(Clone, Copy, Debug, PartialEq)]
|
|
pub struct Table<'a> {
|
|
pub buf: &'a [u8],
|
|
pub loc: usize,
|
|
}
|
|
|
|
impl<'a> Table<'a> {
|
|
#[inline]
|
|
pub fn new(buf: &'a [u8], loc: usize) -> Self {
|
|
Table { buf, loc }
|
|
}
|
|
#[inline]
|
|
pub fn vtable(&self) -> VTable<'a> {
|
|
<BackwardsSOffset<VTable<'a>>>::follow(self.buf, self.loc)
|
|
}
|
|
#[inline]
|
|
pub fn get<T: Follow<'a> + 'a>(
|
|
&self,
|
|
slot_byte_loc: VOffsetT,
|
|
default: Option<T::Inner>,
|
|
) -> Option<T::Inner> {
|
|
let o = self.vtable().get(slot_byte_loc) as usize;
|
|
if o == 0 {
|
|
return default;
|
|
}
|
|
Some(<T>::follow(self.buf, self.loc + o))
|
|
}
|
|
}
|
|
|
|
impl<'a> Follow<'a> for Table<'a> {
|
|
type Inner = Table<'a>;
|
|
#[inline]
|
|
fn follow(buf: &'a [u8], loc: usize) -> Self::Inner {
|
|
Table { buf, loc }
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
pub fn buffer_has_identifier(data: &[u8], ident: &str, size_prefixed: bool) -> bool {
|
|
assert_eq!(ident.len(), FILE_IDENTIFIER_LENGTH);
|
|
|
|
let got = if size_prefixed {
|
|
<SkipSizePrefix<SkipRootOffset<FileIdentifier>>>::follow(data, 0)
|
|
} else {
|
|
<SkipRootOffset<FileIdentifier>>::follow(data, 0)
|
|
};
|
|
|
|
ident.as_bytes() == got
|
|
}
|