1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
|
/*
AngelCode Scripting Library
Copyright (c) 2003-2021 Andreas Jonsson
This software is provided 'as-is', without any express or implied
warranty. In no event will the authors be held liable for any
damages arising from the use of this software.
Permission is granted to anyone to use this software for any
purpose, including commercial applications, and to alter it and
redistribute it freely, subject to the following restrictions:
1. The origin of this software must not be misrepresented; you
must not claim that you wrote the original software. If you use
this software in a product, an acknowledgment in the product
documentation would be appreciated but is not required.
2. Altered source versions must be plainly marked as such, and
must not be misrepresented as being the original software.
3. This notice may not be removed or altered from any source
distribution.
The original version of this library can be located at:
http://www.angelcode.com/angelscript/
Andreas Jonsson
andreas@angelcode.com
*/
//
// as_restore.h
//
// Functions for saving and restoring module bytecode
// asCRestore was originally written by Dennis Bollyn, dennis@gyrbo.be
// It was later split in two classes asCReader and asCWriter by me
#ifndef AS_RESTORE_H
#define AS_RESTORE_H
#include "as_scriptengine.h"
#include "as_context.h"
#include "as_map.h"
BEGIN_AS_NAMESPACE
class asCReader {
public:
asCReader(asCModule *module, asIBinaryStream *stream, asCScriptEngine *engine);
int Read(bool *wasDebugInfoStripped);
protected:
asCModule *module;
asIBinaryStream *stream;
asCScriptEngine *engine;
bool noDebugInfo;
bool error;
asUINT bytesRead;
int Error(const char *msg);
int ReadInner();
int ReadData(void *data, asUINT size);
void ReadString(asCString *str);
asCScriptFunction *ReadFunction(bool &isNew, bool addToModule = true, bool addToEngine = true, bool addToGC = true, bool *isExternal = 0);
void ReadFunctionSignature(asCScriptFunction *func, asCObjectType **parentClass = 0);
void ReadGlobalProperty();
void ReadObjectProperty(asCObjectType *ot);
void ReadDataType(asCDataType *dt);
asCTypeInfo *ReadTypeInfo();
void ReadTypeDeclaration(asCTypeInfo *ot, int phase, bool *isExternal = 0);
void ReadByteCode(asCScriptFunction *func);
asWORD ReadEncodedUInt16();
asUINT ReadEncodedUInt();
int ReadEncodedInt();
asQWORD ReadEncodedUInt64();
asUINT SanityCheck(asUINT val, asUINT max);
int SanityCheck(int val, asUINT max);
void ReadUsedTypeIds();
void ReadUsedFunctions();
void ReadUsedGlobalProps();
void ReadUsedStringConstants();
void ReadUsedObjectProps();
asCTypeInfo *FindType(int idx);
int FindTypeId(int idx);
short FindObjectPropOffset(asWORD index);
asCScriptFunction *FindFunction(int idx);
// After loading, each function needs to be translated to update pointers, function ids, etc
void TranslateFunction(asCScriptFunction *func);
void CalculateAdjustmentByPos(asCScriptFunction *func);
int AdjustStackPosition(int pos);
int AdjustGetOffset(int offset, asCScriptFunction *func, asDWORD programPos);
void CalculateStackNeeded(asCScriptFunction *func);
asCScriptFunction *GetCalledFunction(asCScriptFunction *func, asDWORD programPos);
// Temporary storage for persisting variable data
asCArray<int> usedTypeIds;
asCArray<asCTypeInfo *> usedTypes;
asCArray<asCScriptFunction *> usedFunctions;
asCArray<void *> usedGlobalProperties;
asCArray<void *> usedStringConstants;
asCArray<asCScriptFunction *> savedFunctions;
asCArray<asCDataType> savedDataTypes;
asCArray<asCString> savedStrings;
asCArray<int> adjustByPos;
asCArray<int> adjustNegativeStackByPos;
struct SObjProp {
asCObjectType *objType;
asCObjectProperty *prop;
};
asCArray<SObjProp> usedObjectProperties;
asCMap<void *, bool> existingShared;
asCMap<asCScriptFunction *, bool> dontTranslate;
// Helper class for adjusting offsets within initialization list buffers
struct SListAdjuster {
SListAdjuster(asCReader *rd, asDWORD *bc, asCObjectType *ot);
void AdjustAllocMem();
int AdjustOffset(int offset);
void SetRepeatCount(asUINT rc);
void SetNextType(int typeId);
struct SInfo {
asUINT repeatCount;
asSListPatternNode *startNode;
};
asCArray<SInfo> stack;
asCReader *reader;
asDWORD *allocMemBC;
asUINT maxOffset;
asCObjectType *patternType;
asUINT repeatCount;
int lastOffset;
int nextOffset;
asUINT lastAdjustedOffset;
asSListPatternNode *patternNode;
int nextTypeId;
};
asCArray<SListAdjuster *> listAdjusters;
};
#ifndef AS_NO_COMPILER
class asCWriter {
public:
asCWriter(asCModule *module, asIBinaryStream *stream, asCScriptEngine *engine, bool stripDebugInfo);
int Write();
protected:
asCModule *module;
asIBinaryStream *stream;
asCScriptEngine *engine;
bool stripDebugInfo;
bool error;
asUINT bytesWritten;
int Error(const char *msg);
int WriteData(const void *data, asUINT size);
void WriteString(asCString *str);
void WriteFunction(asCScriptFunction *func);
void WriteFunctionSignature(asCScriptFunction *func);
void WriteGlobalProperty(asCGlobalProperty *prop);
void WriteObjectProperty(asCObjectProperty *prop);
void WriteDataType(const asCDataType *dt);
void WriteTypeInfo(asCTypeInfo *ot);
void WriteTypeDeclaration(asCTypeInfo *ot, int phase);
void WriteByteCode(asCScriptFunction *func);
void WriteEncodedInt64(asINT64 i);
// Helper functions for storing variable data
int FindTypeInfoIdx(asCTypeInfo *ti);
int FindTypeIdIdx(int typeId);
int FindFunctionIndex(asCScriptFunction *func);
int FindGlobalPropPtrIndex(void *);
int FindStringConstantIndex(void *str);
int FindObjectPropIndex(short offset, int typeId, asDWORD *bc);
void CalculateAdjustmentByPos(asCScriptFunction *func);
int AdjustStackPosition(int pos);
int AdjustProgramPosition(int pos);
int AdjustGetOffset(int offset, asCScriptFunction *func, asDWORD programPos);
// Intermediate data used for storing that which isn't constant, function id's, pointers, etc
void WriteUsedTypeIds();
void WriteUsedFunctions();
void WriteUsedGlobalProps();
void WriteUsedStringConstants();
void WriteUsedObjectProps();
// Temporary storage for persisting variable data
asCArray<int> usedTypeIds;
asCArray<asCTypeInfo *> usedTypes;
asCArray<asCScriptFunction *> usedFunctions;
asCArray<void *> usedGlobalProperties;
asCArray<void *> usedStringConstants;
asCMap<void *, int> stringToIndexMap;
asCArray<asCScriptFunction *> savedFunctions;
asCArray<asCDataType> savedDataTypes;
asCArray<asCString> savedStrings;
asCMap<asCString, int> stringToIdMap;
asCArray<int> adjustStackByPos;
asCArray<int> adjustNegativeStackByPos;
asCArray<int> bytecodeNbrByPos;
struct SObjProp {
asCObjectType *objType;
asCObjectProperty *prop;
};
asCArray<SObjProp> usedObjectProperties;
// Helper class for adjusting offsets within initialization list buffers
struct SListAdjuster {
SListAdjuster(asCObjectType *ot);
int AdjustOffset(int offset, asCObjectType *listPatternType);
void SetRepeatCount(asUINT rc);
void SetNextType(int typeId);
struct SInfo {
asUINT repeatCount;
asSListPatternNode *startNode;
};
asCArray<SInfo> stack;
asCObjectType *patternType;
asUINT repeatCount;
asSListPatternNode *patternNode;
asUINT entries;
int lastOffset; // Last offset adjusted
int nextOffset; // next expected offset to be adjusted
int nextTypeId;
};
asCArray<SListAdjuster *> listAdjusters;
};
#endif
END_AS_NAMESPACE
#endif // AS_RESTORE_H
|
