chrome/browser/ash/smb_client/smb_service_helper.cc

// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ash/smb_client/smb_service_helper.h"

#include "base/strings/string_split.h"
#include "base/strings/string_util.h"

namespace ash::smb_client {

bool ParseUserPrincipalName(const std::string& user_principal_name,
                            std::string* user_name,
                            std::string* workgroup) {
  DCHECK(user_name);
  DCHECK(workgroup);
  std::vector<std::string> parts = base::SplitString(
      user_principal_name, "@", base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL);
  if (parts.size() != 2 || parts.at(0).empty() || parts.at(1).empty()) {
    // Don't log user_principal_name, it might contain sensitive data.
    LOG(ERROR) << "Failed to parse user principal name. Expected form "
                  "'user@some.realm'.";
    return false;
  }
  *user_name = std::move(parts.at(0));
  *workgroup = base::ToUpperASCII(std::move(parts.at(1)));
  return true;
}

bool ParseDownLevelLogonName(const std::string& logon_name,
                             std::string* user_name,
                             std::string* workgroup) {
  DCHECK(user_name);
  DCHECK(workgroup);
  std::vector<std::string> parts = base::SplitString(
      logon_name, "\\", base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL);
  if (parts.size() != 2 || parts.at(0).empty() || parts.at(1).empty()) {
    // Don't log logon_name, it might contain sensitive data.
    LOG(ERROR) << "Failed to parse down-level logon name. Expected form "
                  "'DOMAIN\\user'.";
    return false;
  }
  *workgroup = base::ToUpperASCII(std::move(parts.at(0)));
  *user_name = std::move(parts.at(1));
  return true;
}

bool ParseUserName(const std::string& name,
                   std::string* user_name,
                   std::string* workgroup) {
  if (name.find('@') != std::string::npos) {
    return ParseUserPrincipalName(name, user_name, workgroup);
  } else if (name.find('\\') != std::string::npos) {
    return ParseDownLevelLogonName(name, user_name, workgroup);
  }
  // If user principal or down-level logon name format is not detected, fall
  // back to treating the name as a plain user name.
  *user_name = name;
  return true;
}

}  // namespace ash::smb_client