mirror of
https://github.com/RPCS3/llvm.git
synced 2024-11-28 14:10:55 +00:00
01337d5a14
As a follow up of the bad alloc handler patch, this patch introduces nullptr checks on pointers returned from the malloc/realloc/calloc functions. In addition some memory size assignments are moved behind the allocation of the corresponding memory to fulfill exception safe memory management (RAII). patch by Klaus Kretzschmar Differential Revision: https://reviews.llvm.org/D35414 git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@308576 91177308-0d34-0410-b5e6-96231b3b80d8
45 lines
1.7 KiB
C++
45 lines
1.7 KiB
C++
//===- llvm/ADT/SmallVector.cpp - 'Normally small' vectors ----------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements the SmallVector class.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ADT/SmallVector.h"
|
|
using namespace llvm;
|
|
|
|
/// grow_pod - This is an implementation of the grow() method which only works
|
|
/// on POD-like datatypes and is out of line to reduce code duplication.
|
|
void SmallVectorBase::grow_pod(void *FirstEl, size_t MinSizeInBytes,
|
|
size_t TSize) {
|
|
size_t CurSizeBytes = size_in_bytes();
|
|
size_t NewCapacityInBytes = 2 * capacity_in_bytes() + TSize; // Always grow.
|
|
if (NewCapacityInBytes < MinSizeInBytes)
|
|
NewCapacityInBytes = MinSizeInBytes;
|
|
|
|
void *NewElts;
|
|
if (BeginX == FirstEl) {
|
|
NewElts = malloc(NewCapacityInBytes);
|
|
if (NewElts == nullptr)
|
|
report_bad_alloc_error("Allocation of SmallVector element failed.");
|
|
|
|
// Copy the elements over. No need to run dtors on PODs.
|
|
memcpy(NewElts, this->BeginX, CurSizeBytes);
|
|
} else {
|
|
// If this wasn't grown from the inline copy, grow the allocated space.
|
|
NewElts = realloc(this->BeginX, NewCapacityInBytes);
|
|
if (NewElts == nullptr)
|
|
report_bad_alloc_error("Reallocation of SmallVector element failed.");
|
|
}
|
|
|
|
this->EndX = (char*)NewElts+CurSizeBytes;
|
|
this->BeginX = NewElts;
|
|
this->CapacityX = (char*)this->BeginX + NewCapacityInBytes;
|
|
}
|