dbb1f8cf37
671fc805 update test/cybozu 8ca86231 remove mutable in Address 8b93498f add cmpsb/scasb/... 7eb62750 avoid core_sharing_data_cache = 0 for some cloud envrionment 85767e95 support mingw64 59573e6e add PROTECT_RE mode for protect() 71b75f65 fix push(qword[mem]) 811f4959 Merge branch 'rsdubtso-master' 8e3cb711 Account for potentially zero 0xb leaf when parsing cache/topology via cpuid a816249f update version fe083912 fix to avoid zero division for some virtual machine f0a8f7fa update version cac09b7a Merge pull request #62 from mgouicem/master 1f96b5e0 Fixes an error raised by clang < 3.9 c0f885ac Merge pull request #61 from mgouicem/master bfe2d201 Change default value for n_cores in setCacheHierarchy. fd587b55 change format and add getter for data_cache_size 80b3c7b9 remove macro 88189609 Merge branch 'mgouicem-master' e6b79723 Adding queries to get the cpu topology on Intel architectures. 221384f0 vmov* supports [mem]|k|z c04141ef define XBYAK_NO_OP_NAMES for test af7f05ee add const for Label git-subtree-dir: externals/xbyak git-subtree-split: 671fc805d09d075f48d4625f183ef2e1ef725106
189 lines
3.9 KiB
C++
189 lines
3.9 KiB
C++
#if defined(_MSC_VER) && (_MSC_VER <= 1200)
|
|
#pragma warning(disable:4514)
|
|
#pragma warning(disable:4786)
|
|
#endif
|
|
#if defined(_MSC_VER) && (_MSC_VER >= 1900)
|
|
#pragma warning(disable:4456)
|
|
#endif
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#define XBYAK_NO_OP_NAMES
|
|
#include "xbyak/xbyak.h"
|
|
|
|
class Sample : public Xbyak::CodeGenerator {
|
|
void operator=(const Sample&);
|
|
public:
|
|
Sample(void *userPtr = 0, size_t size = Xbyak::DEFAULT_MAX_CODE_SIZE) : Xbyak::CodeGenerator(size, userPtr)
|
|
{
|
|
inLocalLabel(); // use local label for multiple instance
|
|
#ifdef XBYAK32
|
|
mov(ecx, ptr [esp + 4]); // n
|
|
#elif defined(XBYAK64_GCC)
|
|
mov(ecx, edi); // n
|
|
#else
|
|
// n = ecx
|
|
#endif
|
|
xor_(eax, eax); // sum
|
|
test(ecx, ecx);
|
|
jz(".exit");
|
|
xor_(edx, edx); // i
|
|
L(".lp");
|
|
add(eax, edx);
|
|
inc(edx);
|
|
|
|
cmp(edx, ecx);
|
|
jbe(".lp"); // jmp to previous @@
|
|
L(".exit"); // <B>
|
|
ret();
|
|
outLocalLabel(); // end of local label
|
|
}
|
|
};
|
|
|
|
class AddFunc : public Xbyak::CodeGenerator {
|
|
void operator=(const AddFunc&);
|
|
public:
|
|
AddFunc(int y)
|
|
{
|
|
#ifdef XBYAK32
|
|
mov(eax, ptr [esp + 4]);
|
|
add(eax, y);
|
|
#elif defined(XBYAK64_WIN)
|
|
lea(rax, ptr [rcx + y]);
|
|
#else
|
|
lea(eax, ptr [edi + y]);
|
|
#endif
|
|
ret();
|
|
}
|
|
int (*get() const)(int) { return getCode<int(*)(int)>(); }
|
|
};
|
|
|
|
class CallAtoi : public Xbyak::CodeGenerator {
|
|
void operator=(const CallAtoi&);
|
|
public:
|
|
CallAtoi()
|
|
{
|
|
#ifdef XBYAK64
|
|
#ifdef XBYAK64_WIN
|
|
sub(rsp, 32); // return-address is destroied if 64bit debug mode
|
|
#endif
|
|
mov(rax, (size_t)atoi);
|
|
call(rax);
|
|
#ifdef XBYAK64_WIN
|
|
add(rsp, 32);
|
|
#endif
|
|
#else
|
|
mov(eax, ptr [esp + 4]);
|
|
push(eax);
|
|
#ifdef XBYAK_VARIADIC_TEMPLATE
|
|
call(atoi);
|
|
#else
|
|
call(Xbyak::CastTo<void*>(atoi));
|
|
#endif
|
|
add(esp, 4);
|
|
#endif
|
|
ret();
|
|
}
|
|
int (*get() const)(const char *) { return getCode<int (*)(const char *)>(); }
|
|
};
|
|
|
|
class JmpAtoi : public Xbyak::CodeGenerator {
|
|
void operator=(const JmpAtoi&);
|
|
public:
|
|
JmpAtoi()
|
|
{
|
|
/* already pushed "456" */
|
|
#ifdef XBYAK64
|
|
mov(rax, (size_t)atoi);
|
|
jmp(rax);
|
|
#else
|
|
jmp(Xbyak::CastTo<void*>(atoi));
|
|
#endif
|
|
}
|
|
int (*get() const)(const char *) { return getCode<int (*)(const char *)>(); }
|
|
};
|
|
|
|
struct Reset : public Xbyak::CodeGenerator {
|
|
void init(int n)
|
|
{
|
|
xor_(eax, eax);
|
|
mov(ecx, n);
|
|
test(ecx, ecx);
|
|
jnz("@f");
|
|
ret();
|
|
L("@@");
|
|
for (int i = 0; i < 10 - n; i++) {
|
|
add(eax, ecx);
|
|
}
|
|
sub(ecx, 1);
|
|
jnz("@b");
|
|
ret();
|
|
}
|
|
};
|
|
|
|
void testReset()
|
|
{
|
|
puts("testReset");
|
|
Reset code;
|
|
int (*f)(int) = code.getCode<int(*)(int)>();
|
|
for (int i = 0; i < 10; i++) {
|
|
code.init(i);
|
|
int v = f(i);
|
|
printf("%d %d\n", i, v);
|
|
code.reset();
|
|
}
|
|
}
|
|
|
|
int main()
|
|
{
|
|
try {
|
|
Sample s;
|
|
printf("Xbyak version=%s\n", s.getVersionString());
|
|
#ifdef XBYAK64_GCC
|
|
puts("64bit mode(gcc)");
|
|
#elif defined(XBYAK64_WIN)
|
|
puts("64bit mode(win)");
|
|
#else
|
|
puts("32bit");
|
|
#endif
|
|
int (*func)(int) = s.getCode<int (*)(int)>();
|
|
for (int i = 0; i <= 10; i++) {
|
|
printf("0 + ... + %d = %d\n", i, func(i));
|
|
}
|
|
for (int i = 0; i < 10; i++) {
|
|
AddFunc a(i);
|
|
int (*add)(int) = a.get();
|
|
int y = add(i);
|
|
printf("%d + %d = %d\n", i, i, y);
|
|
}
|
|
CallAtoi c;
|
|
printf("call atoi(\"123\") = %d\n", c.get()("123"));
|
|
JmpAtoi j;
|
|
printf("jmp atoi(\"456\") = %d\n", j.get()("456"));
|
|
{
|
|
// use memory allocated by user
|
|
using namespace Xbyak;
|
|
const size_t codeSize = 4096;
|
|
uint8 buf[codeSize + 16];
|
|
uint8 *p = CodeArray::getAlignedAddress(buf);
|
|
Sample s(p, codeSize);
|
|
if (!CodeArray::protect(p, codeSize, CodeArray::PROTECT_RWE)) {
|
|
fprintf(stderr, "can't protect\n");
|
|
return 1;
|
|
}
|
|
int (*func)(int) = s.getCode<int (*)(int)>();
|
|
if (Xbyak::CastTo<uint8*>(func) != p) {
|
|
fprintf(stderr, "internal error %p %p\n", p, Xbyak::CastTo<uint8*>(func));
|
|
return 1;
|
|
}
|
|
printf("0 + ... + %d = %d\n", 100, func(100));
|
|
CodeArray::protect(p, codeSize, CodeArray::PROTECT_RW);
|
|
}
|
|
puts("OK");
|
|
testReset();
|
|
} catch (std::exception& e) {
|
|
printf("ERR:%s\n", e.what());
|
|
} catch (...) {
|
|
printf("unknown error\n");
|
|
}
|
|
}
|
|
|