summaryrefslogtreecommitdiffstats
path: root/chrome/browser/local_discovery/privetv3_crypto_provider.cc
blob: df27aa62c41a3ca055aab929ad57ef3efcf37f51 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/local_discovery/privetv3_crypto_provider.h"

#include "base/logging.h"

namespace local_discovery {

namespace {

// A stub session type used for development/debugging.
const char kAuthMethodEmpty[] = "empty";

const char kHandshakeStateComplete[] = "complete";
const char kStubVerificationCode[] = "SAMPLE";
}

class PrivetV3CryptoProviderEmpty : public PrivetV3CryptoProvider {
 public:
  PrivetV3CryptoProviderEmpty();
  virtual ~PrivetV3CryptoProviderEmpty();

  // PrivetV3CryptoProvider implementation.
  virtual HandshakeState GetState() OVERRIDE;
  virtual std::string GetAuthMethod() OVERRIDE;
  virtual HandshakeState GetNextStep(int* step, std::string* package) OVERRIDE;
  virtual HandshakeState SetStepResponse(int step,
                                         const std::string& state,
                                         const std::string& package) OVERRIDE;
  virtual std::string GetVerificationCode() OVERRIDE;
  virtual HandshakeState AcceptVerificationCode() OVERRIDE;
  virtual bool EncryptData(const std::string& input,
                           std::string* output) OVERRIDE;

 private:
  HandshakeState state_;
};

scoped_ptr<PrivetV3CryptoProvider> Create(
    const std::vector<std::string>& available_auth_methods) {
  for (size_t i = 0; i < available_auth_methods.size(); i++) {
    if (available_auth_methods[i] == kAuthMethodEmpty) {
      return scoped_ptr<PrivetV3CryptoProvider>(
          new PrivetV3CryptoProviderEmpty());
    }
  }

  return scoped_ptr<PrivetV3CryptoProvider>();
}

PrivetV3CryptoProviderEmpty::PrivetV3CryptoProviderEmpty()
    : state_(IN_PROGRESS) {
}

PrivetV3CryptoProviderEmpty::~PrivetV3CryptoProviderEmpty() {
}

PrivetV3CryptoProvider::HandshakeState PrivetV3CryptoProviderEmpty::GetState() {
  return state_;
}

std::string PrivetV3CryptoProviderEmpty::GetAuthMethod() {
  return kAuthMethodEmpty;
}

PrivetV3CryptoProvider::HandshakeState PrivetV3CryptoProviderEmpty::GetNextStep(
    int* step,
    std::string* package) {
  DCHECK(state_ == IN_PROGRESS);
  *step = 0;
  package->clear();
  state_ = AWAITING_RESPONSE;

  return state_;
}

PrivetV3CryptoProvider::HandshakeState
PrivetV3CryptoProviderEmpty::SetStepResponse(int step,
                                             const std::string& state,
                                             const std::string& package) {
  DCHECK(state_ == AWAITING_RESPONSE);

  bool success =
      (step == 0 && package.empty() && state == kHandshakeStateComplete);

  if (success) {
    state_ = AWAITING_USER_VERIFICATION;
  } else {
    state_ = HANDSHAKE_ERROR;
  }

  return state_;
}

std::string PrivetV3CryptoProviderEmpty::GetVerificationCode() {
  DCHECK(state_ == AWAITING_USER_VERIFICATION);
  return kStubVerificationCode;
}

PrivetV3CryptoProvider::HandshakeState
PrivetV3CryptoProviderEmpty::AcceptVerificationCode() {
  DCHECK(state_ == AWAITING_USER_VERIFICATION);
  return (state_ = HANDSHAKE_COMPLETE);
}

bool PrivetV3CryptoProviderEmpty::EncryptData(const std::string& input,
                                              std::string* output) {
  *output = input;
  return true;
}

}  // namespace local_discovery