haiku/src/kits/support/Archivable.cpp
ejakowatz ff1ade6b3d More tests and fixes for bugs exposed by them. Also removed spurious
instantiation of BRoster from BArchivable, line 323, as per Tyler's
mention. =)


git-svn-id: file:///srv/svn/repos/haiku/trunk/current@852 a95241bf-73f2-0310-859d-f6bbb57e9c96
2002-08-22 03:43:06 +00:00

571 lines
15 KiB
C++

//------------------------------------------------------------------------------
// Copyright (c) 2001-2002, OpenBeOS
//
// Permission is hereby granted, free of charge, to any person obtaining a
// copy of this software and associated documentation files (the "Software"),
// to deal in the Software without restriction, including without limitation
// the rights to use, copy, modify, merge, publish, distribute, sublicense,
// and/or sell copies of the Software, and to permit persons to whom the
// Software is furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
// DEALINGS IN THE SOFTWARE.
//
// File Name: Archivable.cpp
// Author: Erik Jaesler (erik@cgsoftware.com)
// Description: BArchivable mix-in class defines the archiving
// protocol. Also some global archiving functions.
//------------------------------------------------------------------------------
// Standard Includes -----------------------------------------------------------
#include <ctype.h>
#include <errno.h>
#include <stdlib.h>
#include <stdio.h>
#include <string>
#include <typeinfo>
#include <vector>
// System Includes -------------------------------------------------------------
#include <AppFileInfo.h>
#include <Archivable.h>
#include <Entry.h>
#include <List.h>
#include <OS.h>
#include <Path.h>
#include <Roster.h>
#include <String.h>
#include <SupportDefs.h>
#include <syslog.h>
// Project Includes ------------------------------------------------------------
// Local Includes --------------------------------------------------------------
// Local Defines ---------------------------------------------------------------
// Globals ---------------------------------------------------------------------
using std::string;
using std::vector;
const char* B_CLASS_FIELD = "class";
const char* B_ADD_ON_FIELD = "add_on";
const int32 FUNC_NAME_LEN = 1024;
// TODO: consider moving these
// to a separate module, and making them more full-featured (e.g., taking
// NS::ClassName::Function(Param p) instead of just NS::ClassName)
static void Demangle(const char *name, BString &out);
static void Mangle(const char *name, BString &out);
static instantiation_func FindFuncInImage(BString& funcName, image_id id,
status_t& err);
static bool CheckSig(const char* sig, image_info& info);
/*
// TODO: Where do these get triggered from?
Log entries graciously coughed up by the Be implementation:
Nov 28 01:40:45 instantiate_object failed: NULL BMessage argument
Nov 28 01:40:45 instantiate_object failed: Failed to find an entrydefining the class name (Name not found).
Nov 28 01:40:45 instantiate_object failed: No signature specified in archive, looking for class "TInvalidClassName".
Nov 28 01:40:45 instantiate_object failed: Error finding app with signature "application/x-vnd.InvalidSignature" (Application could not be found)
Nov 28 01:40:45 instantiate_object failed: Application could not be found (8000200b)
Nov 28 01:40:45 instantiate_object failed: Failed to find exported Instantiate static function for class TInvalidClassName.
Nov 28 01:40:45 instantiate_object failed: Invalid argument (80000005)
Nov 28 01:40:45 instantiate_object failed: No signature specified in archive, looking for class "TRemoteTestObject".
Nov 28 01:40:45 instantiate_object - couldn't get mime sig for /boot/home/src/projects/OpenBeOS/app_kit/test/lib/support/BArchivable/./BArchivableSystemTester
Nov 28 01:40:45 instantiate_object failed: Error finding app with signature "application/x-vnd.InvalidSignature" (Application could not be found)
Nov 28 01:40:45 instantiate_object failed: Application could not be found (8000200b)
Nov 28 01:40:45 instantiate_object failed: Error finding app with signature "application/x-vnd.InvalidSignature" (Application could not be found)
Nov 28 01:40:45 instantiate_object failed: Application could not be found (8000200b)
Nov 28 01:40:45 instantiate_object failed: Error finding app with signature "application/x-vnd.InvalidSignature" (Application could not be found)
Nov 28 01:40:45 instantiate_object failed: Application could not be found (8000200b)
*/
//------------------------------------------------------------------------------
BArchivable::BArchivable()
{
;
}
//------------------------------------------------------------------------------
BArchivable::BArchivable(BMessage* from)
{
;
}
//------------------------------------------------------------------------------
BArchivable::~BArchivable()
{
;
}
//------------------------------------------------------------------------------
status_t BArchivable::Archive(BMessage* into, bool deep) const
{
if (!into)
{
// TODO: logging/other error reporting?
return B_BAD_VALUE;
}
BString name;
Demangle(typeid(*this).name(), name);
return into->AddString(B_CLASS_FIELD, name);
}
//------------------------------------------------------------------------------
BArchivable* BArchivable::Instantiate(BMessage* from)
{
debugger("Can't create a plain BArchivable object");
return NULL;
}
//------------------------------------------------------------------------------
status_t BArchivable::Perform(perform_code d, void* arg)
{
// TODO: Check against original
return B_ERROR;
}
//------------------------------------------------------------------------------
void BArchivable::_ReservedArchivable1()
{
;
}
//------------------------------------------------------------------------------
void BArchivable::_ReservedArchivable2()
{
;
}
//------------------------------------------------------------------------------
void BArchivable::_ReservedArchivable3()
{
;
}
//------------------------------------------------------------------------------
void BuildFuncName(const char* className, BString& funcName)
{
funcName = "";
// This is what we're after:
// Instantiate__Q28OpenBeOS11BArchivableP8BMessage
Mangle(className, funcName);
funcName.Prepend("Instantiate__");
funcName.Append("P8BMessage");
}
//------------------------------------------------------------------------------
BArchivable* instantiate_object(BMessage* archive, image_id* id)
{
errno = B_OK;
// Check our params
if (id)
{
*id = B_BAD_VALUE;
}
if (!archive)
{
// TODO: extended error handling
errno = B_BAD_VALUE;
syslog(LOG_ERR, "instantiate_object failed: NULL BMessage argument");
return NULL;
}
// Get class name from archive
const char* name = NULL;
status_t err = archive->FindString(B_CLASS_FIELD, &name);
if (err)
{
// TODO: extended error handling
syslog(LOG_ERR, "instantiate_object failed: Failed to find an entry "
"defining the class name (%s).", strerror(err));
return NULL;
}
// Get sig from archive
const char* sig = NULL;
bool hasSig = (archive->FindString(B_ADD_ON_FIELD, &sig) == B_OK);
instantiation_func iFunc = find_instantiation_func(name, sig);
// if find_instantiation_func() can't locate Class::Instantiate()
// and a signature was specified
if (!iFunc && hasSig)
{
// use BRoster::FindApp() to locate an app or add-on with the symbol
BRoster Roster;
entry_ref ref;
err = Roster.FindApp(sig, &ref);
BEntry entry;
// if an entry_ref is obtained
if (!err)
{
err = entry.SetTo(&ref);
}
if (err)
{
syslog(LOG_ERR, "instantiate_object failed: Error finding app "
"with signature \"%s\" (%s)", sig, strerror(err));
}
if (!err)
{
BPath path;
err = entry.GetPath(&path);
if (!err)
{
// load the app/add-on
image_id theImage = load_add_on(path.Path());
if (theImage < 0)
{
// TODO: extended error handling
return NULL;
}
// Save the image_id
if (id)
{
*id = theImage;
}
BString funcName;
BuildFuncName(name, funcName);
iFunc = FindFuncInImage(funcName, theImage, err);
if (!iFunc)
{
syslog(LOG_ERR, "instantiate_object failed: Failed to find exported "
"Instantiate static function for class %s.", name);
}
}
}
}
else if (!iFunc)
{
syslog(LOG_ERR, "instantiate_object failed: No signature specified "
"in archive, looking for class \"%s\".", name);
errno = B_BAD_VALUE;
}
if (err)
{
// TODO: extended error handling
syslog(LOG_ERR, "instantiate_object failed: %s (%x)",
strerror(err), err);
errno = err;
return NULL;
}
// if Class::Instantiate(BMessage*) was found
if (iFunc)
{
// use to create and return an object instance
return iFunc(archive);
}
return NULL;
}
//------------------------------------------------------------------------------
BArchivable* instantiate_object(BMessage* from)
{
return instantiate_object(from, NULL);
}
//------------------------------------------------------------------------------
bool validate_instantiation(BMessage* from, const char* class_name)
{
errno = B_OK;
// Make sure our params are kosher -- original skimped here =P
if (!from)
{
// Not standard; Be implementation has a segment
// violation on this error mode
errno = B_BAD_VALUE;
return false;
}
status_t err = B_OK;
const char* data;
for (int32 index = 0; err == B_OK; ++index)
{
err = from->FindString(B_CLASS_FIELD, index, &data);
if (!err && strcmp(data, class_name) == 0)
{
return true;
}
}
errno = B_MISMATCHED_VALUES;
syslog(LOG_ERR, "validate_instantiation failed on class %s.", class_name);
return false;
}
//------------------------------------------------------------------------------
instantiation_func find_instantiation_func(const char* class_name,
const char* sig)
{
errno = B_OK;
if (!class_name)
{
errno = B_BAD_VALUE;
return NULL;
}
instantiation_func theFunc = NULL;
BString funcName;
BuildFuncName(class_name, funcName);
thread_id tid = find_thread(NULL);
thread_info ti;
status_t err = get_thread_info(tid, &ti);
if (!err)
{
// for each image_id in team_id
image_info info;
int32 cookie = 0;
while (!theFunc && (get_next_image_info(ti.team, &cookie, &info) == B_OK))
{
theFunc = FindFuncInImage(funcName, info.id, err);
}
if (theFunc && !CheckSig(sig, info))
{
// TODO: extended error handling
theFunc = NULL;
}
}
return theFunc;
}
//------------------------------------------------------------------------------
instantiation_func find_instantiation_func(const char* class_name)
{
return find_instantiation_func(class_name, NULL);
}
//------------------------------------------------------------------------------
instantiation_func find_instantiation_func(BMessage* archive_data)
{
errno = B_OK;
if (!archive_data)
{
// TODO: extended error handling
errno = B_BAD_VALUE;
return NULL;
}
const char* name = NULL;
const char* sig = NULL;
status_t err;
err = archive_data->FindString(B_CLASS_FIELD, &name);
if (err)
{
// TODO: extended error handling
return NULL;
}
err = archive_data->FindString(B_ADD_ON_FIELD, &sig);
return find_instantiation_func(name, sig);
}
//------------------------------------------------------------------------------
//------------------------------------------------------------------------------
int GetNumber(const char*& name)
{
int val = atoi(name);
while (isdigit(*name))
{
++name;
}
return val;
}
//------------------------------------------------------------------------------
void Demangle(const char* name, BString& out)
{
// TODO: add support for template classes
// _find__t12basic_string3ZcZt18string_char_traits1ZcZt24__default_alloc_template2b0i0PCccUlUl
out = "";
// Are we in a namespace?
if (*name == 'Q')
{
// Yessir, we are; how many deep are we?
int nsCount = 0;
++name;
if (*name == '_') // more than 10 deep
{
++name;
if (!isdigit(*name))
; // TODO: error handling
nsCount = GetNumber(name);
if (*name == '_') // more than 10 deep
++name;
else
; // this should be an error condition
}
else
{
nsCount = *name - '0';
++name;
}
int nameLen = 0;
for (int i = 0; i < nsCount - 1; ++i)
{
if (!isdigit(*name))
; // TODO: error handling
nameLen = GetNumber(name);
out.Append(name, nameLen);
out += "::";
name += nameLen;
}
}
out.Append(name, GetNumber(name));
}
//------------------------------------------------------------------------------
void Mangle(const char* name, BString& out)
{
// TODO: add support for template classes
// _find__t12basic_string3ZcZt18string_char_traits1ZcZt24__default_alloc_template2b0i0PCccUlUl
// Chop this:
// testthree::testfour::Testthree::Testfour
// up into little bite-sized pieces
int count = 0;
string origName(name);
vector<string> spacenames;
string::size_type pos = 0;
string::size_type oldpos = 0;
while (pos != string::npos)
{
pos = origName.find_first_of("::", oldpos);
spacenames.push_back(string(origName, oldpos, pos - oldpos));
pos = origName.find_first_not_of("::", pos);
oldpos = pos;
++count;
}
// Now mangle it into this:
// Q49testthree8testfour9Testthree8Testfour
out = "";
if (count > 1)
{
out += 'Q';
if (count > 10)
out += '_';
out << count;
if (count > 10)
out += '_';
}
for (unsigned int i = 0; i < spacenames.size(); ++i)
{
out << (int)spacenames[i].length();
out += spacenames[i].c_str();
}
}
//------------------------------------------------------------------------------
instantiation_func FindFuncInImage(BString& funcName, image_id id,
status_t& err)
{
err = B_OK;
instantiation_func theFunc = NULL;
// Don't need to do it this way ...
#if 0
char foundFuncName[FUNC_NAME_LEN];
int32 symbolType;
int32 funcNameLen;
// for each B_SYMBOL_TYPE_TEXT in image_id
for (int32 i = 0; !err; ++i)
{
funcNameLen = FUNC_NAME_LEN;
err = get_nth_image_symbol(id, i, foundFuncName, &funcNameLen,
&symbolType, (void**)&theFunc);
if (!err && symbolType == B_SYMBOL_TYPE_TEXT)
{
// try to match Class::Instantiate(BMessage*) signature
if (funcName.ICompare(foundFuncName, funcNameLen) == 0)
break;
else
theFunc = NULL;
}
}
#endif
err = get_image_symbol(id, funcName.String(), B_SYMBOL_TYPE_TEXT,
(void**)&theFunc);
if (err)
{
// TODO: error handling
theFunc = NULL;
}
return theFunc;
}
//------------------------------------------------------------------------------
bool CheckSig(const char* sig, image_info& info)
{
if (!sig)
{
// If it wasn't specified, anything "matches"
return true;
}
status_t err = B_OK;
// Get image signature
BFile ImageFile(info.name, B_READ_ONLY);
err = ImageFile.InitCheck();
if (err)
{
// TODO: extended error handling
return false;
}
char imageSig[B_MIME_TYPE_LENGTH];
BAppFileInfo AFI(&ImageFile);
err = AFI.GetSignature(imageSig);
if (err)
{
// TODO: extended error handling
syslog(LOG_ERR, "instantiate_object - couldn't get mime sig for %s",
info.name);
return false;
}
return strcmp(sig, imageSig) == 0;
}
//------------------------------------------------------------------------------
/*
* $Log $
*
* $Id $
*
*/