asBSTR asBStrFactory(asUINT length, const char *s)
{
asBSTR o = asBStrAlloc(length);
memcpy(o, s, length);
return o;
}
asBSTR *asBStrCopy(const asBSTR *src, asBSTR *dst)
{
// Free the destination bstr
asBStrFree(*dst);
// Make a copy of the source bstr
int len = asBStrLength(*src);
*dst = asBStrAlloc(len);
memcpy(*dst, *src, len);
return dst;
}
asBSTR asBStrFormatBits(asDWORD bits)
{
char str[50];
sprintf(str, "%X", (unsigned int)bits);
int len = (int)strlen(str);
// We must allocate a new bstr that the script engine will free afterwards
asBSTR bstr = asBStrAlloc(len);
memcpy(bstr, str, len);
return bstr;
}
asBSTR asBStrFormat(double number)
{
char str[50];
sprintf(str, "%#6g", number);
int len = (int)strlen(str);
// We must allocate a new bstr that the script engine will free afterwards
asBSTR bstr = asBStrAlloc(len);
memcpy(bstr, str, len);
return bstr;
}
asBSTR asBStrConcatenate(const asBSTR *left, const asBSTR *right)
{
int ll = asBStrLength(*left);
int lr = asBStrLength(*right);
// Allocate a new string for the concatenation
asBSTR str = asBStrAlloc(ll + lr);
memcpy(str, *left, ll);
memcpy(str+ll, *right, lr);
return str;
}
asBSTR asBStrSubstr(const asBSTR &str, asUINT start, asUINT count)
{
asUINT len = asBStrLength(str);
len -= start;
if( len < count ) count = len;
// We must allocate a new bstr that the script engine will free afterwards
asBSTR sub = asBStrAlloc(count);
memcpy(sub, str+start, count);
return sub;
}
bool TestBStr()
{
if( strstr(asGetLibraryOptions(), "AS_MAX_PORTABILITY") )
{
printf("%s: Test skipped due to AS_MAX_PORTABILITY\n", TESTNAME);
return false;
}
bool ret = false;
asIScriptEngine *engine = asCreateScriptEngine(ANGELSCRIPT_VERSION);
RegisterBStr(engine);
engine->RegisterGlobalFunction("bstr NewString(int)", asFUNCTION(NewString), asCALL_CDECL);
engine->RegisterGlobalFunction("void assert(bool)", asFUNCTION(Assert), asCALL_GENERIC);
int r = engine->ExecuteString(0, "bstr s = NewString(10)");
if( r < 0 )
{
printf("%s: ExecuteString() failed\n", TESTNAME);
ret = true;
}
else if( r != asEXECUTION_FINISHED )
{
printf("%s: ExecuteString() returned %d\n", TESTNAME, r);
ret = true;
}
// Test passing bstr strings to a script function
asIScriptModule *mod = engine->GetModule(0, asGM_ALWAYS_CREATE);
mod->AddScriptSection("script", script);
r = mod->Build();
if( r < 0 )
{
ret = true;
}
int funcId = mod->GetFunctionIdByIndex(0);
asIScriptContext *ctx = engine->CreateContext();
ctx->Prepare(funcId);
// Create the object and initialize it, then give
// the pointer directly to the script engine.
// The script engine will free the object.
asBSTR *a = (asBSTR*)engine->CreateScriptObject(engine->GetTypeIdByDecl("bstr"));
*a = asBStrAlloc(1);
strcpy((char*)*a, "a");
*(asBSTR**)ctx->GetArgPointer(0) = a;
// Create a local instance and have the script engine copy it.
// The application must free its copy of the object.
asBSTR b = asBStrAlloc(1);
strcpy((char*)b, "b");
ctx->SetArgObject(1, &b);
asBStrFree(b);
r = ctx->Execute();
if( r != asEXECUTION_FINISHED )
ret = true;
if( ctx ) ctx->Release();
engine->Release();
return ret;
}
static asBSTR NewString(int length)
{
return asBStrAlloc(length);
}