]>
Commit | Line | Data |
---|---|---|
f67539c2 TL |
1 | // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- |
2 | // vim: ts=8 sw=2 smarttab | |
3 | ||
4 | #ifndef CEPH_LIBRBD_CRYPTO_CRYPTO_CONTEXT_POOL_H | |
5 | #define CEPH_LIBRBD_CRYPTO_CRYPTO_CONTEXT_POOL_H | |
6 | ||
7 | #include "librbd/crypto/DataCryptor.h" | |
8 | #include "common/allocator.h" | |
9 | #include "include/ceph_assert.h" | |
10 | #include <boost/lockfree/queue.hpp> | |
11 | ||
12 | namespace librbd { | |
13 | namespace crypto { | |
14 | ||
15 | template <typename T> | |
16 | class CryptoContextPool : public DataCryptor<T> { | |
17 | ||
18 | public: | |
19 | CryptoContextPool(DataCryptor<T>* data_cryptor, uint32_t pool_size); | |
20 | ~CryptoContextPool(); | |
21 | ||
22 | T* get_context(CipherMode mode) override; | |
23 | void return_context(T* ctx, CipherMode mode) override; | |
24 | ||
25 | inline uint32_t get_block_size() const override { | |
26 | return m_data_cryptor->get_block_size(); | |
27 | } | |
28 | inline uint32_t get_iv_size() const override { | |
29 | return m_data_cryptor->get_iv_size(); | |
30 | } | |
31 | inline int get_key_length() const override { | |
32 | return m_data_cryptor->get_key_length(); | |
33 | } | |
34 | inline const unsigned char* get_key() const override { | |
35 | return m_data_cryptor->get_key(); | |
36 | } | |
37 | inline int init_context(T* ctx, const unsigned char* iv, | |
38 | uint32_t iv_length) const override { | |
39 | return m_data_cryptor->init_context(ctx, iv, iv_length); | |
40 | } | |
41 | inline int update_context(T* ctx, const unsigned char* in, | |
42 | unsigned char* out, | |
43 | uint32_t len) const override { | |
44 | return m_data_cryptor->update_context(ctx, in, out, len); | |
45 | } | |
46 | ||
47 | typedef boost::lockfree::queue< | |
48 | T*, | |
49 | boost::lockfree::allocator<ceph::allocator<void>>> ContextQueue; | |
50 | ||
51 | private: | |
52 | DataCryptor<T>* m_data_cryptor; | |
53 | ContextQueue m_encrypt_contexts; | |
54 | ContextQueue m_decrypt_contexts; | |
55 | ||
56 | inline ContextQueue& get_contexts(CipherMode mode) { | |
57 | switch(mode) { | |
58 | case CIPHER_MODE_ENC: | |
59 | return m_encrypt_contexts; | |
60 | case CIPHER_MODE_DEC: | |
61 | return m_decrypt_contexts; | |
62 | default: | |
63 | ceph_assert(false); | |
64 | } | |
65 | } | |
66 | }; | |
67 | ||
68 | } // namespace crypto | |
69 | } // namespace librbd | |
70 | ||
71 | #endif // CEPH_LIBRBD_CRYPTO_CRYPTO_CONTEXT_POOL_H |