127f654740f2a26ad62a5c155af9199af9e69b889claireho// Copyright (C) 2009-2010, International Business Machines 285bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho// Corporation and others. All Rights Reserved. 385bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho// 485bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho// Copyright 2007 Google Inc. All Rights Reserved. 585bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho// Author: sanjay@google.com (Sanjay Ghemawat) 685bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho 785bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho#include "unicode/utypes.h" 885bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho#include "unicode/bytestream.h" 985bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho#include "cmemory.h" 1085bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho 1185bf2e2fbc60a9f938064abc8127d61da7d19882Claire HoU_NAMESPACE_BEGIN 1285bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho 1385bf2e2fbc60a9f938064abc8127d61da7d19882Claire Hochar* ByteSink::GetAppendBuffer(int32_t min_capacity, 1485bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho int32_t /*desired_capacity_hint*/, 1585bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho char* scratch, int32_t scratch_capacity, 1685bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho int32_t* result_capacity) { 1785bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho if (min_capacity < 1 || scratch_capacity < min_capacity) { 1885bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho *result_capacity = 0; 1985bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho return NULL; 2085bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho } 2185bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho *result_capacity = scratch_capacity; 2285bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho return scratch; 2385bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho} 2485bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho 2585bf2e2fbc60a9f938064abc8127d61da7d19882Claire Hovoid ByteSink::Flush() {} 2685bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho 2785bf2e2fbc60a9f938064abc8127d61da7d19882Claire HoCheckedArrayByteSink::CheckedArrayByteSink(char* outbuf, int32_t capacity) 2827f654740f2a26ad62a5c155af9199af9e69b889claireho : outbuf_(outbuf), capacity_(capacity < 0 ? 0 : capacity), 2927f654740f2a26ad62a5c155af9199af9e69b889claireho size_(0), appended_(0), overflowed_(FALSE) { 3027f654740f2a26ad62a5c155af9199af9e69b889claireho} 3127f654740f2a26ad62a5c155af9199af9e69b889claireho 3227f654740f2a26ad62a5c155af9199af9e69b889clairehoCheckedArrayByteSink& CheckedArrayByteSink::Reset() { 3327f654740f2a26ad62a5c155af9199af9e69b889claireho size_ = appended_ = 0; 3427f654740f2a26ad62a5c155af9199af9e69b889claireho overflowed_ = FALSE; 3527f654740f2a26ad62a5c155af9199af9e69b889claireho return *this; 3685bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho} 3785bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho 3885bf2e2fbc60a9f938064abc8127d61da7d19882Claire Hovoid CheckedArrayByteSink::Append(const char* bytes, int32_t n) { 3985bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho if (n <= 0) { 4085bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho return; 4185bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho } 4227f654740f2a26ad62a5c155af9199af9e69b889claireho appended_ += n; 4385bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho int32_t available = capacity_ - size_; 4485bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho if (n > available) { 4585bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho n = available; 4627f654740f2a26ad62a5c155af9199af9e69b889claireho overflowed_ = TRUE; 4785bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho } 4885bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho if (n > 0 && bytes != (outbuf_ + size_)) { 4985bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho uprv_memcpy(outbuf_ + size_, bytes, n); 5085bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho } 5185bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho size_ += n; 5285bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho} 5385bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho 5485bf2e2fbc60a9f938064abc8127d61da7d19882Claire Hochar* CheckedArrayByteSink::GetAppendBuffer(int32_t min_capacity, 5585bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho int32_t /*desired_capacity_hint*/, 5685bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho char* scratch, 5785bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho int32_t scratch_capacity, 5885bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho int32_t* result_capacity) { 5985bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho if (min_capacity < 1 || scratch_capacity < min_capacity) { 6085bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho *result_capacity = 0; 6185bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho return NULL; 6285bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho } 6385bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho int32_t available = capacity_ - size_; 6485bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho if (available >= min_capacity) { 6585bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho *result_capacity = available; 6685bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho return outbuf_ + size_; 6785bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho } else { 6885bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho *result_capacity = scratch_capacity; 6985bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho return scratch; 7085bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho } 7185bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho} 7285bf2e2fbc60a9f938064abc8127d61da7d19882Claire Ho 7385bf2e2fbc60a9f938064abc8127d61da7d19882Claire HoU_NAMESPACE_END 74