Add stack creation and push

This commit is contained in:
Steven Fackler 2016-11-06 23:46:42 -08:00
parent 8f7df7b205
commit 597d05b8f8
6 changed files with 39 additions and 5 deletions

View File

@ -602,10 +602,12 @@ extern {
pub fn EVP_MD_CTX_create() -> *mut EVP_MD_CTX; pub fn EVP_MD_CTX_create() -> *mut EVP_MD_CTX;
pub fn EVP_MD_CTX_destroy(ctx: *mut EVP_MD_CTX); pub fn EVP_MD_CTX_destroy(ctx: *mut EVP_MD_CTX);
pub fn sk_new_null() -> *mut _STACK;
pub fn sk_num(st: *const _STACK) -> c_int; pub fn sk_num(st: *const _STACK) -> c_int;
pub fn sk_value(st: *const _STACK, n: c_int) -> *mut c_void; pub fn sk_value(st: *const _STACK, n: c_int) -> *mut c_void;
pub fn sk_free(st: *mut _STACK); pub fn sk_free(st: *mut _STACK);
pub fn sk_pop_free(st: *mut _STACK, free: Option<unsafe extern "C" fn (*mut c_void)>); pub fn sk_pop_free(st: *mut _STACK, free: Option<unsafe extern "C" fn (*mut c_void)>);
pub fn sk_push(st: *mut _STACK, data: *mut c_void) -> c_int;
pub fn sk_pop(st: *mut _STACK) -> *mut c_void; pub fn sk_pop(st: *mut _STACK) -> *mut c_void;
pub fn SSLeay() -> c_ulong; pub fn SSLeay() -> c_ulong;

View File

@ -151,7 +151,9 @@ extern {
pub fn OpenSSL_version_num() -> c_ulong; pub fn OpenSSL_version_num() -> c_ulong;
pub fn OpenSSL_version(key: c_int) -> *const c_char; pub fn OpenSSL_version(key: c_int) -> *const c_char;
pub fn OPENSSL_sk_new_null() -> *mut ::OPENSSL_STACK;
pub fn OPENSSL_sk_free(st: *mut ::OPENSSL_STACK); pub fn OPENSSL_sk_free(st: *mut ::OPENSSL_STACK);
pub fn OPENSSL_sk_pop_free(st: *mut ::OPENSSL_STACK, free: Option<unsafe extern "C" fn (*mut c_void)>); pub fn OPENSSL_sk_pop_free(st: *mut ::OPENSSL_STACK, free: Option<unsafe extern "C" fn (*mut c_void)>);
pub fn OPENSSL_sk_push(st: *mut ::OPENSSL_STACK, data: *const c_void) -> c_int;
pub fn OPENSSL_sk_pop(st: *mut ::OPENSSL_STACK) -> *mut c_void; pub fn OPENSSL_sk_pop(st: *mut ::OPENSSL_STACK) -> *mut c_void;
} }

View File

@ -31,6 +31,10 @@ macro_rules! type_ {
unsafe fn from_ptr(ptr: *mut $c) -> $n { unsafe fn from_ptr(ptr: *mut $c) -> $n {
$n(ptr) $n(ptr)
} }
fn as_ptr(&self) -> *mut $c {
self.0
}
} }
impl Drop for $n { impl Drop for $n {

View File

@ -822,6 +822,10 @@ impl OpenSslType for SslCipher {
unsafe fn from_ptr(ptr: *mut ffi::SSL_CIPHER) -> SslCipher { unsafe fn from_ptr(ptr: *mut ffi::SSL_CIPHER) -> SslCipher {
SslCipher(ptr) SslCipher(ptr)
} }
fn as_ptr(&self) -> *mut ffi::SSL_CIPHER {
self.0
}
} }
impl Deref for SslCipher { impl Deref for SslCipher {

View File

@ -6,14 +6,18 @@ use std::marker::PhantomData;
use libc::c_int; use libc::c_int;
use std::mem; use std::mem;
use {cvt, cvt_p};
use error::ErrorStack;
use types::{OpenSslType, OpenSslTypeRef}; use types::{OpenSslType, OpenSslTypeRef};
use util::Opaque; use util::Opaque;
#[cfg(ossl10x)] #[cfg(ossl10x)]
use ffi::{sk_pop as OPENSSL_sk_pop, sk_free as OPENSSL_sk_free, sk_num as OPENSSL_sk_num, use ffi::{sk_pop as OPENSSL_sk_pop, sk_free as OPENSSL_sk_free, sk_num as OPENSSL_sk_num,
sk_value as OPENSSL_sk_value, _STACK as OPENSSL_STACK}; sk_value as OPENSSL_sk_value, _STACK as OPENSSL_STACK,
sk_new_null as OPENSSL_sk_new_null, sk_push as OPENSSL_sk_push};
#[cfg(ossl110)] #[cfg(ossl110)]
use ffi::{OPENSSL_sk_pop, OPENSSL_sk_free, OPENSSL_sk_num, OPENSSL_sk_value, OPENSSL_STACK}; use ffi::{OPENSSL_sk_pop, OPENSSL_sk_free, OPENSSL_sk_num, OPENSSL_sk_value, OPENSSL_STACK,
OPENSSL_sk_new_null, OPENSSL_sk_push};
/// Trait implemented by types which can be placed in a stack. /// Trait implemented by types which can be placed in a stack.
/// ///
@ -31,9 +35,11 @@ pub trait Stackable: OpenSslType {
pub struct Stack<T: Stackable>(*mut T::StackType); pub struct Stack<T: Stackable>(*mut T::StackType);
impl<T: Stackable> Stack<T> { impl<T: Stackable> Stack<T> {
/// Return a new Stack<T>, taking ownership of the handle pub fn new() -> Result<Stack<T>, ErrorStack> {
pub unsafe fn from_ptr(stack: *mut T::StackType) -> Stack<T> { unsafe {
Stack(stack) let ptr = try!(cvt_p(OPENSSL_sk_new_null()));
Ok(Stack(ptr as *mut _))
}
} }
} }
@ -79,6 +85,10 @@ impl<T: Stackable> OpenSslType for Stack<T> {
unsafe fn from_ptr(ptr: *mut T::StackType) -> Stack<T> { unsafe fn from_ptr(ptr: *mut T::StackType) -> Stack<T> {
Stack(ptr) Stack(ptr)
} }
fn as_ptr(&self) -> *mut T::StackType {
self.0
}
} }
impl<T: Stackable> Deref for Stack<T> { impl<T: Stackable> Deref for Stack<T> {
@ -198,6 +208,15 @@ impl<T: Stackable> StackRef<T> {
} }
} }
/// Pushes a value onto the top of the stack.
pub fn push(&mut self, data: T) -> Result<(), ErrorStack> {
unsafe {
try!(cvt(OPENSSL_sk_push(self.as_stack(), data.as_ptr() as *mut _)));
mem::forget(data);
Ok(())
}
}
/// Removes the last element from the stack and returns it. /// Removes the last element from the stack and returns it.
pub fn pop(&mut self) -> Option<T> { pub fn pop(&mut self) -> Option<T> {
unsafe { unsafe {

View File

@ -13,6 +13,9 @@ pub trait OpenSslType: Sized {
/// Constructs an instance of this type from its raw type. /// Constructs an instance of this type from its raw type.
unsafe fn from_ptr(ptr: *mut Self::CType) -> Self; unsafe fn from_ptr(ptr: *mut Self::CType) -> Self;
/// Returns a raw pointer to the wrapped value.
fn as_ptr(&self) -> *mut Self::CType;
} }
/// A trait implemented by types which reference borrowed OpenSSL types. /// A trait implemented by types which reference borrowed OpenSSL types.