mirror of https://github.com/grpc/grpc.git
commit
c876923bc2
18 changed files with 444 additions and 3 deletions
@ -0,0 +1,77 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2015, Google Inc. |
||||||
|
* All rights reserved. |
||||||
|
* |
||||||
|
* Redistribution and use in source and binary forms, with or without |
||||||
|
* modification, are permitted provided that the following conditions are |
||||||
|
* met: |
||||||
|
* |
||||||
|
* * Redistributions of source code must retain the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer. |
||||||
|
* * Redistributions in binary form must reproduce the above |
||||||
|
* copyright notice, this list of conditions and the following disclaimer |
||||||
|
* in the documentation and/or other materials provided with the |
||||||
|
* distribution. |
||||||
|
* * Neither the name of Google Inc. nor the names of its |
||||||
|
* contributors may be used to endorse or promote products derived from |
||||||
|
* this software without specific prior written permission. |
||||||
|
* |
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||||
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||||
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||||
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||||
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||||
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||||
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||||
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||||
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef GRPCXX_AUTH_PROPERTY_ITERATOR_H |
||||||
|
#define GRPCXX_AUTH_PROPERTY_ITERATOR_H |
||||||
|
|
||||||
|
#include <iterator> |
||||||
|
#include <vector> |
||||||
|
|
||||||
|
#include <grpc++/config.h> |
||||||
|
|
||||||
|
struct grpc_auth_context; |
||||||
|
struct grpc_auth_property; |
||||||
|
struct grpc_auth_property_iterator; |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
class SecureAuthContext; |
||||||
|
|
||||||
|
typedef std::pair<grpc::string, grpc::string> AuthProperty; |
||||||
|
|
||||||
|
class AuthPropertyIterator |
||||||
|
: public std::iterator<std::input_iterator_tag, const AuthProperty> { |
||||||
|
public: |
||||||
|
~AuthPropertyIterator(); |
||||||
|
AuthPropertyIterator& operator++(); |
||||||
|
AuthPropertyIterator operator++(int); |
||||||
|
bool operator==(const AuthPropertyIterator& rhs) const; |
||||||
|
bool operator!=(const AuthPropertyIterator& rhs) const; |
||||||
|
const AuthProperty operator*(); |
||||||
|
|
||||||
|
protected: |
||||||
|
AuthPropertyIterator(); |
||||||
|
AuthPropertyIterator(const grpc_auth_property* property, |
||||||
|
const grpc_auth_property_iterator* iter); |
||||||
|
private: |
||||||
|
friend class SecureAuthContext; |
||||||
|
const grpc_auth_property* property_; |
||||||
|
// The following items form a grpc_auth_property_iterator.
|
||||||
|
const grpc_auth_context* ctx_; |
||||||
|
size_t index_; |
||||||
|
const char* name_; |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
#endif // GRPCXX_AUTH_PROPERTY_ITERATOR_H
|
||||||
|
|
@ -0,0 +1,87 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2015, Google Inc. |
||||||
|
* All rights reserved. |
||||||
|
* |
||||||
|
* Redistribution and use in source and binary forms, with or without |
||||||
|
* modification, are permitted provided that the following conditions are |
||||||
|
* met: |
||||||
|
* |
||||||
|
* * Redistributions of source code must retain the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer. |
||||||
|
* * Redistributions in binary form must reproduce the above |
||||||
|
* copyright notice, this list of conditions and the following disclaimer |
||||||
|
* in the documentation and/or other materials provided with the |
||||||
|
* distribution. |
||||||
|
* * Neither the name of Google Inc. nor the names of its |
||||||
|
* contributors may be used to endorse or promote products derived from |
||||||
|
* this software without specific prior written permission. |
||||||
|
* |
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||||
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||||
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||||
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||||
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||||
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||||
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||||
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||||
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#include <grpc++/auth_property_iterator.h> |
||||||
|
|
||||||
|
#include <grpc/grpc_security.h> |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
|
||||||
|
AuthPropertyIterator::AuthPropertyIterator() |
||||||
|
: property_(nullptr), ctx_(nullptr), index_(0), name_(nullptr) {} |
||||||
|
|
||||||
|
AuthPropertyIterator::AuthPropertyIterator( |
||||||
|
const grpc_auth_property* property, const grpc_auth_property_iterator* iter) |
||||||
|
: property_(property), |
||||||
|
ctx_(iter->ctx), |
||||||
|
index_(iter->index), |
||||||
|
name_(iter->name) {} |
||||||
|
|
||||||
|
AuthPropertyIterator::~AuthPropertyIterator() {} |
||||||
|
|
||||||
|
AuthPropertyIterator& AuthPropertyIterator::operator++() { |
||||||
|
grpc_auth_property_iterator iter = {ctx_, index_, name_}; |
||||||
|
property_ = grpc_auth_property_iterator_next(&iter); |
||||||
|
ctx_ = iter.ctx; |
||||||
|
index_ = iter.index; |
||||||
|
name_ = iter.name; |
||||||
|
return *this; |
||||||
|
} |
||||||
|
|
||||||
|
AuthPropertyIterator AuthPropertyIterator::operator++(int) { |
||||||
|
AuthPropertyIterator tmp(*this); |
||||||
|
operator++(); |
||||||
|
return tmp; |
||||||
|
} |
||||||
|
|
||||||
|
bool AuthPropertyIterator::operator==( |
||||||
|
const AuthPropertyIterator& rhs) const { |
||||||
|
if (property_ == nullptr || rhs.property_ == nullptr) { |
||||||
|
return property_ == rhs.property_; |
||||||
|
} else { |
||||||
|
return index_ == rhs.index_; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
bool AuthPropertyIterator::operator!=( |
||||||
|
const AuthPropertyIterator& rhs) const { |
||||||
|
return !operator==(rhs); |
||||||
|
} |
||||||
|
|
||||||
|
const AuthProperty AuthPropertyIterator::operator*() { |
||||||
|
return std::make_pair<grpc::string, grpc::string>( |
||||||
|
grpc::string(property_->name), |
||||||
|
grpc::string(property_->value, property_->value_length)); |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace grpc
|
@ -0,0 +1,101 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2015, Google Inc. |
||||||
|
* All rights reserved. |
||||||
|
* |
||||||
|
* Redistribution and use in source and binary forms, with or without |
||||||
|
* modification, are permitted provided that the following conditions are |
||||||
|
* met: |
||||||
|
* |
||||||
|
* * Redistributions of source code must retain the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer. |
||||||
|
* * Redistributions in binary form must reproduce the above |
||||||
|
* copyright notice, this list of conditions and the following disclaimer |
||||||
|
* in the documentation and/or other materials provided with the |
||||||
|
* distribution. |
||||||
|
* * Neither the name of Google Inc. nor the names of its |
||||||
|
* contributors may be used to endorse or promote products derived from |
||||||
|
* this software without specific prior written permission. |
||||||
|
* |
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||||
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||||
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||||
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||||
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||||
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||||
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||||
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||||
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#include <grpc++/auth_context.h> |
||||||
|
#include <gtest/gtest.h> |
||||||
|
#include "src/cpp/common/secure_auth_context.h" |
||||||
|
#include "src/core/security/security_context.h" |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
namespace { |
||||||
|
|
||||||
|
class TestAuthPropertyIterator : public AuthPropertyIterator { |
||||||
|
public: |
||||||
|
TestAuthPropertyIterator() {} |
||||||
|
TestAuthPropertyIterator(const grpc_auth_property* property, |
||||||
|
const grpc_auth_property_iterator* iter) |
||||||
|
: AuthPropertyIterator(property, iter) {} |
||||||
|
}; |
||||||
|
|
||||||
|
class AuthPropertyIteratorTest : public ::testing::Test { |
||||||
|
protected: |
||||||
|
void SetUp() GRPC_OVERRIDE { |
||||||
|
ctx_ = grpc_auth_context_create(NULL, 3); |
||||||
|
ctx_->properties[0] = grpc_auth_property_init_from_cstring("name", "chapi"); |
||||||
|
ctx_->properties[1] = grpc_auth_property_init_from_cstring("name", "chapo"); |
||||||
|
ctx_->properties[2] = grpc_auth_property_init_from_cstring("foo", "bar"); |
||||||
|
ctx_->peer_identity_property_name = ctx_->properties[0].name; |
||||||
|
} |
||||||
|
void TearDown() GRPC_OVERRIDE { |
||||||
|
GRPC_AUTH_CONTEXT_UNREF(ctx_, "AuthPropertyIteratorTest"); |
||||||
|
} |
||||||
|
grpc_auth_context* ctx_; |
||||||
|
|
||||||
|
}; |
||||||
|
|
||||||
|
TEST_F(AuthPropertyIteratorTest, DefaultCtor) { |
||||||
|
TestAuthPropertyIterator iter1; |
||||||
|
TestAuthPropertyIterator iter2; |
||||||
|
EXPECT_EQ(iter1, iter2); |
||||||
|
} |
||||||
|
|
||||||
|
TEST_F(AuthPropertyIteratorTest, GeneralTest) { |
||||||
|
grpc_auth_property_iterator c_iter = |
||||||
|
grpc_auth_context_property_iterator(ctx_); |
||||||
|
const grpc_auth_property* property = |
||||||
|
grpc_auth_property_iterator_next(&c_iter); |
||||||
|
TestAuthPropertyIterator iter(property, &c_iter); |
||||||
|
TestAuthPropertyIterator empty_iter; |
||||||
|
EXPECT_FALSE(iter == empty_iter); |
||||||
|
AuthProperty p0 = *iter; |
||||||
|
++iter; |
||||||
|
AuthProperty p1 = *iter; |
||||||
|
iter++; |
||||||
|
AuthProperty p2 = *iter; |
||||||
|
EXPECT_EQ("name", p0.first); |
||||||
|
EXPECT_EQ("chapi", p0.second); |
||||||
|
EXPECT_EQ("name", p1.first); |
||||||
|
EXPECT_EQ("chapo", p1.second); |
||||||
|
EXPECT_EQ("foo", p2.first); |
||||||
|
EXPECT_EQ("bar", p2.second); |
||||||
|
++iter; |
||||||
|
EXPECT_EQ(empty_iter, iter); |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace
|
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
int main(int argc, char **argv) { |
||||||
|
::testing::InitGoogleTest(&argc, argv); |
||||||
|
return RUN_ALL_TESTS(); |
||||||
|
} |
Loading…
Reference in new issue