summaryrefslogtreecommitdiffstats
path: root/chromium/components/policy/test_support/request_handler_for_register_cert_based.cc
blob: eded2038f25f1e3cb19b3f0c493b4cb0695424fd (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
// Copyright 2021 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 "components/policy/test_support/request_handler_for_register_cert_based.h"

#include <set>
#include <string>

#include "base/guid.h"
#include "base/notreached.h"
#include "base/strings/stringprintf.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/request_handler_for_register_device_and_user.h"
#include "components/policy/test_support/test_server_helpers.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 {

RequestHandlerForRegisterCertBased::RequestHandlerForRegisterCertBased(
    ClientStorage* client_storage,
    PolicyStorage* policy_storage)
    : RequestHandlerForRegisterDeviceAndUser(client_storage, policy_storage) {}

RequestHandlerForRegisterCertBased::~RequestHandlerForRegisterCertBased() =
    default;

std::string RequestHandlerForRegisterCertBased::RequestType() {
  return dm_protocol::kValueRequestCertBasedRegister;
}

std::unique_ptr<HttpResponse> RequestHandlerForRegisterCertBased::HandleRequest(
    const HttpRequest& request) {
  em::DeviceManagementRequest device_management_request;
  device_management_request.ParseFromString(request.content);
  const em::SignedData& signed_req =
      device_management_request.certificate_based_register_request()
          .signed_request();
  em::CertificateBasedDeviceRegistrationData parsed_req;
  std::string data = signed_req.data().substr(
      0, signed_req.data().size() - signed_req.extra_data_bytes());
  if (!parsed_req.ParseFromString(data))
    return CreateHttpResponse(net::HTTP_BAD_REQUEST, "Invalid request");
  if (parsed_req.certificate_type() !=
      em::CertificateBasedDeviceRegistrationData::
          ENTERPRISE_ENROLLMENT_CERTIFICATE) {
    return CreateHttpResponse(net::HTTP_FORBIDDEN,
                              "Invalid certificate type for registration");
  }
  const em::DeviceRegisterRequest& register_request =
      parsed_req.device_register_request();

  return RegisterDeviceAndSendResponse(request, register_request, "");
}

}  // namespace policy