mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
53 lines
1.9 KiB
C++
53 lines
1.9 KiB
C++
//===--- PunycodeUTF8.cpp - Unicode to Punycode transcoding -----*- C++ -*-===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2015 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See http://swift.org/LICENSE.txt for license information
|
|
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "swift/Basic/Punycode.h"
|
|
#include "llvm/Support/ConvertUTF.h"
|
|
#include <vector>
|
|
|
|
using namespace swift;
|
|
|
|
bool Punycode::encodePunycodeUTF8(StringRef InputUTF8,
|
|
std::string &OutPunycode) {
|
|
std::vector<uint32_t> InputCodePoints(InputUTF8.size());
|
|
const UTF8 *SourceStart = reinterpret_cast<const UTF8 *>(InputUTF8.data());
|
|
UTF32 *TargetStart = InputCodePoints.data();
|
|
auto ConvStatus = ConvertUTF8toUTF32(
|
|
&SourceStart, SourceStart + InputUTF8.size(), &TargetStart,
|
|
InputCodePoints.data() + InputCodePoints.size(), strictConversion);
|
|
if (ConvStatus != conversionOK)
|
|
return false;
|
|
return encodePunycode(InputCodePoints, OutPunycode);
|
|
}
|
|
|
|
bool Punycode::decodePunycodeUTF8(StringRef InputPunycode,
|
|
std::string &OutUTF8) {
|
|
std::vector<uint32_t> OutCodePoints;
|
|
if (!decodePunycode(InputPunycode, OutCodePoints))
|
|
return false;
|
|
|
|
const size_t SizeUpperBound = OutCodePoints.size() * 4;
|
|
std::vector<UTF8> Result(SizeUpperBound);
|
|
const UTF32 *SourceStart = OutCodePoints.data();
|
|
UTF8 *TargetStart = Result.data();
|
|
auto ConvStatus = ConvertUTF32toUTF8(
|
|
&SourceStart, SourceStart + OutCodePoints.size(), &TargetStart,
|
|
Result.data() + Result.size(), strictConversion);
|
|
if (ConvStatus != conversionOK) {
|
|
OutUTF8.clear();
|
|
return false;
|
|
}
|
|
OutUTF8 = std::string(Result.data(), TargetStart);
|
|
return true;
|
|
}
|
|
|