| // Copyright 2014 The Chromium OS 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 <brillo/mime_utils.h> |
| |
| #include <algorithm> |
| #include <base/strings/string_util.h> |
| #include <brillo/strings/string_utils.h> |
| |
| namespace brillo { |
| |
| // *************************************************************************** |
| // ******************************* MIME types ******************************** |
| // *************************************************************************** |
| const char mime::types::kApplication[] = "application"; |
| const char mime::types::kAudio[] = "audio"; |
| const char mime::types::kImage[] = "image"; |
| const char mime::types::kMessage[] = "message"; |
| const char mime::types::kMultipart[] = "multipart"; |
| const char mime::types::kText[] = "text"; |
| const char mime::types::kVideo[] = "video"; |
| |
| const char mime::parameters::kCharset[] = "charset"; |
| |
| const char mime::image::kJpeg[] = "image/jpeg"; |
| const char mime::image::kPng[] = "image/png"; |
| const char mime::image::kBmp[] = "image/bmp"; |
| const char mime::image::kTiff[] = "image/tiff"; |
| const char mime::image::kGif[] = "image/gif"; |
| |
| const char mime::text::kPlain[] = "text/plain"; |
| const char mime::text::kHtml[] = "text/html"; |
| const char mime::text::kXml[] = "text/xml"; |
| |
| const char mime::application::kOctet_stream[] = "application/octet-stream"; |
| const char mime::application::kJson[] = "application/json"; |
| const char mime::application::kWwwFormUrlEncoded[] = |
| "application/x-www-form-urlencoded"; |
| const char mime::application::kProtobuf[] = "application/x-protobuf"; |
| |
| const char mime::multipart::kFormData[] = "multipart/form-data"; |
| const char mime::multipart::kMixed[] = "multipart/mixed"; |
| |
| // *************************************************************************** |
| // **************************** Utility Functions **************************** |
| // *************************************************************************** |
| static std::string EncodeParam(const std::string& param) { |
| // If the string contains one of "tspecials" characters as |
| // specified in RFC 1521, enclose it in quotes. |
| if (param.find_first_of("()<>@,;:\\\"/[]?=") != std::string::npos) { |
| return '"' + param + '"'; |
| } |
| return param; |
| } |
| |
| static std::string DecodeParam(const std::string& param) { |
| if (param.size() > 1 && param.front() == '"' && param.back() == '"') { |
| return param.substr(1, param.size() - 2); |
| } |
| return param; |
| } |
| |
| // *************************************************************************** |
| // ******************** Main MIME manipulation functions ********************* |
| // *************************************************************************** |
| |
| bool mime::Split(const std::string& mime_string, |
| std::string* type, |
| std::string* subtype, |
| mime::Parameters* parameters) { |
| std::vector<std::string> parts = |
| brillo::string_utils::Split(mime_string, ";"); |
| if (parts.empty()) |
| return false; |
| |
| if (!mime::Split(parts.front(), type, subtype)) |
| return false; |
| |
| if (parameters) { |
| parameters->clear(); |
| parameters->reserve(parts.size() - 1); |
| for (size_t i = 1; i < parts.size(); i++) { |
| auto pair = brillo::string_utils::SplitAtFirst(parts[i], "="); |
| pair.second = DecodeParam(pair.second); |
| parameters->push_back(pair); |
| } |
| } |
| return true; |
| } |
| |
| bool mime::Split(const std::string& mime_string, |
| std::string* type, |
| std::string* subtype) { |
| std::string mime = mime::RemoveParameters(mime_string); |
| auto types = brillo::string_utils::SplitAtFirst(mime, "/"); |
| |
| if (type) |
| *type = types.first; |
| |
| if (subtype) |
| *subtype = types.second; |
| |
| return !types.first.empty() && !types.second.empty(); |
| } |
| |
| std::string mime::Combine(const std::string& type, |
| const std::string& subtype, |
| const mime::Parameters& parameters) { |
| std::vector<std::string> parts; |
| parts.push_back(brillo::string_utils::Join("/", type, subtype)); |
| for (const auto& pair : parameters) { |
| parts.push_back( |
| brillo::string_utils::Join("=", pair.first, EncodeParam(pair.second))); |
| } |
| return brillo::string_utils::Join("; ", parts); |
| } |
| |
| std::string mime::GetType(const std::string& mime_string) { |
| std::string mime = mime::RemoveParameters(mime_string); |
| return brillo::string_utils::SplitAtFirst(mime, "/").first; |
| } |
| |
| std::string mime::GetSubtype(const std::string& mime_string) { |
| std::string mime = mime::RemoveParameters(mime_string); |
| return brillo::string_utils::SplitAtFirst(mime, "/").second; |
| } |
| |
| mime::Parameters mime::GetParameters(const std::string& mime_string) { |
| std::string type; |
| std::string subtype; |
| mime::Parameters parameters; |
| |
| if (mime::Split(mime_string, &type, &subtype, ¶meters)) |
| return parameters; |
| |
| return mime::Parameters(); |
| } |
| |
| std::string mime::RemoveParameters(const std::string& mime_string) { |
| return brillo::string_utils::SplitAtFirst(mime_string, ";").first; |
| } |
| |
| std::string mime::AppendParameter(const std::string& mime_string, |
| const std::string& paramName, |
| const std::string& paramValue) { |
| std::string mime(mime_string); |
| mime += "; "; |
| mime += brillo::string_utils::Join("=", paramName, EncodeParam(paramValue)); |
| return mime; |
| } |
| |
| std::string mime::GetParameterValue(const std::string& mime_string, |
| const std::string& paramName) { |
| mime::Parameters params = mime::GetParameters(mime_string); |
| for (const auto& pair : params) { |
| if (base::EqualsCaseInsensitiveASCII(pair.first.c_str(), paramName.c_str())) |
| return pair.second; |
| } |
| return std::string(); |
| } |
| |
| } // namespace brillo |