From 246a908734f137e918d88a9f18a4bcb1f18ff8b4 Mon Sep 17 00:00:00 2001
From: Michael Lotz <mmlr@mlotz.ch>
Date: Sat, 13 Sep 2008 16:30:48 +0000
Subject: [PATCH] * CID 1012: Fix leaking the OffsetFile in case it fails to
 initialize. * Use std::nothrow (the rest of the file already used it). * We
 also have to keep the source file around and properly dispose it, as  
 OffsetFile doesn't take ownership of it.

git-svn-id: file:///srv/svn/repos/haiku/haiku/trunk@27486 a95241bf-73f2-0310-859d-f6bbb57e9c96
---
 src/servers/app/ServerPicture.cpp | 25 ++++++++++++++++---------
 src/servers/app/ServerPicture.h   |  3 +++
 2 files changed, 19 insertions(+), 9 deletions(-)

diff --git a/src/servers/app/ServerPicture.cpp b/src/servers/app/ServerPicture.cpp
index 55796ed093..ea352607dc 100644
--- a/src/servers/app/ServerPicture.cpp
+++ b/src/servers/app/ServerPicture.cpp
@@ -754,6 +754,7 @@ const static void *kTableEntries[] = {
 ServerPicture::ServerPicture()
 	:
 	PictureDataWriter(),
+	fFile(NULL),
 	fData(NULL),
 	fPictures(NULL),
 	fUsurped(NULL)
@@ -768,6 +769,7 @@ ServerPicture::ServerPicture()
 ServerPicture::ServerPicture(const ServerPicture &picture)
 	:
 	PictureDataWriter(),
+	fFile(NULL),
 	fData(NULL),
 	fPictures(NULL),
 	fUsurped(NULL)
@@ -793,19 +795,25 @@ ServerPicture::ServerPicture(const ServerPicture &picture)
 ServerPicture::ServerPicture(const char *fileName, const int32 &offset)
 	:
 	PictureDataWriter(),
+	fFile(NULL),
 	fData(NULL),
 	fPictures(NULL),
 	fUsurped(NULL)
 {
-	BPrivate::Storage::OffsetFile *file =
-		new BPrivate::Storage::OffsetFile(new BFile(fileName, B_READ_WRITE), (off_t)offset);
-
-	if (file == NULL || file->InitCheck() != B_OK)
+	fFile = new (std::nothrow) BFile(fileName, B_READ_WRITE);
+	if (fFile == NULL)
 		return;
-	
-	fData = file;
+
+	BPrivate::Storage::OffsetFile *offsetFile =
+		new (std::nothrow) BPrivate::Storage::OffsetFile(fFile, (off_t)offset);
+	if (offsetFile == NULL || offsetFile->InitCheck() != B_OK) {
+		delete offsetFile;
+		return;
+	}
+
+	fData = offsetFile;
 	fToken = gTokenSpace.NewToken(kPictureToken, this);
-	
+
 	PictureDataWriter::SetTo(fData);
 }
 
@@ -813,6 +821,7 @@ ServerPicture::ServerPicture(const char *fileName, const int32 &offset)
 ServerPicture::~ServerPicture()
 {
 	delete fData;
+	delete fFile;
 	gTokenSpace.RemoveToken(fToken);
 	
 	// We only delete the subpictures list, not the subpictures themselves,
@@ -1044,5 +1053,3 @@ ServerPicture::ExportData(BPrivate::PortLink &link)
 	fData->Seek(oldPosition, SEEK_SET);
 	return status;
 }
-
-
diff --git a/src/servers/app/ServerPicture.h b/src/servers/app/ServerPicture.h
index 4f5869645a..ec0427dc4c 100644
--- a/src/servers/app/ServerPicture.h
+++ b/src/servers/app/ServerPicture.h
@@ -16,6 +16,8 @@
 
 class ServerApp;
 class View;
+class BFile;
+
 namespace BPrivate {
 	class LinkReceiver;
 	class PortLink;
@@ -53,6 +55,7 @@ friend class	ServerApp;
 		~ServerPicture();
 
 		int32		fToken;
+		BFile		*fFile;
 		BPositionIO	*fData;
 		// DrawState	*fState;
 		BList		*fPictures;