зеркало из https://github.com/mozilla/gecko-dev.git
162 строки
3.9 KiB
C++
162 строки
3.9 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
|
|
*
|
|
* The contents of this file are subject to the Netscape Public
|
|
* License Version 1.1 (the "License"); you may not use this file
|
|
* except in compliance with the License. You may obtain a copy of
|
|
* the License at http://www.mozilla.org/NPL/
|
|
*
|
|
* Software distributed under the License is distributed on an "AS
|
|
* IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
|
|
* implied. See the License for the specific language governing
|
|
* rights and limitations under the License.
|
|
*
|
|
* The Original Code is mozilla.org code.
|
|
*
|
|
* The Initial Developer of the Original Code is Netscape
|
|
* Communications Corporation. Portions created by Netscape are
|
|
* Copyright (C) 1998 Netscape Communications Corporation. All
|
|
* Rights Reserved.
|
|
*
|
|
* Contributor(s):
|
|
*/
|
|
|
|
#include "nsUnicharBuffer.h"
|
|
#include "nsIUnicharInputStream.h"
|
|
#include "nsCRT.h"
|
|
|
|
#define MIN_BUFFER_SIZE 32
|
|
|
|
UnicharBufferImpl::UnicharBufferImpl()
|
|
: mBuffer(NULL), mSpace(0), mLength(0)
|
|
{
|
|
NS_INIT_REFCNT();
|
|
}
|
|
|
|
NS_METHOD
|
|
UnicharBufferImpl::Create(nsISupports *aOuter, REFNSIID aIID, void **aResult)
|
|
{
|
|
if (aOuter)
|
|
return NS_ERROR_NO_AGGREGATION;
|
|
|
|
UnicharBufferImpl* it = new UnicharBufferImpl();
|
|
if (it == nsnull)
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
NS_ADDREF(it);
|
|
nsresult rv = it->QueryInterface(aIID, aResult);
|
|
NS_RELEASE(it);
|
|
return rv;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
UnicharBufferImpl::Init(PRUint32 aBufferSize)
|
|
{
|
|
if (aBufferSize < MIN_BUFFER_SIZE) {
|
|
aBufferSize = MIN_BUFFER_SIZE;
|
|
}
|
|
mSpace = aBufferSize;
|
|
mLength = 0;
|
|
mBuffer = new PRUnichar[aBufferSize];
|
|
return mBuffer ? NS_OK : NS_ERROR_OUT_OF_MEMORY;
|
|
}
|
|
|
|
NS_IMPL_ISUPPORTS1(UnicharBufferImpl, nsIUnicharBuffer)
|
|
|
|
UnicharBufferImpl::~UnicharBufferImpl()
|
|
{
|
|
if (nsnull != mBuffer) {
|
|
delete[] mBuffer;
|
|
mBuffer = nsnull;
|
|
}
|
|
mLength = 0;
|
|
}
|
|
|
|
NS_IMETHODIMP_(PRInt32)
|
|
UnicharBufferImpl::GetLength() const
|
|
{
|
|
return mLength;
|
|
}
|
|
|
|
NS_IMETHODIMP_(PRInt32)
|
|
UnicharBufferImpl::GetBufferSize() const
|
|
{
|
|
return mSpace;
|
|
}
|
|
|
|
NS_IMETHODIMP_(PRUnichar*)
|
|
UnicharBufferImpl::GetBuffer() const
|
|
{
|
|
return mBuffer;
|
|
}
|
|
|
|
NS_IMETHODIMP_(PRBool)
|
|
UnicharBufferImpl::Grow(PRInt32 aNewSize)
|
|
{
|
|
if (PRUint32(aNewSize) < MIN_BUFFER_SIZE) {
|
|
aNewSize = MIN_BUFFER_SIZE;
|
|
}
|
|
PRUnichar* newbuf = new PRUnichar[aNewSize];
|
|
if (nsnull != newbuf) {
|
|
if (0 != mLength) {
|
|
nsCRT::memcpy(newbuf, mBuffer, mLength * sizeof(PRUnichar));
|
|
}
|
|
delete[] mBuffer;
|
|
mBuffer = newbuf;
|
|
return PR_TRUE;
|
|
}
|
|
return PR_FALSE;
|
|
}
|
|
|
|
NS_IMETHODIMP_(PRInt32)
|
|
UnicharBufferImpl::Fill(nsresult* aErrorCode,
|
|
nsIUnicharInputStream* aStream,
|
|
PRInt32 aKeep)
|
|
{
|
|
NS_PRECONDITION(nsnull != aStream, "null stream");
|
|
NS_PRECONDITION(PRUint32(aKeep) < PRUint32(mLength), "illegal keep count");
|
|
if ((nsnull == aStream) || (PRUint32(aKeep) >= PRUint32(mLength))) {
|
|
// whoops
|
|
*aErrorCode = NS_BASE_STREAM_ILLEGAL_ARGS;
|
|
return -1;
|
|
}
|
|
|
|
if (0 != aKeep) {
|
|
// Slide over kept data
|
|
nsCRT::memmove(mBuffer, mBuffer + (mLength - aKeep),
|
|
aKeep * sizeof(PRUnichar));
|
|
}
|
|
|
|
// Read in some new data
|
|
mLength = aKeep;
|
|
PRInt32 amount = mSpace - aKeep;
|
|
PRUint32 nb;
|
|
NS_ASSERTION(aKeep >= 0, "unsigned madness");
|
|
NS_ASSERTION(amount >= 0, "unsigned madness");
|
|
*aErrorCode = aStream->Read(mBuffer, (PRUint32)aKeep, (PRUint32)amount, &nb);
|
|
if (NS_SUCCEEDED(*aErrorCode)) {
|
|
mLength += nb;
|
|
}
|
|
else
|
|
nb = 0;
|
|
return nb;
|
|
}
|
|
|
|
NS_COM nsresult
|
|
NS_NewUnicharBuffer(nsIUnicharBuffer** aInstancePtrResult,
|
|
nsISupports* aOuter,
|
|
PRUint32 aBufferSize)
|
|
{
|
|
nsresult rv;
|
|
nsIUnicharBuffer* buf;
|
|
rv = UnicharBufferImpl::Create(aOuter, nsIUnicharBuffer::GetIID(),
|
|
(void**)&buf);
|
|
if (NS_FAILED(rv)) return rv;
|
|
rv = buf->Init(aBufferSize);
|
|
if (NS_FAILED(rv)) {
|
|
NS_RELEASE(buf);
|
|
return rv;
|
|
}
|
|
*aInstancePtrResult = buf;
|
|
return rv;
|
|
}
|