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
|
// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/policy/test_support/request_handler_for_auto_enrollment.h"
#include "components/policy/core/common/cloud/cloud_policy_constants.h"
#include "components/policy/proto/device_management_backend.pb.h"
#include "components/policy/test_support/client_storage.h"
#include "components/policy/test_support/policy_storage.h"
#include "components/policy/test_support/test_server_helpers.h"
#include "net/base/url_util.h"
#include "net/http/http_status_code.h"
#include "net/test/embedded_test_server/http_request.h"
#include "net/test/embedded_test_server/http_response.h"
using net::test_server::HttpRequest;
using net::test_server::HttpResponse;
namespace em = enterprise_management;
namespace policy {
namespace {
void AddHashes(const std::vector<std::string>& hashes,
em::DeviceAutoEnrollmentResponse* response) {
for (const std::string& hash : hashes) {
*response->add_hashes() = hash;
}
}
} // namespace
RequestHandlerForAutoEnrollment::RequestHandlerForAutoEnrollment(
EmbeddedPolicyTestServer* parent)
: EmbeddedPolicyTestServer::RequestHandler(parent) {}
RequestHandlerForAutoEnrollment::~RequestHandlerForAutoEnrollment() = default;
std::string RequestHandlerForAutoEnrollment::RequestType() {
return dm_protocol::kValueRequestAutoEnrollment;
}
std::unique_ptr<HttpResponse> RequestHandlerForAutoEnrollment::HandleRequest(
const HttpRequest& request) {
em::DeviceManagementRequest device_management_request;
device_management_request.ParseFromString(request.content);
const em::DeviceAutoEnrollmentRequest& enrollment_request =
device_management_request.auto_enrollment_request();
em::DeviceManagementResponse device_management_response;
em::DeviceAutoEnrollmentResponse* enrollment_response =
device_management_response.mutable_auto_enrollment_response();
switch (enrollment_request.modulus()) {
case 1:
if (enrollment_request.enrollment_check_type() ==
enterprise_management::DeviceAutoEnrollmentRequest::
ENROLLMENT_CHECK_TYPE_FRE) {
AddHashes(
client_storage()->GetMatchingStateKeyHashes(
enrollment_request.modulus(), enrollment_request.remainder()),
enrollment_response);
} else if (enrollment_request.enrollment_check_type() ==
enterprise_management::DeviceAutoEnrollmentRequest::
ENROLLMENT_CHECK_TYPE_FORCED_ENROLLMENT) {
AddHashes(
policy_storage()->GetMatchingSerialHashes(
enrollment_request.modulus(), enrollment_request.remainder()),
enrollment_response);
}
break;
case 32:
enrollment_response->set_expected_modulus(1);
break;
}
return CreateHttpResponse(net::HTTP_OK, device_management_response);
}
} // namespace policy
|
