50 lines
1.4 KiB
C++
50 lines
1.4 KiB
C++
// Copyright 2018 Google LLC. All Rights Reserved. This file and proprietary
|
|
// source code may only be used and distributed under the Widevine License
|
|
// Agreement.
|
|
#ifndef CDM_TEST_URL_REQUEST_H_
|
|
#define CDM_TEST_URL_REQUEST_H_
|
|
|
|
#include <map>
|
|
#include <string>
|
|
|
|
#include "http_socket.h"
|
|
#include "wv_class_utils.h"
|
|
|
|
namespace wvcdm {
|
|
// Provides simple HTTP request and response service.
|
|
// Only POST request method is implemented.
|
|
class UrlRequest {
|
|
public:
|
|
UrlRequest() = delete;
|
|
WVCDM_DISALLOW_COPY_AND_MOVE(UrlRequest);
|
|
|
|
explicit UrlRequest(const std::string& url);
|
|
~UrlRequest();
|
|
|
|
bool is_connected() const { return is_connected_; }
|
|
void Reconnect();
|
|
|
|
bool PostRequest(const std::string& data);
|
|
bool PostCertRequestInQueryString(const std::string& data);
|
|
|
|
bool GetResponse(std::string* message);
|
|
static int GetStatusCode(const std::string& response);
|
|
// Get the response, and expect the status is OK.
|
|
// It will retry if the response code is in the 500 range.
|
|
void AssertOkResponseWithRetry(std::string* message);
|
|
|
|
static bool GetDebugHeaderFields(
|
|
const std::string& response,
|
|
std::map<std::string, std::string>* header_fields);
|
|
|
|
private:
|
|
bool PostRequestWithPath(const std::string& path, const std::string& data);
|
|
bool SendRequestOnce();
|
|
|
|
bool is_connected_;
|
|
HttpSocket socket_;
|
|
std::string request_;
|
|
}; // class UrlRequest
|
|
} // namespace wvcdm
|
|
#endif // CDM_TEST_URL_REQUEST_H_
|