| Index: webrtc/base/cryptstring.h
|
| diff --git a/webrtc/base/cryptstring.h b/webrtc/base/cryptstring.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..e1ee309f65cffd9eeef95ee1483fae32ef65c656
|
| --- /dev/null
|
| +++ b/webrtc/base/cryptstring.h
|
| @@ -0,0 +1,167 @@
|
| +/*
|
| + * Copyright 2004 The WebRTC Project Authors. All rights reserved.
|
| + *
|
| + * Use of this source code is governed by a BSD-style license
|
| + * that can be found in the LICENSE file in the root of the source
|
| + * tree. An additional intellectual property rights grant can be found
|
| + * in the file PATENTS. All contributing project authors may
|
| + * be found in the AUTHORS file in the root of the source tree.
|
| + */
|
| +
|
| +#ifndef _WEBRTC_BASE_CRYPTSTRING_H_
|
| +#define _WEBRTC_BASE_CRYPTSTRING_H_
|
| +
|
| +#include <string.h>
|
| +
|
| +#include <memory>
|
| +#include <string>
|
| +#include <vector>
|
| +
|
| +namespace rtc {
|
| +
|
| +class CryptStringImpl {
|
| +public:
|
| + virtual ~CryptStringImpl() {}
|
| + virtual size_t GetLength() const = 0;
|
| + virtual void CopyTo(char * dest, bool nullterminate) const = 0;
|
| + virtual std::string UrlEncode() const = 0;
|
| + virtual CryptStringImpl * Copy() const = 0;
|
| + virtual void CopyRawTo(std::vector<unsigned char> * dest) const = 0;
|
| +};
|
| +
|
| +class EmptyCryptStringImpl : public CryptStringImpl {
|
| +public:
|
| + ~EmptyCryptStringImpl() override {}
|
| + size_t GetLength() const override;
|
| + void CopyTo(char* dest, bool nullterminate) const override;
|
| + std::string UrlEncode() const override;
|
| + CryptStringImpl* Copy() const override;
|
| + void CopyRawTo(std::vector<unsigned char>* dest) const override;
|
| +};
|
| +
|
| +class CryptString {
|
| + public:
|
| + CryptString();
|
| + size_t GetLength() const { return impl_->GetLength(); }
|
| + void CopyTo(char * dest, bool nullterminate) const { impl_->CopyTo(dest, nullterminate); }
|
| + CryptString(const CryptString& other);
|
| + explicit CryptString(const CryptStringImpl& impl);
|
| + ~CryptString();
|
| + CryptString & operator=(const CryptString & other) {
|
| + if (this != &other) {
|
| + impl_.reset(other.impl_->Copy());
|
| + }
|
| + return *this;
|
| + }
|
| + void Clear() { impl_.reset(new EmptyCryptStringImpl()); }
|
| + std::string UrlEncode() const { return impl_->UrlEncode(); }
|
| + void CopyRawTo(std::vector<unsigned char> * dest) const {
|
| + return impl_->CopyRawTo(dest);
|
| + }
|
| +
|
| + private:
|
| + std::unique_ptr<const CryptStringImpl> impl_;
|
| +};
|
| +
|
| +
|
| +// Used for constructing strings where a password is involved and we
|
| +// need to ensure that we zero memory afterwards
|
| +class FormatCryptString {
|
| +public:
|
| + FormatCryptString() {
|
| + storage_ = new char[32];
|
| + capacity_ = 32;
|
| + length_ = 0;
|
| + storage_[0] = 0;
|
| + }
|
| +
|
| + void Append(const std::string & text) {
|
| + Append(text.data(), text.length());
|
| + }
|
| +
|
| + void Append(const char * data, size_t length) {
|
| + EnsureStorage(length_ + length + 1);
|
| + memcpy(storage_ + length_, data, length);
|
| + length_ += length;
|
| + storage_[length_] = '\0';
|
| + }
|
| +
|
| + void Append(const CryptString * password) {
|
| + size_t len = password->GetLength();
|
| + EnsureStorage(length_ + len + 1);
|
| + password->CopyTo(storage_ + length_, true);
|
| + length_ += len;
|
| + }
|
| +
|
| + size_t GetLength() {
|
| + return length_;
|
| + }
|
| +
|
| + const char * GetData() {
|
| + return storage_;
|
| + }
|
| +
|
| +
|
| + // Ensures storage of at least n bytes
|
| + void EnsureStorage(size_t n) {
|
| + if (capacity_ >= n) {
|
| + return;
|
| + }
|
| +
|
| + size_t old_capacity = capacity_;
|
| + char * old_storage = storage_;
|
| +
|
| + for (;;) {
|
| + capacity_ *= 2;
|
| + if (capacity_ >= n)
|
| + break;
|
| + }
|
| +
|
| + storage_ = new char[capacity_];
|
| +
|
| + if (old_capacity) {
|
| + memcpy(storage_, old_storage, length_);
|
| +
|
| + // zero memory in a way that an optimizer won't optimize it out
|
| + old_storage[0] = 0;
|
| + for (size_t i = 1; i < old_capacity; i++) {
|
| + old_storage[i] = old_storage[i - 1];
|
| + }
|
| + delete[] old_storage;
|
| + }
|
| + }
|
| +
|
| + ~FormatCryptString() {
|
| + if (capacity_) {
|
| + storage_[0] = 0;
|
| + for (size_t i = 1; i < capacity_; i++) {
|
| + storage_[i] = storage_[i - 1];
|
| + }
|
| + }
|
| + delete[] storage_;
|
| + }
|
| +private:
|
| + char * storage_;
|
| + size_t capacity_;
|
| + size_t length_;
|
| +};
|
| +
|
| +class InsecureCryptStringImpl : public CryptStringImpl {
|
| + public:
|
| + std::string& password() { return password_; }
|
| + const std::string& password() const { return password_; }
|
| +
|
| + ~InsecureCryptStringImpl() override = default;
|
| + size_t GetLength() const override;
|
| + void CopyTo(char* dest, bool nullterminate) const override;
|
| + std::string UrlEncode() const override;
|
| + CryptStringImpl* Copy() const override;
|
| + void CopyRawTo(std::vector<unsigned char>* dest) const override;
|
| +
|
| + private:
|
| + std::string password_;
|
| +};
|
| +
|
| +}
|
| +
|
| +#endif // _WEBRTC_BASE_CRYPTSTRING_H_
|
|
|