mirror of
https://github.com/luau-lang/luau.git
synced 2024-11-15 14:25:44 +08:00
Sync to upstream/release/542 (#649)
- Fix DeprecatedGlobal warning text in cases when the global is deprecated without a suggested alternative - Fix an off-by-one error in type error text for incorrect use of string.format - Reduce stack consumption further during parsing, hopefully eliminating stack overflows during parsing/compilation for good - Mark interpolated string support as experimental (requires --fflags=LuauInterpolatedStringBaseSupport to enable) - Simplify garbage collection treatment of upvalues, reducing cache misses during sweeping stage and reducing the cost of upvalue assignment (SETUPVAL); supersedes #643 - Simplify garbage collection treatment of sleeping threads - Simplify sweeping of alive threads, reducing cache misses during sweeping stage - Simplify management of string buffers, removing redundant linked list operations
This commit is contained in:
parent
da9d8e8c60
commit
b2f9f53ae3
@ -35,7 +35,7 @@ struct PackSubtypeConstraint
|
|||||||
TypePackId superPack;
|
TypePackId superPack;
|
||||||
};
|
};
|
||||||
|
|
||||||
// subType ~ gen superType
|
// generalizedType ~ gen sourceType
|
||||||
struct GeneralizationConstraint
|
struct GeneralizationConstraint
|
||||||
{
|
{
|
||||||
TypeId generalizedType;
|
TypeId generalizedType;
|
||||||
|
@ -100,6 +100,8 @@ struct ConstraintSolver
|
|||||||
void unblock(NotNull<const Constraint> progressed);
|
void unblock(NotNull<const Constraint> progressed);
|
||||||
void unblock(TypeId progressed);
|
void unblock(TypeId progressed);
|
||||||
void unblock(TypePackId progressed);
|
void unblock(TypePackId progressed);
|
||||||
|
void unblock(const std::vector<TypeId>& types);
|
||||||
|
void unblock(const std::vector<TypePackId>& packs);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @returns true if the TypeId is in a blocked state.
|
* @returns true if the TypeId is in a blocked state.
|
||||||
|
@ -16,7 +16,7 @@ struct ConstraintSolverLogger
|
|||||||
{
|
{
|
||||||
std::string compileOutput();
|
std::string compileOutput();
|
||||||
void captureBoundarySnapshot(const Scope* rootScope, std::vector<NotNull<const Constraint>>& unsolvedConstraints);
|
void captureBoundarySnapshot(const Scope* rootScope, std::vector<NotNull<const Constraint>>& unsolvedConstraints);
|
||||||
void prepareStepSnapshot(const Scope* rootScope, NotNull<const Constraint> current, std::vector<NotNull<const Constraint>>& unsolvedConstraints);
|
void prepareStepSnapshot(const Scope* rootScope, NotNull<const Constraint> current, std::vector<NotNull<const Constraint>>& unsolvedConstraints, bool force);
|
||||||
void commitPreparedStepSnapshot();
|
void commitPreparedStepSnapshot();
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
@ -33,7 +33,8 @@ struct ToStringOptions
|
|||||||
bool indent = false;
|
bool indent = false;
|
||||||
size_t maxTableLength = size_t(FInt::LuauTableTypeMaximumStringifierLength); // Only applied to TableTypeVars
|
size_t maxTableLength = size_t(FInt::LuauTableTypeMaximumStringifierLength); // Only applied to TableTypeVars
|
||||||
size_t maxTypeLength = size_t(FInt::LuauTypeMaximumStringifierLength);
|
size_t maxTypeLength = size_t(FInt::LuauTypeMaximumStringifierLength);
|
||||||
std::optional<ToStringNameMap> nameMap;
|
ToStringNameMap nameMap;
|
||||||
|
std::optional<ToStringNameMap> DEPRECATED_nameMap;
|
||||||
std::shared_ptr<Scope> scope; // If present, module names will be added and types that are not available in scope will be marked as 'invalid'
|
std::shared_ptr<Scope> scope; // If present, module names will be added and types that are not available in scope will be marked as 'invalid'
|
||||||
std::vector<std::string> namedFunctionOverrideArgNames; // If present, named function argument names will be overridden
|
std::vector<std::string> namedFunctionOverrideArgNames; // If present, named function argument names will be overridden
|
||||||
};
|
};
|
||||||
@ -41,7 +42,7 @@ struct ToStringOptions
|
|||||||
struct ToStringResult
|
struct ToStringResult
|
||||||
{
|
{
|
||||||
std::string name;
|
std::string name;
|
||||||
ToStringNameMap nameMap;
|
ToStringNameMap DEPRECATED_nameMap;
|
||||||
|
|
||||||
bool invalid = false;
|
bool invalid = false;
|
||||||
bool error = false;
|
bool error = false;
|
||||||
@ -49,12 +50,24 @@ struct ToStringResult
|
|||||||
bool truncated = false;
|
bool truncated = false;
|
||||||
};
|
};
|
||||||
|
|
||||||
ToStringResult toStringDetailed(TypeId ty, const ToStringOptions& opts = {});
|
ToStringResult toStringDetailed(TypeId ty, ToStringOptions& opts);
|
||||||
ToStringResult toStringDetailed(TypePackId ty, const ToStringOptions& opts = {});
|
ToStringResult toStringDetailed(TypePackId ty, ToStringOptions& opts);
|
||||||
|
|
||||||
std::string toString(TypeId ty, const ToStringOptions& opts);
|
std::string toString(TypeId ty, ToStringOptions& opts);
|
||||||
std::string toString(TypePackId ty, const ToStringOptions& opts);
|
std::string toString(TypePackId ty, ToStringOptions& opts);
|
||||||
std::string toString(const Constraint& c, ToStringOptions& opts);
|
|
||||||
|
// These overloads are selected when a temporary ToStringOptions is passed. (eg
|
||||||
|
// via an initializer list)
|
||||||
|
inline std::string toString(TypePackId ty, ToStringOptions&& opts)
|
||||||
|
{
|
||||||
|
// Delegate to the overload (TypePackId, ToStringOptions&)
|
||||||
|
return toString(ty, opts);
|
||||||
|
}
|
||||||
|
inline std::string toString(TypeId ty, ToStringOptions&& opts)
|
||||||
|
{
|
||||||
|
// Delegate to the overload (TypeId, ToStringOptions&)
|
||||||
|
return toString(ty, opts);
|
||||||
|
}
|
||||||
|
|
||||||
// These are offered as overloads rather than a default parameter so that they can be easily invoked from within the MSVC debugger.
|
// These are offered as overloads rather than a default parameter so that they can be easily invoked from within the MSVC debugger.
|
||||||
// You can use them in watch expressions!
|
// You can use them in watch expressions!
|
||||||
@ -66,16 +79,42 @@ inline std::string toString(TypePackId ty)
|
|||||||
{
|
{
|
||||||
return toString(ty, ToStringOptions{});
|
return toString(ty, ToStringOptions{});
|
||||||
}
|
}
|
||||||
inline std::string toString(const Constraint& c)
|
|
||||||
|
std::string toString(const Constraint& c, ToStringOptions& opts);
|
||||||
|
|
||||||
|
inline std::string toString(const Constraint& c, ToStringOptions&& opts)
|
||||||
{
|
{
|
||||||
ToStringOptions opts;
|
|
||||||
return toString(c, opts);
|
return toString(c, opts);
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string toString(const TypeVar& tv, const ToStringOptions& opts = {});
|
inline std::string toString(const Constraint& c)
|
||||||
std::string toString(const TypePackVar& tp, const ToStringOptions& opts = {});
|
{
|
||||||
|
return toString(c, ToStringOptions{});
|
||||||
|
}
|
||||||
|
|
||||||
std::string toStringNamedFunction(const std::string& funcName, const FunctionTypeVar& ftv, const ToStringOptions& opts = {});
|
|
||||||
|
std::string toString(const TypeVar& tv, ToStringOptions& opts);
|
||||||
|
std::string toString(const TypePackVar& tp, ToStringOptions& opts);
|
||||||
|
|
||||||
|
inline std::string toString(const TypeVar& tv)
|
||||||
|
{
|
||||||
|
ToStringOptions opts;
|
||||||
|
return toString(tv, opts);
|
||||||
|
}
|
||||||
|
|
||||||
|
inline std::string toString(const TypePackVar& tp)
|
||||||
|
{
|
||||||
|
ToStringOptions opts;
|
||||||
|
return toString(tp, opts);
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string toStringNamedFunction(const std::string& funcName, const FunctionTypeVar& ftv, ToStringOptions& opts);
|
||||||
|
|
||||||
|
inline std::string toStringNamedFunction(const std::string& funcName, const FunctionTypeVar& ftv)
|
||||||
|
{
|
||||||
|
ToStringOptions opts;
|
||||||
|
return toStringNamedFunction(funcName, ftv, opts);
|
||||||
|
}
|
||||||
|
|
||||||
// It could be useful to see the text representation of a type during a debugging session instead of exploring the content of the class
|
// It could be useful to see the text representation of a type during a debugging session instead of exploring the content of the class
|
||||||
// These functions will dump the type to stdout and can be evaluated in Watch/Immediate windows or as gdb/lldb expression
|
// These functions will dump the type to stdout and can be evaluated in Watch/Immediate windows or as gdb/lldb expression
|
||||||
|
@ -263,6 +263,8 @@ struct TxnLog
|
|||||||
return Luau::get_if<T>(&ty->ty) != nullptr;
|
return Luau::get_if<T>(&ty->ty) != nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
std::pair<std::vector<TypeId>, std::vector<TypePackId>> getChanges() const;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
// unique_ptr is used to give us stable pointers across insertions into the
|
// unique_ptr is used to give us stable pointers across insertions into the
|
||||||
// map. Otherwise, it would be really easy to accidentally invalidate the
|
// map. Otherwise, it would be really easy to accidentally invalidate the
|
||||||
|
@ -210,7 +210,8 @@ void ConstraintGraphBuilder::visit(const ScopePtr& scope, AstStatLocal* local)
|
|||||||
|
|
||||||
for (size_t i = 0; i < local->values.size; ++i)
|
for (size_t i = 0; i < local->values.size; ++i)
|
||||||
{
|
{
|
||||||
if (local->values.data[i]->is<AstExprConstantNil>())
|
AstExpr* value = local->values.data[i];
|
||||||
|
if (value->is<AstExprConstantNil>())
|
||||||
{
|
{
|
||||||
// HACK: we leave nil-initialized things floating under the assumption that they will later be populated.
|
// HACK: we leave nil-initialized things floating under the assumption that they will later be populated.
|
||||||
// See the test TypeInfer/infer_locals_with_nil_value.
|
// See the test TypeInfer/infer_locals_with_nil_value.
|
||||||
@ -218,7 +219,7 @@ void ConstraintGraphBuilder::visit(const ScopePtr& scope, AstStatLocal* local)
|
|||||||
}
|
}
|
||||||
else if (i == local->values.size - 1)
|
else if (i == local->values.size - 1)
|
||||||
{
|
{
|
||||||
TypePackId exprPack = checkPack(scope, local->values.data[i]);
|
TypePackId exprPack = checkPack(scope, value);
|
||||||
|
|
||||||
if (i < local->vars.size)
|
if (i < local->vars.size)
|
||||||
{
|
{
|
||||||
@ -229,7 +230,7 @@ void ConstraintGraphBuilder::visit(const ScopePtr& scope, AstStatLocal* local)
|
|||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
TypeId exprType = check(scope, local->values.data[i]);
|
TypeId exprType = check(scope, value);
|
||||||
if (i < varTypes.size())
|
if (i < varTypes.size())
|
||||||
addConstraint(scope, SubtypeConstraint{varTypes[i], exprType});
|
addConstraint(scope, SubtypeConstraint{varTypes[i], exprType});
|
||||||
}
|
}
|
||||||
@ -1107,9 +1108,7 @@ TypeId ConstraintGraphBuilder::resolveType(const ScopePtr& scope, AstType* ty, b
|
|||||||
|
|
||||||
if (topLevel)
|
if (topLevel)
|
||||||
{
|
{
|
||||||
addConstraint(scope, TypeAliasExpansionConstraint{
|
addConstraint(scope, TypeAliasExpansionConstraint{ /* target */ result });
|
||||||
/* target */ result,
|
|
||||||
});
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -11,6 +11,7 @@
|
|||||||
|
|
||||||
LUAU_FASTFLAGVARIABLE(DebugLuauLogSolver, false);
|
LUAU_FASTFLAGVARIABLE(DebugLuauLogSolver, false);
|
||||||
LUAU_FASTFLAGVARIABLE(DebugLuauLogSolverToJson, false);
|
LUAU_FASTFLAGVARIABLE(DebugLuauLogSolverToJson, false);
|
||||||
|
LUAU_FASTFLAG(LuauFixNameMaps)
|
||||||
|
|
||||||
namespace Luau
|
namespace Luau
|
||||||
{
|
{
|
||||||
@ -19,9 +20,17 @@ namespace Luau
|
|||||||
{
|
{
|
||||||
for (const auto& [k, v] : scope->bindings)
|
for (const auto& [k, v] : scope->bindings)
|
||||||
{
|
{
|
||||||
auto d = toStringDetailed(v.typeId, opts);
|
if (FFlag::LuauFixNameMaps)
|
||||||
opts.nameMap = d.nameMap;
|
{
|
||||||
printf("\t%s : %s\n", k.c_str(), d.name.c_str());
|
auto d = toString(v.typeId, opts);
|
||||||
|
printf("\t%s : %s\n", k.c_str(), d.c_str());
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
auto d = toStringDetailed(v.typeId, opts);
|
||||||
|
opts.DEPRECATED_nameMap = d.DEPRECATED_nameMap;
|
||||||
|
printf("\t%s : %s\n", k.c_str(), d.name.c_str());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
for (NotNull<Scope> child : scope->children)
|
for (NotNull<Scope> child : scope->children)
|
||||||
@ -212,12 +221,22 @@ void dump(NotNull<Scope> rootScope, ToStringOptions& opts)
|
|||||||
void dump(ConstraintSolver* cs, ToStringOptions& opts)
|
void dump(ConstraintSolver* cs, ToStringOptions& opts)
|
||||||
{
|
{
|
||||||
printf("constraints:\n");
|
printf("constraints:\n");
|
||||||
for (const Constraint* c : cs->unsolvedConstraints)
|
for (NotNull<const Constraint> c : cs->unsolvedConstraints)
|
||||||
{
|
{
|
||||||
printf("\t%s\n", toString(*c, opts).c_str());
|
auto it = cs->blockedConstraints.find(c);
|
||||||
|
int blockCount = it == cs->blockedConstraints.end() ? 0 : int(it->second);
|
||||||
|
printf("\t%d\t%s\n", blockCount, toString(*c, opts).c_str());
|
||||||
|
|
||||||
for (const Constraint* dep : c->dependencies)
|
for (NotNull<Constraint> dep : c->dependencies)
|
||||||
printf("\t\t%s\n", toString(*dep, opts).c_str());
|
{
|
||||||
|
auto unsolvedIter = std::find(begin(cs->unsolvedConstraints), end(cs->unsolvedConstraints), dep);
|
||||||
|
if (unsolvedIter == cs->unsolvedConstraints.end())
|
||||||
|
continue;
|
||||||
|
|
||||||
|
auto it = cs->blockedConstraints.find(dep);
|
||||||
|
int blockCount = it == cs->blockedConstraints.end() ? 0 : int(it->second);
|
||||||
|
printf("\t%d\t\t%s\n", blockCount, toString(*dep, opts).c_str());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -273,7 +292,7 @@ void ConstraintSolver::run()
|
|||||||
|
|
||||||
if (FFlag::DebugLuauLogSolverToJson)
|
if (FFlag::DebugLuauLogSolverToJson)
|
||||||
{
|
{
|
||||||
logger.prepareStepSnapshot(rootScope, c, unsolvedConstraints);
|
logger.prepareStepSnapshot(rootScope, c, unsolvedConstraints, force);
|
||||||
}
|
}
|
||||||
|
|
||||||
bool success = tryDispatch(c, force);
|
bool success = tryDispatch(c, force);
|
||||||
@ -282,6 +301,7 @@ void ConstraintSolver::run()
|
|||||||
|
|
||||||
if (success)
|
if (success)
|
||||||
{
|
{
|
||||||
|
unblock(c);
|
||||||
unsolvedConstraints.erase(unsolvedConstraints.begin() + i);
|
unsolvedConstraints.erase(unsolvedConstraints.begin() + i);
|
||||||
|
|
||||||
if (FFlag::DebugLuauLogSolverToJson)
|
if (FFlag::DebugLuauLogSolverToJson)
|
||||||
@ -375,18 +395,12 @@ bool ConstraintSolver::tryDispatch(const SubtypeConstraint& c, NotNull<const Con
|
|||||||
|
|
||||||
unify(c.subType, c.superType, constraint->scope);
|
unify(c.subType, c.superType, constraint->scope);
|
||||||
|
|
||||||
unblock(c.subType);
|
|
||||||
unblock(c.superType);
|
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool ConstraintSolver::tryDispatch(const PackSubtypeConstraint& c, NotNull<const Constraint> constraint, bool force)
|
bool ConstraintSolver::tryDispatch(const PackSubtypeConstraint& c, NotNull<const Constraint> constraint, bool force)
|
||||||
{
|
{
|
||||||
unify(c.subPack, c.superPack, constraint->scope);
|
unify(c.subPack, c.superPack, constraint->scope);
|
||||||
unblock(c.subPack);
|
|
||||||
unblock(c.superPack);
|
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -395,13 +409,12 @@ bool ConstraintSolver::tryDispatch(const GeneralizationConstraint& c, NotNull<co
|
|||||||
if (isBlocked(c.sourceType))
|
if (isBlocked(c.sourceType))
|
||||||
return block(c.sourceType, constraint);
|
return block(c.sourceType, constraint);
|
||||||
|
|
||||||
if (isBlocked(c.generalizedType))
|
|
||||||
asMutable(c.generalizedType)->ty.emplace<BoundTypeVar>(c.sourceType);
|
|
||||||
else
|
|
||||||
unify(c.generalizedType, c.sourceType, constraint->scope);
|
|
||||||
|
|
||||||
TypeId generalized = quantify(arena, c.sourceType, constraint->scope);
|
TypeId generalized = quantify(arena, c.sourceType, constraint->scope);
|
||||||
*asMutable(c.sourceType) = *generalized;
|
|
||||||
|
if (isBlocked(c.generalizedType))
|
||||||
|
asMutable(c.generalizedType)->ty.emplace<BoundTypeVar>(generalized);
|
||||||
|
else
|
||||||
|
unify(c.generalizedType, generalized, constraint->scope);
|
||||||
|
|
||||||
unblock(c.generalizedType);
|
unblock(c.generalizedType);
|
||||||
unblock(c.sourceType);
|
unblock(c.sourceType);
|
||||||
@ -455,23 +468,44 @@ bool ConstraintSolver::tryDispatch(const BinaryConstraint& c, NotNull<const Cons
|
|||||||
{
|
{
|
||||||
TypeId leftType = follow(c.leftType);
|
TypeId leftType = follow(c.leftType);
|
||||||
TypeId rightType = follow(c.rightType);
|
TypeId rightType = follow(c.rightType);
|
||||||
|
TypeId resultType = follow(c.resultType);
|
||||||
|
|
||||||
if (isBlocked(leftType) || isBlocked(rightType))
|
if (isBlocked(leftType) || isBlocked(rightType))
|
||||||
{
|
{
|
||||||
block(leftType, constraint);
|
/* Compound assignments create constraints of the form
|
||||||
block(rightType, constraint);
|
*
|
||||||
return false;
|
* A <: Binary<op, A, B>
|
||||||
|
*
|
||||||
|
* This constraint is the one that is meant to unblock A, so it doesn't
|
||||||
|
* make any sense to stop and wait for someone else to do it.
|
||||||
|
*/
|
||||||
|
if (leftType != resultType && rightType != resultType)
|
||||||
|
{
|
||||||
|
block(c.leftType, constraint);
|
||||||
|
block(c.rightType, constraint);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (isNumber(leftType))
|
if (isNumber(leftType))
|
||||||
{
|
{
|
||||||
unify(leftType, rightType, constraint->scope);
|
unify(leftType, rightType, constraint->scope);
|
||||||
asMutable(c.resultType)->ty.emplace<BoundTypeVar>(leftType);
|
asMutable(resultType)->ty.emplace<BoundTypeVar>(leftType);
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (get<FreeTypeVar>(leftType) && !force)
|
if (!force)
|
||||||
return block(leftType, constraint);
|
{
|
||||||
|
if (get<FreeTypeVar>(leftType))
|
||||||
|
return block(leftType, constraint);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (isBlocked(leftType))
|
||||||
|
{
|
||||||
|
asMutable(resultType)->ty.emplace<BoundTypeVar>(getSingletonTypes().errorRecoveryType());
|
||||||
|
// reportError(constraint->location, CannotInferBinaryOperation{c.op, std::nullopt, CannotInferBinaryOperation::Operation});
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
// TODO metatables, classes
|
// TODO metatables, classes
|
||||||
|
|
||||||
@ -706,17 +740,23 @@ void ConstraintSolver::block_(BlockedConstraintId target, NotNull<const Constrai
|
|||||||
|
|
||||||
void ConstraintSolver::block(NotNull<const Constraint> target, NotNull<const Constraint> constraint)
|
void ConstraintSolver::block(NotNull<const Constraint> target, NotNull<const Constraint> constraint)
|
||||||
{
|
{
|
||||||
|
if (FFlag::DebugLuauLogSolver)
|
||||||
|
printf("block Constraint %s on\t%s\n", toString(*target).c_str(), toString(*constraint).c_str());
|
||||||
block_(target, constraint);
|
block_(target, constraint);
|
||||||
}
|
}
|
||||||
|
|
||||||
bool ConstraintSolver::block(TypeId target, NotNull<const Constraint> constraint)
|
bool ConstraintSolver::block(TypeId target, NotNull<const Constraint> constraint)
|
||||||
{
|
{
|
||||||
|
if (FFlag::DebugLuauLogSolver)
|
||||||
|
printf("block TypeId %s on\t%s\n", toString(target).c_str(), toString(*constraint).c_str());
|
||||||
block_(target, constraint);
|
block_(target, constraint);
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool ConstraintSolver::block(TypePackId target, NotNull<const Constraint> constraint)
|
bool ConstraintSolver::block(TypePackId target, NotNull<const Constraint> constraint)
|
||||||
{
|
{
|
||||||
|
if (FFlag::DebugLuauLogSolver)
|
||||||
|
printf("block TypeId %s on\t%s\n", toString(target).c_str(), toString(*constraint).c_str());
|
||||||
block_(target, constraint);
|
block_(target, constraint);
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
@ -731,6 +771,9 @@ void ConstraintSolver::unblock_(BlockedConstraintId progressed)
|
|||||||
for (NotNull<const Constraint> unblockedConstraint : it->second)
|
for (NotNull<const Constraint> unblockedConstraint : it->second)
|
||||||
{
|
{
|
||||||
auto& count = blockedConstraints[unblockedConstraint];
|
auto& count = blockedConstraints[unblockedConstraint];
|
||||||
|
if (FFlag::DebugLuauLogSolver)
|
||||||
|
printf("Unblocking count=%d\t%s\n", int(count), toString(*unblockedConstraint).c_str());
|
||||||
|
|
||||||
// This assertion being hit indicates that `blocked` and
|
// This assertion being hit indicates that `blocked` and
|
||||||
// `blockedConstraints` desynchronized at some point. This is problematic
|
// `blockedConstraints` desynchronized at some point. This is problematic
|
||||||
// because we rely on this count being correct to skip over blocked
|
// because we rely on this count being correct to skip over blocked
|
||||||
@ -757,6 +800,18 @@ void ConstraintSolver::unblock(TypePackId progressed)
|
|||||||
return unblock_(progressed);
|
return unblock_(progressed);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void ConstraintSolver::unblock(const std::vector<TypeId>& types)
|
||||||
|
{
|
||||||
|
for (TypeId t : types)
|
||||||
|
unblock(t);
|
||||||
|
}
|
||||||
|
|
||||||
|
void ConstraintSolver::unblock(const std::vector<TypePackId>& packs)
|
||||||
|
{
|
||||||
|
for (TypePackId t : packs)
|
||||||
|
unblock(t);
|
||||||
|
}
|
||||||
|
|
||||||
bool ConstraintSolver::isBlocked(TypeId ty)
|
bool ConstraintSolver::isBlocked(TypeId ty)
|
||||||
{
|
{
|
||||||
return nullptr != get<BlockedTypeVar>(follow(ty)) || nullptr != get<PendingExpansionTypeVar>(follow(ty));
|
return nullptr != get<BlockedTypeVar>(follow(ty)) || nullptr != get<PendingExpansionTypeVar>(follow(ty));
|
||||||
@ -774,7 +829,13 @@ void ConstraintSolver::unify(TypeId subType, TypeId superType, NotNull<Scope> sc
|
|||||||
Unifier u{arena, Mode::Strict, scope, Location{}, Covariant, sharedState};
|
Unifier u{arena, Mode::Strict, scope, Location{}, Covariant, sharedState};
|
||||||
|
|
||||||
u.tryUnify(subType, superType);
|
u.tryUnify(subType, superType);
|
||||||
|
|
||||||
|
const auto [changedTypes, changedPacks] = u.log.getChanges();
|
||||||
|
|
||||||
u.log.commit();
|
u.log.commit();
|
||||||
|
|
||||||
|
unblock(changedTypes);
|
||||||
|
unblock(changedPacks);
|
||||||
}
|
}
|
||||||
|
|
||||||
void ConstraintSolver::unify(TypePackId subPack, TypePackId superPack, NotNull<Scope> scope)
|
void ConstraintSolver::unify(TypePackId subPack, TypePackId superPack, NotNull<Scope> scope)
|
||||||
@ -783,7 +844,13 @@ void ConstraintSolver::unify(TypePackId subPack, TypePackId superPack, NotNull<S
|
|||||||
Unifier u{arena, Mode::Strict, scope, Location{}, Covariant, sharedState};
|
Unifier u{arena, Mode::Strict, scope, Location{}, Covariant, sharedState};
|
||||||
|
|
||||||
u.tryUnify(subPack, superPack);
|
u.tryUnify(subPack, superPack);
|
||||||
|
|
||||||
|
const auto [changedTypes, changedPacks] = u.log.getChanges();
|
||||||
|
|
||||||
u.log.commit();
|
u.log.commit();
|
||||||
|
|
||||||
|
unblock(changedTypes);
|
||||||
|
unblock(changedPacks);
|
||||||
}
|
}
|
||||||
|
|
||||||
void ConstraintSolver::pushConstraint(ConstraintV cv, NotNull<Scope> scope)
|
void ConstraintSolver::pushConstraint(ConstraintV cv, NotNull<Scope> scope)
|
||||||
|
@ -4,6 +4,8 @@
|
|||||||
|
|
||||||
#include "Luau/JsonEmitter.h"
|
#include "Luau/JsonEmitter.h"
|
||||||
|
|
||||||
|
LUAU_FASTFLAG(LuauFixNameMaps);
|
||||||
|
|
||||||
namespace Luau
|
namespace Luau
|
||||||
{
|
{
|
||||||
|
|
||||||
@ -17,9 +19,14 @@ static void dumpScopeAndChildren(const Scope* scope, Json::JsonEmitter& emitter,
|
|||||||
|
|
||||||
for (const auto& [name, binding] : scope->bindings)
|
for (const auto& [name, binding] : scope->bindings)
|
||||||
{
|
{
|
||||||
ToStringResult result = toStringDetailed(binding.typeId, opts);
|
if (FFlag::LuauFixNameMaps)
|
||||||
opts.nameMap = std::move(result.nameMap);
|
o.writePair(name.c_str(), toString(binding.typeId, opts));
|
||||||
o.writePair(name.c_str(), result.name);
|
else
|
||||||
|
{
|
||||||
|
ToStringResult result = toStringDetailed(binding.typeId, opts);
|
||||||
|
opts.DEPRECATED_nameMap = std::move(result.DEPRECATED_nameMap);
|
||||||
|
o.writePair(name.c_str(), result.name);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
o.finish();
|
o.finish();
|
||||||
@ -30,6 +37,7 @@ static void dumpScopeAndChildren(const Scope* scope, Json::JsonEmitter& emitter,
|
|||||||
Json::ArrayEmitter a = emitter.writeArray();
|
Json::ArrayEmitter a = emitter.writeArray();
|
||||||
for (const Scope* child : scope->children)
|
for (const Scope* child : scope->children)
|
||||||
{
|
{
|
||||||
|
emitter.writeComma();
|
||||||
dumpScopeAndChildren(child, emitter, opts);
|
dumpScopeAndChildren(child, emitter, opts);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -39,7 +47,8 @@ static void dumpScopeAndChildren(const Scope* scope, Json::JsonEmitter& emitter,
|
|||||||
|
|
||||||
static std::string dumpConstraintsToDot(std::vector<NotNull<const Constraint>>& constraints, ToStringOptions& opts)
|
static std::string dumpConstraintsToDot(std::vector<NotNull<const Constraint>>& constraints, ToStringOptions& opts)
|
||||||
{
|
{
|
||||||
std::string result = "digraph Constraints {\\n";
|
std::string result = "digraph Constraints {\n";
|
||||||
|
result += "rankdir=LR\n";
|
||||||
|
|
||||||
std::unordered_set<NotNull<const Constraint>> contained;
|
std::unordered_set<NotNull<const Constraint>> contained;
|
||||||
for (NotNull<const Constraint> c : constraints)
|
for (NotNull<const Constraint> c : constraints)
|
||||||
@ -49,11 +58,19 @@ static std::string dumpConstraintsToDot(std::vector<NotNull<const Constraint>>&
|
|||||||
|
|
||||||
for (NotNull<const Constraint> c : constraints)
|
for (NotNull<const Constraint> c : constraints)
|
||||||
{
|
{
|
||||||
|
std::string shape;
|
||||||
|
if (get<SubtypeConstraint>(*c))
|
||||||
|
shape = "box";
|
||||||
|
else if (get<PackSubtypeConstraint>(*c))
|
||||||
|
shape = "box3d";
|
||||||
|
else
|
||||||
|
shape = "oval";
|
||||||
|
|
||||||
std::string id = std::to_string(reinterpret_cast<size_t>(c.get()));
|
std::string id = std::to_string(reinterpret_cast<size_t>(c.get()));
|
||||||
result += id;
|
result += id;
|
||||||
result += " [label=\\\"";
|
result += " [label=\"";
|
||||||
result += toString(*c, opts).c_str();
|
result += toString(*c, opts);
|
||||||
result += "\\\"];\\n";
|
result += "\" shape=" + shape + "];\n";
|
||||||
|
|
||||||
for (NotNull<const Constraint> dep : c->dependencies)
|
for (NotNull<const Constraint> dep : c->dependencies)
|
||||||
{
|
{
|
||||||
@ -63,7 +80,7 @@ static std::string dumpConstraintsToDot(std::vector<NotNull<const Constraint>>&
|
|||||||
result += std::to_string(reinterpret_cast<size_t>(dep.get()));
|
result += std::to_string(reinterpret_cast<size_t>(dep.get()));
|
||||||
result += " -> ";
|
result += " -> ";
|
||||||
result += id;
|
result += id;
|
||||||
result += ";\\n";
|
result += ";\n";
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -102,7 +119,7 @@ void ConstraintSolverLogger::captureBoundarySnapshot(const Scope* rootScope, std
|
|||||||
}
|
}
|
||||||
|
|
||||||
void ConstraintSolverLogger::prepareStepSnapshot(
|
void ConstraintSolverLogger::prepareStepSnapshot(
|
||||||
const Scope* rootScope, NotNull<const Constraint> current, std::vector<NotNull<const Constraint>>& unsolvedConstraints)
|
const Scope* rootScope, NotNull<const Constraint> current, std::vector<NotNull<const Constraint>>& unsolvedConstraints, bool force)
|
||||||
{
|
{
|
||||||
Json::JsonEmitter emitter;
|
Json::JsonEmitter emitter;
|
||||||
Json::ObjectEmitter o = emitter.writeObject();
|
Json::ObjectEmitter o = emitter.writeObject();
|
||||||
@ -110,6 +127,7 @@ void ConstraintSolverLogger::prepareStepSnapshot(
|
|||||||
o.writePair("constraintGraph", dumpConstraintsToDot(unsolvedConstraints, opts));
|
o.writePair("constraintGraph", dumpConstraintsToDot(unsolvedConstraints, opts));
|
||||||
o.writePair("currentId", std::to_string(reinterpret_cast<size_t>(current.get())));
|
o.writePair("currentId", std::to_string(reinterpret_cast<size_t>(current.get())));
|
||||||
o.writePair("current", toString(*current, opts));
|
o.writePair("current", toString(*current, opts));
|
||||||
|
o.writePair("force", force);
|
||||||
emitter.writeComma();
|
emitter.writeComma();
|
||||||
Json::write(emitter, "rootScope");
|
Json::write(emitter, "rootScope");
|
||||||
emitter.writeRaw(":");
|
emitter.writeRaw(":");
|
||||||
|
@ -15,6 +15,7 @@
|
|||||||
LUAU_FASTINTVARIABLE(LuauSuggestionDistance, 4)
|
LUAU_FASTINTVARIABLE(LuauSuggestionDistance, 4)
|
||||||
LUAU_FASTFLAGVARIABLE(LuauLintGlobalNeverReadBeforeWritten, false)
|
LUAU_FASTFLAGVARIABLE(LuauLintGlobalNeverReadBeforeWritten, false)
|
||||||
LUAU_FASTFLAGVARIABLE(LuauLintComparisonPrecedence, false)
|
LUAU_FASTFLAGVARIABLE(LuauLintComparisonPrecedence, false)
|
||||||
|
LUAU_FASTFLAGVARIABLE(LuauLintFixDeprecationMessage, false)
|
||||||
|
|
||||||
namespace Luau
|
namespace Luau
|
||||||
{
|
{
|
||||||
@ -306,11 +307,22 @@ private:
|
|||||||
emitWarning(*context, LintWarning::Code_UnknownGlobal, gv->location, "Unknown global '%s'", gv->name.value);
|
emitWarning(*context, LintWarning::Code_UnknownGlobal, gv->location, "Unknown global '%s'", gv->name.value);
|
||||||
else if (g->deprecated)
|
else if (g->deprecated)
|
||||||
{
|
{
|
||||||
if (*g->deprecated)
|
if (FFlag::LuauLintFixDeprecationMessage)
|
||||||
emitWarning(*context, LintWarning::Code_DeprecatedGlobal, gv->location, "Global '%s' is deprecated, use '%s' instead",
|
{
|
||||||
gv->name.value, *g->deprecated);
|
if (const char* replacement = *g->deprecated; replacement && strlen(replacement))
|
||||||
|
emitWarning(*context, LintWarning::Code_DeprecatedGlobal, gv->location, "Global '%s' is deprecated, use '%s' instead",
|
||||||
|
gv->name.value, replacement);
|
||||||
|
else
|
||||||
|
emitWarning(*context, LintWarning::Code_DeprecatedGlobal, gv->location, "Global '%s' is deprecated", gv->name.value);
|
||||||
|
}
|
||||||
else
|
else
|
||||||
emitWarning(*context, LintWarning::Code_DeprecatedGlobal, gv->location, "Global '%s' is deprecated", gv->name.value);
|
{
|
||||||
|
if (*g->deprecated)
|
||||||
|
emitWarning(*context, LintWarning::Code_DeprecatedGlobal, gv->location, "Global '%s' is deprecated, use '%s' instead",
|
||||||
|
gv->name.value, *g->deprecated);
|
||||||
|
else
|
||||||
|
emitWarning(*context, LintWarning::Code_DeprecatedGlobal, gv->location, "Global '%s' is deprecated", gv->name.value);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -73,7 +73,7 @@ void StateDot::visitChild(TypeId ty, int parentIndex, const char* linkName)
|
|||||||
if (opts.duplicatePrimitives && canDuplicatePrimitive(ty))
|
if (opts.duplicatePrimitives && canDuplicatePrimitive(ty))
|
||||||
{
|
{
|
||||||
if (get<PrimitiveTypeVar>(ty))
|
if (get<PrimitiveTypeVar>(ty))
|
||||||
formatAppend(result, "n%d [label=\"%s\"];\n", index, toStringDetailed(ty, {}).name.c_str());
|
formatAppend(result, "n%d [label=\"%s\"];\n", index, toString(ty).c_str());
|
||||||
else if (get<AnyTypeVar>(ty))
|
else if (get<AnyTypeVar>(ty))
|
||||||
formatAppend(result, "n%d [label=\"any\"];\n", index);
|
formatAppend(result, "n%d [label=\"any\"];\n", index);
|
||||||
}
|
}
|
||||||
@ -233,7 +233,7 @@ void StateDot::visitChildren(TypeId ty, int index)
|
|||||||
}
|
}
|
||||||
else if (get<PrimitiveTypeVar>(ty))
|
else if (get<PrimitiveTypeVar>(ty))
|
||||||
{
|
{
|
||||||
formatAppend(result, "PrimitiveTypeVar %s", toStringDetailed(ty, {}).name.c_str());
|
formatAppend(result, "PrimitiveTypeVar %s", toString(ty).c_str());
|
||||||
finishNodeLabel(ty);
|
finishNodeLabel(ty);
|
||||||
finishNode();
|
finishNode();
|
||||||
}
|
}
|
||||||
|
@ -13,6 +13,7 @@
|
|||||||
LUAU_FASTFLAG(LuauLowerBoundsCalculation)
|
LUAU_FASTFLAG(LuauLowerBoundsCalculation)
|
||||||
LUAU_FASTFLAG(LuauUnknownAndNeverType)
|
LUAU_FASTFLAG(LuauUnknownAndNeverType)
|
||||||
LUAU_FASTFLAGVARIABLE(LuauSpecialTypesAsterisked, false)
|
LUAU_FASTFLAGVARIABLE(LuauSpecialTypesAsterisked, false)
|
||||||
|
LUAU_FASTFLAGVARIABLE(LuauFixNameMaps, false)
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Prefix generic typenames with gen-
|
* Prefix generic typenames with gen-
|
||||||
@ -116,7 +117,7 @@ static std::pair<bool, std::optional<Luau::Name>> canUseTypeNameInScope(ScopePtr
|
|||||||
|
|
||||||
struct StringifierState
|
struct StringifierState
|
||||||
{
|
{
|
||||||
const ToStringOptions& opts;
|
ToStringOptions& opts;
|
||||||
ToStringResult& result;
|
ToStringResult& result;
|
||||||
|
|
||||||
std::unordered_map<TypeId, std::string> cycleNames;
|
std::unordered_map<TypeId, std::string> cycleNames;
|
||||||
@ -127,18 +128,28 @@ struct StringifierState
|
|||||||
|
|
||||||
bool exhaustive;
|
bool exhaustive;
|
||||||
|
|
||||||
StringifierState(const ToStringOptions& opts, ToStringResult& result, const std::optional<ToStringNameMap>& nameMap)
|
StringifierState(ToStringOptions& opts, ToStringResult& result, const std::optional<ToStringNameMap>& DEPRECATED_nameMap)
|
||||||
: opts(opts)
|
: opts(opts)
|
||||||
, result(result)
|
, result(result)
|
||||||
, exhaustive(opts.exhaustive)
|
, exhaustive(opts.exhaustive)
|
||||||
{
|
{
|
||||||
if (nameMap)
|
if (!FFlag::LuauFixNameMaps && DEPRECATED_nameMap)
|
||||||
result.nameMap = *nameMap;
|
result.DEPRECATED_nameMap = *DEPRECATED_nameMap;
|
||||||
|
|
||||||
for (const auto& [_, v] : result.nameMap.typeVars)
|
if (!FFlag::LuauFixNameMaps)
|
||||||
usedNames.insert(v);
|
{
|
||||||
for (const auto& [_, v] : result.nameMap.typePacks)
|
for (const auto& [_, v] : result.DEPRECATED_nameMap.typeVars)
|
||||||
usedNames.insert(v);
|
usedNames.insert(v);
|
||||||
|
for (const auto& [_, v] : result.DEPRECATED_nameMap.typePacks)
|
||||||
|
usedNames.insert(v);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
for (const auto& [_, v] : opts.nameMap.typeVars)
|
||||||
|
usedNames.insert(v);
|
||||||
|
for (const auto& [_, v] : opts.nameMap.typePacks)
|
||||||
|
usedNames.insert(v);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
bool hasSeen(const void* tv)
|
bool hasSeen(const void* tv)
|
||||||
@ -161,8 +172,8 @@ struct StringifierState
|
|||||||
|
|
||||||
std::string getName(TypeId ty)
|
std::string getName(TypeId ty)
|
||||||
{
|
{
|
||||||
const size_t s = result.nameMap.typeVars.size();
|
const size_t s = FFlag::LuauFixNameMaps ? opts.nameMap.typeVars.size() : result.DEPRECATED_nameMap.typeVars.size();
|
||||||
std::string& n = result.nameMap.typeVars[ty];
|
std::string& n = FFlag::LuauFixNameMaps ? opts.nameMap.typeVars[ty] : result.DEPRECATED_nameMap.typeVars[ty];
|
||||||
if (!n.empty())
|
if (!n.empty())
|
||||||
return n;
|
return n;
|
||||||
|
|
||||||
@ -184,8 +195,8 @@ struct StringifierState
|
|||||||
|
|
||||||
std::string getName(TypePackId ty)
|
std::string getName(TypePackId ty)
|
||||||
{
|
{
|
||||||
const size_t s = result.nameMap.typePacks.size();
|
const size_t s = FFlag::LuauFixNameMaps ? opts.nameMap.typePacks.size() : result.DEPRECATED_nameMap.typePacks.size();
|
||||||
std::string& n = result.nameMap.typePacks[ty];
|
std::string& n = FFlag::LuauFixNameMaps ? opts.nameMap.typePacks[ty] : result.DEPRECATED_nameMap.typePacks[ty];
|
||||||
if (!n.empty())
|
if (!n.empty())
|
||||||
return n;
|
return n;
|
||||||
|
|
||||||
@ -378,7 +389,10 @@ struct TypeVarStringifier
|
|||||||
if (gtv.explicitName)
|
if (gtv.explicitName)
|
||||||
{
|
{
|
||||||
state.usedNames.insert(gtv.name);
|
state.usedNames.insert(gtv.name);
|
||||||
state.result.nameMap.typeVars[ty] = gtv.name;
|
if (FFlag::LuauFixNameMaps)
|
||||||
|
state.opts.nameMap.typeVars[ty] = gtv.name;
|
||||||
|
else
|
||||||
|
state.result.DEPRECATED_nameMap.typeVars[ty] = gtv.name;
|
||||||
state.emit(gtv.name);
|
state.emit(gtv.name);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
@ -989,7 +1003,10 @@ struct TypePackStringifier
|
|||||||
if (pack.explicitName)
|
if (pack.explicitName)
|
||||||
{
|
{
|
||||||
state.usedNames.insert(pack.name);
|
state.usedNames.insert(pack.name);
|
||||||
state.result.nameMap.typePacks[tp] = pack.name;
|
if (FFlag::LuauFixNameMaps)
|
||||||
|
state.opts.nameMap.typePacks[tp] = pack.name;
|
||||||
|
else
|
||||||
|
state.result.DEPRECATED_nameMap.typePacks[tp] = pack.name;
|
||||||
state.emit(pack.name);
|
state.emit(pack.name);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
@ -1070,7 +1087,7 @@ static void assignCycleNames(const std::set<TypeId>& cycles, const std::set<Type
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
ToStringResult toStringDetailed(TypeId ty, const ToStringOptions& opts)
|
ToStringResult toStringDetailed(TypeId ty, ToStringOptions& opts)
|
||||||
{
|
{
|
||||||
/*
|
/*
|
||||||
* 1. Walk the TypeVar and track seen TypeIds. When you reencounter a TypeId, add it to a set of seen cycles.
|
* 1. Walk the TypeVar and track seen TypeIds. When you reencounter a TypeId, add it to a set of seen cycles.
|
||||||
@ -1082,7 +1099,9 @@ ToStringResult toStringDetailed(TypeId ty, const ToStringOptions& opts)
|
|||||||
|
|
||||||
ToStringResult result;
|
ToStringResult result;
|
||||||
|
|
||||||
StringifierState state{opts, result, opts.nameMap};
|
StringifierState state = FFlag::LuauFixNameMaps
|
||||||
|
? StringifierState{opts, result, opts.nameMap}
|
||||||
|
: StringifierState{opts, result, opts.DEPRECATED_nameMap};
|
||||||
|
|
||||||
std::set<TypeId> cycles;
|
std::set<TypeId> cycles;
|
||||||
std::set<TypePackId> cycleTPs;
|
std::set<TypePackId> cycleTPs;
|
||||||
@ -1183,7 +1202,7 @@ ToStringResult toStringDetailed(TypeId ty, const ToStringOptions& opts)
|
|||||||
return result;
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
ToStringResult toStringDetailed(TypePackId tp, const ToStringOptions& opts)
|
ToStringResult toStringDetailed(TypePackId tp, ToStringOptions& opts)
|
||||||
{
|
{
|
||||||
/*
|
/*
|
||||||
* 1. Walk the TypeVar and track seen TypeIds. When you reencounter a TypeId, add it to a set of seen cycles.
|
* 1. Walk the TypeVar and track seen TypeIds. When you reencounter a TypeId, add it to a set of seen cycles.
|
||||||
@ -1192,7 +1211,9 @@ ToStringResult toStringDetailed(TypePackId tp, const ToStringOptions& opts)
|
|||||||
* 4. Print out the root of the type using the same algorithm as step 3.
|
* 4. Print out the root of the type using the same algorithm as step 3.
|
||||||
*/
|
*/
|
||||||
ToStringResult result;
|
ToStringResult result;
|
||||||
StringifierState state{opts, result, opts.nameMap};
|
StringifierState state = FFlag::LuauFixNameMaps
|
||||||
|
? StringifierState{opts, result, opts.nameMap}
|
||||||
|
: StringifierState{opts, result, opts.DEPRECATED_nameMap};
|
||||||
|
|
||||||
std::set<TypeId> cycles;
|
std::set<TypeId> cycles;
|
||||||
std::set<TypePackId> cycleTPs;
|
std::set<TypePackId> cycleTPs;
|
||||||
@ -1258,30 +1279,32 @@ ToStringResult toStringDetailed(TypePackId tp, const ToStringOptions& opts)
|
|||||||
return result;
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string toString(TypeId ty, const ToStringOptions& opts)
|
std::string toString(TypeId ty, ToStringOptions& opts)
|
||||||
{
|
{
|
||||||
return toStringDetailed(ty, opts).name;
|
return toStringDetailed(ty, opts).name;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string toString(TypePackId tp, const ToStringOptions& opts)
|
std::string toString(TypePackId tp, ToStringOptions& opts)
|
||||||
{
|
{
|
||||||
return toStringDetailed(tp, opts).name;
|
return toStringDetailed(tp, opts).name;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string toString(const TypeVar& tv, const ToStringOptions& opts)
|
std::string toString(const TypeVar& tv, ToStringOptions& opts)
|
||||||
{
|
{
|
||||||
return toString(const_cast<TypeId>(&tv), std::move(opts));
|
return toString(const_cast<TypeId>(&tv), opts);
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string toString(const TypePackVar& tp, const ToStringOptions& opts)
|
std::string toString(const TypePackVar& tp, ToStringOptions& opts)
|
||||||
{
|
{
|
||||||
return toString(const_cast<TypePackId>(&tp), std::move(opts));
|
return toString(const_cast<TypePackId>(&tp), opts);
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string toStringNamedFunction(const std::string& funcName, const FunctionTypeVar& ftv, const ToStringOptions& opts)
|
std::string toStringNamedFunction(const std::string& funcName, const FunctionTypeVar& ftv, ToStringOptions& opts)
|
||||||
{
|
{
|
||||||
ToStringResult result;
|
ToStringResult result;
|
||||||
StringifierState state(opts, result, opts.nameMap);
|
StringifierState state = FFlag::LuauFixNameMaps
|
||||||
|
? StringifierState{opts, result, opts.nameMap}
|
||||||
|
: StringifierState{opts, result, opts.DEPRECATED_nameMap};
|
||||||
TypeVarStringifier tvs{state};
|
TypeVarStringifier tvs{state};
|
||||||
|
|
||||||
state.emit(funcName);
|
state.emit(funcName);
|
||||||
@ -1413,69 +1436,67 @@ std::string toString(const Constraint& constraint, ToStringOptions& opts)
|
|||||||
auto go = [&opts](auto&& c) {
|
auto go = [&opts](auto&& c) {
|
||||||
using T = std::decay_t<decltype(c)>;
|
using T = std::decay_t<decltype(c)>;
|
||||||
|
|
||||||
|
// TODO: Inline and delete this function when clipping FFlag::LuauFixNameMaps
|
||||||
|
auto tos = [](auto&& a, ToStringOptions& opts)
|
||||||
|
{
|
||||||
|
if (FFlag::LuauFixNameMaps)
|
||||||
|
return toString(a, opts);
|
||||||
|
else
|
||||||
|
{
|
||||||
|
ToStringResult tsr = toStringDetailed(a, opts);
|
||||||
|
opts.DEPRECATED_nameMap = std::move(tsr.DEPRECATED_nameMap);
|
||||||
|
return tsr.name;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
if constexpr (std::is_same_v<T, SubtypeConstraint>)
|
if constexpr (std::is_same_v<T, SubtypeConstraint>)
|
||||||
{
|
{
|
||||||
ToStringResult subStr = toStringDetailed(c.subType, opts);
|
std::string subStr = tos(c.subType, opts);
|
||||||
opts.nameMap = std::move(subStr.nameMap);
|
std::string superStr = tos(c.superType, opts);
|
||||||
ToStringResult superStr = toStringDetailed(c.superType, opts);
|
return subStr + " <: " + superStr;
|
||||||
opts.nameMap = std::move(superStr.nameMap);
|
|
||||||
return subStr.name + " <: " + superStr.name;
|
|
||||||
}
|
}
|
||||||
else if constexpr (std::is_same_v<T, PackSubtypeConstraint>)
|
else if constexpr (std::is_same_v<T, PackSubtypeConstraint>)
|
||||||
{
|
{
|
||||||
ToStringResult subStr = toStringDetailed(c.subPack, opts);
|
std::string subStr = tos(c.subPack, opts);
|
||||||
opts.nameMap = std::move(subStr.nameMap);
|
std::string superStr = tos(c.superPack, opts);
|
||||||
ToStringResult superStr = toStringDetailed(c.superPack, opts);
|
return subStr + " <: " + superStr;
|
||||||
opts.nameMap = std::move(superStr.nameMap);
|
|
||||||
return subStr.name + " <: " + superStr.name;
|
|
||||||
}
|
}
|
||||||
else if constexpr (std::is_same_v<T, GeneralizationConstraint>)
|
else if constexpr (std::is_same_v<T, GeneralizationConstraint>)
|
||||||
{
|
{
|
||||||
ToStringResult subStr = toStringDetailed(c.generalizedType, opts);
|
std::string subStr = tos(c.generalizedType, opts);
|
||||||
opts.nameMap = std::move(subStr.nameMap);
|
std::string superStr = tos(c.sourceType, opts);
|
||||||
ToStringResult superStr = toStringDetailed(c.sourceType, opts);
|
return subStr + " ~ gen " + superStr;
|
||||||
opts.nameMap = std::move(superStr.nameMap);
|
|
||||||
return subStr.name + " ~ gen " + superStr.name;
|
|
||||||
}
|
}
|
||||||
else if constexpr (std::is_same_v<T, InstantiationConstraint>)
|
else if constexpr (std::is_same_v<T, InstantiationConstraint>)
|
||||||
{
|
{
|
||||||
ToStringResult subStr = toStringDetailed(c.subType, opts);
|
std::string subStr = tos(c.subType, opts);
|
||||||
opts.nameMap = std::move(subStr.nameMap);
|
std::string superStr = tos(c.superType, opts);
|
||||||
ToStringResult superStr = toStringDetailed(c.superType, opts);
|
return subStr + " ~ inst " + superStr;
|
||||||
opts.nameMap = std::move(superStr.nameMap);
|
|
||||||
return subStr.name + " ~ inst " + superStr.name;
|
|
||||||
}
|
}
|
||||||
else if constexpr (std::is_same_v<T, UnaryConstraint>)
|
else if constexpr (std::is_same_v<T, UnaryConstraint>)
|
||||||
{
|
{
|
||||||
ToStringResult resultStr = toStringDetailed(c.resultType, opts);
|
std::string resultStr = tos(c.resultType, opts);
|
||||||
opts.nameMap = std::move(resultStr.nameMap);
|
std::string operandStr = tos(c.operandType, opts);
|
||||||
ToStringResult operandStr = toStringDetailed(c.operandType, opts);
|
|
||||||
opts.nameMap = std::move(operandStr.nameMap);
|
|
||||||
|
|
||||||
return resultStr.name + " ~ Unary<" + toString(c.op) + ", " + operandStr.name + ">";
|
return resultStr + " ~ Unary<" + toString(c.op) + ", " + operandStr + ">";
|
||||||
}
|
}
|
||||||
else if constexpr (std::is_same_v<T, BinaryConstraint>)
|
else if constexpr (std::is_same_v<T, BinaryConstraint>)
|
||||||
{
|
{
|
||||||
ToStringResult resultStr = toStringDetailed(c.resultType);
|
std::string resultStr = tos(c.resultType, opts);
|
||||||
opts.nameMap = std::move(resultStr.nameMap);
|
std::string leftStr = tos(c.leftType, opts);
|
||||||
ToStringResult leftStr = toStringDetailed(c.leftType);
|
std::string rightStr = tos(c.rightType, opts);
|
||||||
opts.nameMap = std::move(leftStr.nameMap);
|
|
||||||
ToStringResult rightStr = toStringDetailed(c.rightType);
|
|
||||||
opts.nameMap = std::move(rightStr.nameMap);
|
|
||||||
|
|
||||||
return resultStr.name + " ~ Binary<" + toString(c.op) + ", " + leftStr.name + ", " + rightStr.name + ">";
|
return resultStr + " ~ Binary<" + toString(c.op) + ", " + leftStr + ", " + rightStr + ">";
|
||||||
}
|
}
|
||||||
else if constexpr (std::is_same_v<T, NameConstraint>)
|
else if constexpr (std::is_same_v<T, NameConstraint>)
|
||||||
{
|
{
|
||||||
ToStringResult namedStr = toStringDetailed(c.namedType, opts);
|
std::string namedStr = tos(c.namedType, opts);
|
||||||
opts.nameMap = std::move(namedStr.nameMap);
|
return "@name(" + namedStr + ") = " + c.name;
|
||||||
return "@name(" + namedStr.name + ") = " + c.name;
|
|
||||||
}
|
}
|
||||||
else if constexpr (std::is_same_v<T, TypeAliasExpansionConstraint>)
|
else if constexpr (std::is_same_v<T, TypeAliasExpansionConstraint>)
|
||||||
{
|
{
|
||||||
ToStringResult targetStr = toStringDetailed(c.target, opts);
|
std::string targetStr = tos(c.target, opts);
|
||||||
opts.nameMap = std::move(targetStr.nameMap);
|
return "expand " + targetStr;
|
||||||
return "expand " + targetStr.name;
|
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
static_assert(always_false_v<T>, "Non-exhaustive constraint switch");
|
static_assert(always_false_v<T>, "Non-exhaustive constraint switch");
|
||||||
|
@ -344,4 +344,16 @@ TypePackId TxnLog::follow(TypePackId tp) const
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
std::pair<std::vector<TypeId>, std::vector<TypePackId>> TxnLog::getChanges() const
|
||||||
|
{
|
||||||
|
std::pair<std::vector<TypeId>, std::vector<TypePackId>> result;
|
||||||
|
|
||||||
|
for (const auto& [typeId, _newState] : typeVarChanges)
|
||||||
|
result.first.push_back(typeId);
|
||||||
|
for (const auto& [typePackId, _newState] : typePackChanges)
|
||||||
|
result.second.push_back(typePackId);
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
} // namespace Luau
|
} // namespace Luau
|
||||||
|
@ -27,6 +27,7 @@ LUAU_FASTFLAG(LuauUnknownAndNeverType)
|
|||||||
LUAU_FASTFLAGVARIABLE(LuauDeduceGmatchReturnTypes, false)
|
LUAU_FASTFLAGVARIABLE(LuauDeduceGmatchReturnTypes, false)
|
||||||
LUAU_FASTFLAGVARIABLE(LuauMaybeGenericIntersectionTypes, false)
|
LUAU_FASTFLAGVARIABLE(LuauMaybeGenericIntersectionTypes, false)
|
||||||
LUAU_FASTFLAGVARIABLE(LuauDeduceFindMatchReturnTypes, false)
|
LUAU_FASTFLAGVARIABLE(LuauDeduceFindMatchReturnTypes, false)
|
||||||
|
LUAU_FASTFLAGVARIABLE(LuauStringFormatArgumentErrorFix, false)
|
||||||
|
|
||||||
namespace Luau
|
namespace Luau
|
||||||
{
|
{
|
||||||
@ -1139,11 +1140,21 @@ std::optional<WithPredicate<TypePackId>> magicFunctionFormat(
|
|||||||
}
|
}
|
||||||
|
|
||||||
// if we know the argument count or if we have too many arguments for sure, we can issue an error
|
// if we know the argument count or if we have too many arguments for sure, we can issue an error
|
||||||
size_t actualParamSize = params.size() - paramOffset;
|
if (FFlag::LuauStringFormatArgumentErrorFix)
|
||||||
|
{
|
||||||
|
size_t numActualParams = params.size();
|
||||||
|
size_t numExpectedParams = expected.size() + 1; // + 1 for the format string
|
||||||
|
|
||||||
if (expected.size() != actualParamSize && (!tail || expected.size() < actualParamSize))
|
if (numExpectedParams != numActualParams && (!tail || numExpectedParams < numActualParams))
|
||||||
typechecker.reportError(TypeError{expr.location, CountMismatch{expected.size(), actualParamSize}});
|
typechecker.reportError(TypeError{expr.location, CountMismatch{numExpectedParams, numActualParams}});
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
size_t actualParamSize = params.size() - paramOffset;
|
||||||
|
|
||||||
|
if (expected.size() != actualParamSize && (!tail || expected.size() < actualParamSize))
|
||||||
|
typechecker.reportError(TypeError{expr.location, CountMismatch{expected.size(), actualParamSize}});
|
||||||
|
}
|
||||||
return WithPredicate<TypePackId>{arena.addTypePack({typechecker.stringType})};
|
return WithPredicate<TypePackId>{arena.addTypePack({typechecker.stringType})};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -598,9 +598,9 @@ public:
|
|||||||
LUAU_RTTI(AstExprFunction)
|
LUAU_RTTI(AstExprFunction)
|
||||||
|
|
||||||
AstExprFunction(const Location& location, const AstArray<AstGenericType>& generics, const AstArray<AstGenericTypePack>& genericPacks,
|
AstExprFunction(const Location& location, const AstArray<AstGenericType>& generics, const AstArray<AstGenericTypePack>& genericPacks,
|
||||||
AstLocal* self, const AstArray<AstLocal*>& args, std::optional<Location> vararg, AstStatBlock* body, size_t functionDepth,
|
AstLocal* self, const AstArray<AstLocal*>& args, bool vararg, const Location& varargLocation, AstStatBlock* body, size_t functionDepth,
|
||||||
const AstName& debugname, std::optional<AstTypeList> returnAnnotation = {}, AstTypePack* varargAnnotation = nullptr, bool hasEnd = false,
|
const AstName& debugname, const std::optional<AstTypeList>& returnAnnotation = {}, AstTypePack* varargAnnotation = nullptr,
|
||||||
std::optional<Location> argLocation = std::nullopt);
|
bool hasEnd = false, const std::optional<Location>& argLocation = std::nullopt);
|
||||||
|
|
||||||
void visit(AstVisitor* visitor) override;
|
void visit(AstVisitor* visitor) override;
|
||||||
|
|
||||||
|
@ -11,6 +11,7 @@
|
|||||||
|
|
||||||
#include <initializer_list>
|
#include <initializer_list>
|
||||||
#include <optional>
|
#include <optional>
|
||||||
|
#include <tuple>
|
||||||
|
|
||||||
namespace Luau
|
namespace Luau
|
||||||
{
|
{
|
||||||
@ -109,8 +110,10 @@ private:
|
|||||||
// for namelist in explist do block end |
|
// for namelist in explist do block end |
|
||||||
AstStat* parseFor();
|
AstStat* parseFor();
|
||||||
|
|
||||||
// function funcname funcbody |
|
|
||||||
// funcname ::= Name {`.' Name} [`:' Name]
|
// funcname ::= Name {`.' Name} [`:' Name]
|
||||||
|
AstExpr* parseFunctionName(Location start, bool& hasself, AstName& debugname);
|
||||||
|
|
||||||
|
// function funcname funcbody
|
||||||
AstStat* parseFunctionStat();
|
AstStat* parseFunctionStat();
|
||||||
|
|
||||||
// local function Name funcbody |
|
// local function Name funcbody |
|
||||||
@ -135,8 +138,10 @@ private:
|
|||||||
// var [`+=' | `-=' | `*=' | `/=' | `%=' | `^=' | `..='] exp
|
// var [`+=' | `-=' | `*=' | `/=' | `%=' | `^=' | `..='] exp
|
||||||
AstStat* parseCompoundAssignment(AstExpr* initial, AstExprBinary::Op op);
|
AstStat* parseCompoundAssignment(AstExpr* initial, AstExprBinary::Op op);
|
||||||
|
|
||||||
// funcbody ::= `(' [parlist] `)' block end
|
std::pair<AstLocal*, AstArray<AstLocal*>> prepareFunctionArguments(const Location& start, bool hasself, const TempVector<Binding>& args);
|
||||||
// parlist ::= namelist [`,' `...'] | `...'
|
|
||||||
|
// funcbodyhead ::= `(' [namelist [`,' `...'] | `...'] `)' [`:` TypeAnnotation]
|
||||||
|
// funcbody ::= funcbodyhead block end
|
||||||
std::pair<AstExprFunction*, AstLocal*> parseFunctionBody(
|
std::pair<AstExprFunction*, AstLocal*> parseFunctionBody(
|
||||||
bool hasself, const Lexeme& matchFunction, const AstName& debugname, const Name* localName);
|
bool hasself, const Lexeme& matchFunction, const AstName& debugname, const Name* localName);
|
||||||
|
|
||||||
@ -148,7 +153,7 @@ private:
|
|||||||
|
|
||||||
// bindinglist ::= (binding | `...') {`,' bindinglist}
|
// bindinglist ::= (binding | `...') {`,' bindinglist}
|
||||||
// Returns the location of the vararg ..., or std::nullopt if the function is not vararg.
|
// Returns the location of the vararg ..., or std::nullopt if the function is not vararg.
|
||||||
std::pair<std::optional<Location>, AstTypePack*> parseBindingList(TempVector<Binding>& result, bool allowDot3 = false);
|
std::tuple<bool, Location, AstTypePack*> parseBindingList(TempVector<Binding>& result, bool allowDot3 = false);
|
||||||
|
|
||||||
AstType* parseOptionalTypeAnnotation();
|
AstType* parseOptionalTypeAnnotation();
|
||||||
|
|
||||||
|
@ -160,17 +160,17 @@ void AstExprIndexExpr::visit(AstVisitor* visitor)
|
|||||||
}
|
}
|
||||||
|
|
||||||
AstExprFunction::AstExprFunction(const Location& location, const AstArray<AstGenericType>& generics, const AstArray<AstGenericTypePack>& genericPacks,
|
AstExprFunction::AstExprFunction(const Location& location, const AstArray<AstGenericType>& generics, const AstArray<AstGenericTypePack>& genericPacks,
|
||||||
AstLocal* self, const AstArray<AstLocal*>& args, std::optional<Location> vararg, AstStatBlock* body, size_t functionDepth,
|
AstLocal* self, const AstArray<AstLocal*>& args, bool vararg, const Location& varargLocation, AstStatBlock* body, size_t functionDepth,
|
||||||
const AstName& debugname, std::optional<AstTypeList> returnAnnotation, AstTypePack* varargAnnotation, bool hasEnd,
|
const AstName& debugname, const std::optional<AstTypeList>& returnAnnotation, AstTypePack* varargAnnotation, bool hasEnd,
|
||||||
std::optional<Location> argLocation)
|
const std::optional<Location>& argLocation)
|
||||||
: AstExpr(ClassIndex(), location)
|
: AstExpr(ClassIndex(), location)
|
||||||
, generics(generics)
|
, generics(generics)
|
||||||
, genericPacks(genericPacks)
|
, genericPacks(genericPacks)
|
||||||
, self(self)
|
, self(self)
|
||||||
, args(args)
|
, args(args)
|
||||||
, returnAnnotation(returnAnnotation)
|
, returnAnnotation(returnAnnotation)
|
||||||
, vararg(vararg.has_value())
|
, vararg(vararg)
|
||||||
, varargLocation(vararg.value_or(Location()))
|
, varargLocation(varargLocation)
|
||||||
, varargAnnotation(varargAnnotation)
|
, varargAnnotation(varargAnnotation)
|
||||||
, body(body)
|
, body(body)
|
||||||
, functionDepth(functionDepth)
|
, functionDepth(functionDepth)
|
||||||
|
@ -605,16 +605,11 @@ AstStat* Parser::parseFor()
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// function funcname funcbody |
|
|
||||||
// funcname ::= Name {`.' Name} [`:' Name]
|
// funcname ::= Name {`.' Name} [`:' Name]
|
||||||
AstStat* Parser::parseFunctionStat()
|
AstExpr* Parser::parseFunctionName(Location start, bool& hasself, AstName& debugname)
|
||||||
{
|
{
|
||||||
Location start = lexer.current().location;
|
if (lexer.current().type == Lexeme::Name)
|
||||||
|
debugname = AstName(lexer.current().name);
|
||||||
Lexeme matchFunction = lexer.current();
|
|
||||||
nextLexeme();
|
|
||||||
|
|
||||||
AstName debugname = (lexer.current().type == Lexeme::Name) ? AstName(lexer.current().name) : AstName();
|
|
||||||
|
|
||||||
// parse funcname into a chain of indexing operators
|
// parse funcname into a chain of indexing operators
|
||||||
AstExpr* expr = parseNameExpr("function name");
|
AstExpr* expr = parseNameExpr("function name");
|
||||||
@ -640,8 +635,6 @@ AstStat* Parser::parseFunctionStat()
|
|||||||
recursionCounter = recursionCounterOld;
|
recursionCounter = recursionCounterOld;
|
||||||
|
|
||||||
// finish with :
|
// finish with :
|
||||||
bool hasself = false;
|
|
||||||
|
|
||||||
if (lexer.current().type == ':')
|
if (lexer.current().type == ':')
|
||||||
{
|
{
|
||||||
Position opPosition = lexer.current().location.begin;
|
Position opPosition = lexer.current().location.begin;
|
||||||
@ -657,6 +650,21 @@ AstStat* Parser::parseFunctionStat()
|
|||||||
hasself = true;
|
hasself = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return expr;
|
||||||
|
}
|
||||||
|
|
||||||
|
// function funcname funcbody
|
||||||
|
AstStat* Parser::parseFunctionStat()
|
||||||
|
{
|
||||||
|
Location start = lexer.current().location;
|
||||||
|
|
||||||
|
Lexeme matchFunction = lexer.current();
|
||||||
|
nextLexeme();
|
||||||
|
|
||||||
|
bool hasself = false;
|
||||||
|
AstName debugname;
|
||||||
|
AstExpr* expr = parseFunctionName(start, hasself, debugname);
|
||||||
|
|
||||||
matchRecoveryStopOnToken[Lexeme::ReservedEnd]++;
|
matchRecoveryStopOnToken[Lexeme::ReservedEnd]++;
|
||||||
|
|
||||||
AstExprFunction* body = parseFunctionBody(hasself, matchFunction, debugname, nullptr).first;
|
AstExprFunction* body = parseFunctionBody(hasself, matchFunction, debugname, nullptr).first;
|
||||||
@ -785,10 +793,11 @@ AstDeclaredClassProp Parser::parseDeclaredClassMethod()
|
|||||||
|
|
||||||
TempVector<Binding> args(scratchBinding);
|
TempVector<Binding> args(scratchBinding);
|
||||||
|
|
||||||
std::optional<Location> vararg = std::nullopt;
|
bool vararg = false;
|
||||||
|
Location varargLocation;
|
||||||
AstTypePack* varargAnnotation = nullptr;
|
AstTypePack* varargAnnotation = nullptr;
|
||||||
if (lexer.current().type != ')')
|
if (lexer.current().type != ')')
|
||||||
std::tie(vararg, varargAnnotation) = parseBindingList(args, /* allowDot3 */ true);
|
std::tie(vararg, varargLocation, varargAnnotation) = parseBindingList(args, /* allowDot3 */ true);
|
||||||
|
|
||||||
expectMatchAndConsume(')', matchParen);
|
expectMatchAndConsume(')', matchParen);
|
||||||
|
|
||||||
@ -842,11 +851,12 @@ AstStat* Parser::parseDeclaration(const Location& start)
|
|||||||
|
|
||||||
TempVector<Binding> args(scratchBinding);
|
TempVector<Binding> args(scratchBinding);
|
||||||
|
|
||||||
std::optional<Location> vararg;
|
bool vararg = false;
|
||||||
|
Location varargLocation;
|
||||||
AstTypePack* varargAnnotation = nullptr;
|
AstTypePack* varargAnnotation = nullptr;
|
||||||
|
|
||||||
if (lexer.current().type != ')')
|
if (lexer.current().type != ')')
|
||||||
std::tie(vararg, varargAnnotation) = parseBindingList(args, /* allowDot3= */ true);
|
std::tie(vararg, varargLocation, varargAnnotation) = parseBindingList(args, /* allowDot3= */ true);
|
||||||
|
|
||||||
expectMatchAndConsume(')', matchParen);
|
expectMatchAndConsume(')', matchParen);
|
||||||
|
|
||||||
@ -969,6 +979,21 @@ AstStat* Parser::parseCompoundAssignment(AstExpr* initial, AstExprBinary::Op op)
|
|||||||
return allocator.alloc<AstStatCompoundAssign>(Location(initial->location, value->location), op, initial, value);
|
return allocator.alloc<AstStatCompoundAssign>(Location(initial->location, value->location), op, initial, value);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
std::pair<AstLocal*, AstArray<AstLocal*>> Parser::prepareFunctionArguments(const Location& start, bool hasself, const TempVector<Binding>& args)
|
||||||
|
{
|
||||||
|
AstLocal* self = nullptr;
|
||||||
|
|
||||||
|
if (hasself)
|
||||||
|
self = pushLocal(Binding(Name(nameSelf, start), nullptr));
|
||||||
|
|
||||||
|
TempVector<AstLocal*> vars(scratchLocal);
|
||||||
|
|
||||||
|
for (size_t i = 0; i < args.size(); ++i)
|
||||||
|
vars.push_back(pushLocal(args[i]));
|
||||||
|
|
||||||
|
return {self, copy(vars)};
|
||||||
|
}
|
||||||
|
|
||||||
// funcbody ::= `(' [parlist] `)' [`:' ReturnType] block end
|
// funcbody ::= `(' [parlist] `)' [`:' ReturnType] block end
|
||||||
// parlist ::= bindinglist [`,' `...'] | `...'
|
// parlist ::= bindinglist [`,' `...'] | `...'
|
||||||
std::pair<AstExprFunction*, AstLocal*> Parser::parseFunctionBody(
|
std::pair<AstExprFunction*, AstLocal*> Parser::parseFunctionBody(
|
||||||
@ -983,15 +1008,18 @@ std::pair<AstExprFunction*, AstLocal*> Parser::parseFunctionBody(
|
|||||||
|
|
||||||
TempVector<Binding> args(scratchBinding);
|
TempVector<Binding> args(scratchBinding);
|
||||||
|
|
||||||
std::optional<Location> vararg;
|
bool vararg = false;
|
||||||
|
Location varargLocation;
|
||||||
AstTypePack* varargAnnotation = nullptr;
|
AstTypePack* varargAnnotation = nullptr;
|
||||||
|
|
||||||
if (lexer.current().type != ')')
|
if (lexer.current().type != ')')
|
||||||
std::tie(vararg, varargAnnotation) = parseBindingList(args, /* allowDot3= */ true);
|
std::tie(vararg, varargLocation, varargAnnotation) = parseBindingList(args, /* allowDot3= */ true);
|
||||||
|
|
||||||
|
std::optional<Location> argLocation;
|
||||||
|
|
||||||
|
if (matchParen.type == Lexeme::Type('(') && lexer.current().type == Lexeme::Type(')'))
|
||||||
|
argLocation = Location(matchParen.position, lexer.current().location.end);
|
||||||
|
|
||||||
std::optional<Location> argLocation = matchParen.type == Lexeme::Type('(') && lexer.current().type == Lexeme::Type(')')
|
|
||||||
? std::make_optional(Location(matchParen.position, lexer.current().location.end))
|
|
||||||
: std::nullopt;
|
|
||||||
expectMatchAndConsume(')', matchParen, true);
|
expectMatchAndConsume(')', matchParen, true);
|
||||||
|
|
||||||
std::optional<AstTypeList> typelist = parseOptionalReturnTypeAnnotation();
|
std::optional<AstTypeList> typelist = parseOptionalReturnTypeAnnotation();
|
||||||
@ -1004,19 +1032,11 @@ std::pair<AstExprFunction*, AstLocal*> Parser::parseFunctionBody(
|
|||||||
unsigned int localsBegin = saveLocals();
|
unsigned int localsBegin = saveLocals();
|
||||||
|
|
||||||
Function fun;
|
Function fun;
|
||||||
fun.vararg = vararg.has_value();
|
fun.vararg = vararg;
|
||||||
|
|
||||||
functionStack.push_back(fun);
|
functionStack.emplace_back(fun);
|
||||||
|
|
||||||
AstLocal* self = nullptr;
|
auto [self, vars] = prepareFunctionArguments(start, hasself, args);
|
||||||
|
|
||||||
if (hasself)
|
|
||||||
self = pushLocal(Binding(Name(nameSelf, start), nullptr));
|
|
||||||
|
|
||||||
TempVector<AstLocal*> vars(scratchLocal);
|
|
||||||
|
|
||||||
for (size_t i = 0; i < args.size(); ++i)
|
|
||||||
vars.push_back(pushLocal(args[i]));
|
|
||||||
|
|
||||||
AstStatBlock* body = parseBlock();
|
AstStatBlock* body = parseBlock();
|
||||||
|
|
||||||
@ -1028,8 +1048,8 @@ std::pair<AstExprFunction*, AstLocal*> Parser::parseFunctionBody(
|
|||||||
|
|
||||||
bool hasEnd = expectMatchEndAndConsume(Lexeme::ReservedEnd, matchFunction);
|
bool hasEnd = expectMatchEndAndConsume(Lexeme::ReservedEnd, matchFunction);
|
||||||
|
|
||||||
return {allocator.alloc<AstExprFunction>(Location(start, end), generics, genericPacks, self, copy(vars), vararg, body, functionStack.size(),
|
return {allocator.alloc<AstExprFunction>(Location(start, end), generics, genericPacks, self, vars, vararg, varargLocation, body,
|
||||||
debugname, typelist, varargAnnotation, hasEnd, argLocation),
|
functionStack.size(), debugname, typelist, varargAnnotation, hasEnd, argLocation),
|
||||||
funLocal};
|
funLocal};
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1060,7 +1080,7 @@ Parser::Binding Parser::parseBinding()
|
|||||||
}
|
}
|
||||||
|
|
||||||
// bindinglist ::= (binding | `...') [`,' bindinglist]
|
// bindinglist ::= (binding | `...') [`,' bindinglist]
|
||||||
std::pair<std::optional<Location>, AstTypePack*> Parser::parseBindingList(TempVector<Binding>& result, bool allowDot3)
|
std::tuple<bool, Location, AstTypePack*> Parser::parseBindingList(TempVector<Binding>& result, bool allowDot3)
|
||||||
{
|
{
|
||||||
while (true)
|
while (true)
|
||||||
{
|
{
|
||||||
@ -1076,7 +1096,7 @@ std::pair<std::optional<Location>, AstTypePack*> Parser::parseBindingList(TempVe
|
|||||||
tailAnnotation = parseVariadicArgumentAnnotation();
|
tailAnnotation = parseVariadicArgumentAnnotation();
|
||||||
}
|
}
|
||||||
|
|
||||||
return {varargLocation, tailAnnotation};
|
return {true, varargLocation, tailAnnotation};
|
||||||
}
|
}
|
||||||
|
|
||||||
result.push_back(parseBinding());
|
result.push_back(parseBinding());
|
||||||
@ -1086,7 +1106,7 @@ std::pair<std::optional<Location>, AstTypePack*> Parser::parseBindingList(TempVe
|
|||||||
nextLexeme();
|
nextLexeme();
|
||||||
}
|
}
|
||||||
|
|
||||||
return {std::nullopt, nullptr};
|
return {false, Location(), nullptr};
|
||||||
}
|
}
|
||||||
|
|
||||||
AstType* Parser::parseOptionalTypeAnnotation()
|
AstType* Parser::parseOptionalTypeAnnotation()
|
||||||
|
@ -12,6 +12,7 @@ inline bool isFlagExperimental(const char* flag)
|
|||||||
// or critical bugs that are found after the code has been submitted.
|
// or critical bugs that are found after the code has been submitted.
|
||||||
static const char* kList[] = {
|
static const char* kList[] = {
|
||||||
"LuauLowerBoundsCalculation",
|
"LuauLowerBoundsCalculation",
|
||||||
|
"LuauInterpolatedStringBaseSupport",
|
||||||
nullptr, // makes sure we always have at least one entry
|
nullptr, // makes sure we always have at least one entry
|
||||||
};
|
};
|
||||||
|
|
||||||
|
@ -3043,6 +3043,18 @@ struct Compiler
|
|||||||
uint8_t valueReg = kInvalidReg;
|
uint8_t valueReg = kInvalidReg;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// This function analyzes assignments and marks assignment conflicts: cases when a variable is assigned on lhs
|
||||||
|
// but subsequently used on the rhs, assuming assignments are performed in order. Note that it's also possible
|
||||||
|
// for a variable to conflict on the lhs, if it's used in an lvalue expression after it's assigned.
|
||||||
|
// When conflicts are found, Assignment::conflictReg is allocated and that's where assignment is performed instead,
|
||||||
|
// until the final fixup in compileStatAssign. Assignment::valueReg is allocated by compileStatAssign as well.
|
||||||
|
//
|
||||||
|
// Per Lua manual, section 3.3.3 (Assignments), the proper assignment order is only guaranteed to hold for syntactic access:
|
||||||
|
//
|
||||||
|
// Note that this guarantee covers only accesses syntactically inside the assignment statement. If a function or a metamethod called
|
||||||
|
// during the assignment changes the value of a variable, Lua gives no guarantees about the order of that access.
|
||||||
|
//
|
||||||
|
// As such, we currently don't check if an assigned local is captured, which may mean it gets reassigned during a function call.
|
||||||
void resolveAssignConflicts(AstStat* stat, std::vector<Assignment>& vars, const AstArray<AstExpr*>& values)
|
void resolveAssignConflicts(AstStat* stat, std::vector<Assignment>& vars, const AstArray<AstExpr*>& values)
|
||||||
{
|
{
|
||||||
struct Visitor : AstVisitor
|
struct Visitor : AstVisitor
|
||||||
@ -3945,7 +3957,8 @@ void compileOrThrow(BytecodeBuilder& bytecode, const ParseResult& parseResult, c
|
|||||||
compiler.compileFunction(expr);
|
compiler.compileFunction(expr);
|
||||||
|
|
||||||
AstExprFunction main(root->location, /*generics= */ AstArray<AstGenericType>(), /*genericPacks= */ AstArray<AstGenericTypePack>(),
|
AstExprFunction main(root->location, /*generics= */ AstArray<AstGenericType>(), /*genericPacks= */ AstArray<AstGenericTypePack>(),
|
||||||
/* self= */ nullptr, AstArray<AstLocal*>(), /* vararg= */ Luau::Location(), root, /* functionDepth= */ 0, /* debugname= */ AstName());
|
/* self= */ nullptr, AstArray<AstLocal*>(), /* vararg= */ true, /* varargLocation= */ Luau::Location(), root, /* functionDepth= */ 0,
|
||||||
|
/* debugname= */ AstName());
|
||||||
uint32_t mainid = compiler.compileFunction(&main);
|
uint32_t mainid = compiler.compileFunction(&main);
|
||||||
|
|
||||||
const Compiler::Function* mainf = compiler.functions.find(&main);
|
const Compiler::Function* mainf = compiler.functions.find(&main);
|
||||||
|
8
Makefile
8
Makefile
@ -93,7 +93,7 @@ endif
|
|||||||
|
|
||||||
ifeq ($(config),fuzz)
|
ifeq ($(config),fuzz)
|
||||||
CXX=clang++ # our fuzzing infra relies on llvm fuzzer
|
CXX=clang++ # our fuzzing infra relies on llvm fuzzer
|
||||||
CXXFLAGS+=-fsanitize=address,fuzzer -Ibuild/libprotobuf-mutator -Ibuild/libprotobuf-mutator/external.protobuf/include -O2
|
CXXFLAGS+=-fsanitize=address,fuzzer -Ibuild/libprotobuf-mutator -O2
|
||||||
LDFLAGS+=-fsanitize=address,fuzzer
|
LDFLAGS+=-fsanitize=address,fuzzer
|
||||||
endif
|
endif
|
||||||
|
|
||||||
@ -115,7 +115,7 @@ $(FUZZ_OBJECTS): CXXFLAGS+=-std=c++17 -ICommon/include -IAst/include -ICompiler/
|
|||||||
|
|
||||||
$(TESTS_TARGET): LDFLAGS+=-lpthread
|
$(TESTS_TARGET): LDFLAGS+=-lpthread
|
||||||
$(REPL_CLI_TARGET): LDFLAGS+=-lpthread
|
$(REPL_CLI_TARGET): LDFLAGS+=-lpthread
|
||||||
fuzz-proto fuzz-prototest: LDFLAGS+=build/libprotobuf-mutator/src/libfuzzer/libprotobuf-mutator-libfuzzer.a build/libprotobuf-mutator/src/libprotobuf-mutator.a build/libprotobuf-mutator/external.protobuf/lib/libprotobuf.a
|
fuzz-proto fuzz-prototest: LDFLAGS+=build/libprotobuf-mutator/src/libfuzzer/libprotobuf-mutator-libfuzzer.a build/libprotobuf-mutator/src/libprotobuf-mutator.a -lprotobuf
|
||||||
|
|
||||||
# pseudo targets
|
# pseudo targets
|
||||||
.PHONY: all test clean coverage format luau-size aliases
|
.PHONY: all test clean coverage format luau-size aliases
|
||||||
@ -195,7 +195,7 @@ $(BUILD)/%.c.o: %.c
|
|||||||
|
|
||||||
# protobuf fuzzer setup
|
# protobuf fuzzer setup
|
||||||
fuzz/luau.pb.cpp: fuzz/luau.proto build/libprotobuf-mutator
|
fuzz/luau.pb.cpp: fuzz/luau.proto build/libprotobuf-mutator
|
||||||
cd fuzz && ../build/libprotobuf-mutator/external.protobuf/bin/protoc luau.proto --cpp_out=.
|
cd fuzz && protoc luau.proto --cpp_out=.
|
||||||
mv fuzz/luau.pb.cc fuzz/luau.pb.cpp
|
mv fuzz/luau.pb.cc fuzz/luau.pb.cpp
|
||||||
|
|
||||||
$(BUILD)/fuzz/proto.cpp.o: fuzz/luau.pb.cpp
|
$(BUILD)/fuzz/proto.cpp.o: fuzz/luau.pb.cpp
|
||||||
@ -203,7 +203,7 @@ $(BUILD)/fuzz/protoprint.cpp.o: fuzz/luau.pb.cpp
|
|||||||
|
|
||||||
build/libprotobuf-mutator:
|
build/libprotobuf-mutator:
|
||||||
git clone https://github.com/google/libprotobuf-mutator build/libprotobuf-mutator
|
git clone https://github.com/google/libprotobuf-mutator build/libprotobuf-mutator
|
||||||
CXX= cmake -S build/libprotobuf-mutator -B build/libprotobuf-mutator -D CMAKE_BUILD_TYPE=Release -D LIB_PROTO_MUTATOR_DOWNLOAD_PROTOBUF=ON -D LIB_PROTO_MUTATOR_TESTING=OFF
|
CXX= cmake -S build/libprotobuf-mutator -B build/libprotobuf-mutator -D CMAKE_BUILD_TYPE=Release -D LIB_PROTO_MUTATOR_TESTING=OFF
|
||||||
make -C build/libprotobuf-mutator -j8
|
make -C build/libprotobuf-mutator -j8
|
||||||
|
|
||||||
# picks up include dependencies for all object files
|
# picks up include dependencies for all object files
|
||||||
|
@ -24,16 +24,18 @@
|
|||||||
* The caller is expected to handle stack reservation (by using less than LUA_MINSTACK slots or by calling lua_checkstack).
|
* The caller is expected to handle stack reservation (by using less than LUA_MINSTACK slots or by calling lua_checkstack).
|
||||||
* To ensure this is handled correctly, use api_incr_top(L) when pushing values to the stack.
|
* To ensure this is handled correctly, use api_incr_top(L) when pushing values to the stack.
|
||||||
*
|
*
|
||||||
* Functions that push any collectable objects to the stack *should* call luaC_checkthreadsleep. Failure to do this can result
|
* Functions that push any collectable objects to the stack *should* call luaC_threadbarrier. Failure to do this can result
|
||||||
* in stack references that point to dead objects since sleeping threads don't get rescanned.
|
* in stack references that point to dead objects since black threads don't get rescanned.
|
||||||
*
|
*
|
||||||
* Functions that push newly created objects to the stack *should* call luaC_checkGC in addition to luaC_checkthreadsleep.
|
* Functions that push newly created objects to the stack *should* call luaC_checkGC in addition to luaC_threadbarrier.
|
||||||
* Failure to do this can result in OOM since GC may never run.
|
* Failure to do this can result in OOM since GC may never run.
|
||||||
*
|
*
|
||||||
* Note that luaC_checkGC may scan the thread and put it back to sleep; functions that call both before pushing objects must
|
* Note that luaC_checkGC may mark the thread and paint it black; functions that call both before pushing objects must
|
||||||
* therefore call luaC_checkGC before luaC_checkthreadsleep to guarantee the object is pushed to an awake thread.
|
* therefore call luaC_checkGC before luaC_threadbarrier to guarantee the object is pushed to a gray thread.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
LUAU_FASTFLAG(LuauSimplerUpval)
|
||||||
|
|
||||||
const char* lua_ident = "$Lua: Lua 5.1.4 Copyright (C) 1994-2008 Lua.org, PUC-Rio $\n"
|
const char* lua_ident = "$Lua: Lua 5.1.4 Copyright (C) 1994-2008 Lua.org, PUC-Rio $\n"
|
||||||
"$Authors: R. Ierusalimschy, L. H. de Figueiredo & W. Celes $\n"
|
"$Authors: R. Ierusalimschy, L. H. de Figueiredo & W. Celes $\n"
|
||||||
"$URL: www.lua.org $\n";
|
"$URL: www.lua.org $\n";
|
||||||
@ -152,7 +154,7 @@ void lua_xmove(lua_State* from, lua_State* to, int n)
|
|||||||
api_checknelems(from, n);
|
api_checknelems(from, n);
|
||||||
api_check(from, from->global == to->global);
|
api_check(from, from->global == to->global);
|
||||||
api_check(from, to->ci->top - to->top >= n);
|
api_check(from, to->ci->top - to->top >= n);
|
||||||
luaC_checkthreadsleep(to);
|
luaC_threadbarrier(to);
|
||||||
|
|
||||||
StkId ttop = to->top;
|
StkId ttop = to->top;
|
||||||
StkId ftop = from->top - n;
|
StkId ftop = from->top - n;
|
||||||
@ -168,7 +170,7 @@ void lua_xmove(lua_State* from, lua_State* to, int n)
|
|||||||
void lua_xpush(lua_State* from, lua_State* to, int idx)
|
void lua_xpush(lua_State* from, lua_State* to, int idx)
|
||||||
{
|
{
|
||||||
api_check(from, from->global == to->global);
|
api_check(from, from->global == to->global);
|
||||||
luaC_checkthreadsleep(to);
|
luaC_threadbarrier(to);
|
||||||
setobj2s(to, to->top, index2addr(from, idx));
|
setobj2s(to, to->top, index2addr(from, idx));
|
||||||
api_incr_top(to);
|
api_incr_top(to);
|
||||||
return;
|
return;
|
||||||
@ -177,7 +179,7 @@ void lua_xpush(lua_State* from, lua_State* to, int idx)
|
|||||||
lua_State* lua_newthread(lua_State* L)
|
lua_State* lua_newthread(lua_State* L)
|
||||||
{
|
{
|
||||||
luaC_checkGC(L);
|
luaC_checkGC(L);
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
lua_State* L1 = luaE_newthread(L);
|
lua_State* L1 = luaE_newthread(L);
|
||||||
setthvalue(L, L->top, L1);
|
setthvalue(L, L->top, L1);
|
||||||
api_incr_top(L);
|
api_incr_top(L);
|
||||||
@ -236,7 +238,7 @@ void lua_remove(lua_State* L, int idx)
|
|||||||
|
|
||||||
void lua_insert(lua_State* L, int idx)
|
void lua_insert(lua_State* L, int idx)
|
||||||
{
|
{
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
StkId p = index2addr(L, idx);
|
StkId p = index2addr(L, idx);
|
||||||
api_checkvalidindex(L, p);
|
api_checkvalidindex(L, p);
|
||||||
for (StkId q = L->top; q > p; q--)
|
for (StkId q = L->top; q > p; q--)
|
||||||
@ -248,7 +250,7 @@ void lua_insert(lua_State* L, int idx)
|
|||||||
void lua_replace(lua_State* L, int idx)
|
void lua_replace(lua_State* L, int idx)
|
||||||
{
|
{
|
||||||
api_checknelems(L, 1);
|
api_checknelems(L, 1);
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
StkId o = index2addr(L, idx);
|
StkId o = index2addr(L, idx);
|
||||||
api_checkvalidindex(L, o);
|
api_checkvalidindex(L, o);
|
||||||
if (idx == LUA_ENVIRONINDEX)
|
if (idx == LUA_ENVIRONINDEX)
|
||||||
@ -276,7 +278,7 @@ void lua_replace(lua_State* L, int idx)
|
|||||||
|
|
||||||
void lua_pushvalue(lua_State* L, int idx)
|
void lua_pushvalue(lua_State* L, int idx)
|
||||||
{
|
{
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
StkId o = index2addr(L, idx);
|
StkId o = index2addr(L, idx);
|
||||||
setobj2s(L, L->top, o);
|
setobj2s(L, L->top, o);
|
||||||
api_incr_top(L);
|
api_incr_top(L);
|
||||||
@ -427,7 +429,7 @@ const char* lua_tolstring(lua_State* L, int idx, size_t* len)
|
|||||||
StkId o = index2addr(L, idx);
|
StkId o = index2addr(L, idx);
|
||||||
if (!ttisstring(o))
|
if (!ttisstring(o))
|
||||||
{
|
{
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
if (!luaV_tostring(L, o))
|
if (!luaV_tostring(L, o))
|
||||||
{ // conversion failed?
|
{ // conversion failed?
|
||||||
if (len != NULL)
|
if (len != NULL)
|
||||||
@ -607,7 +609,7 @@ void lua_pushvector(lua_State* L, float x, float y, float z)
|
|||||||
void lua_pushlstring(lua_State* L, const char* s, size_t len)
|
void lua_pushlstring(lua_State* L, const char* s, size_t len)
|
||||||
{
|
{
|
||||||
luaC_checkGC(L);
|
luaC_checkGC(L);
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
setsvalue2s(L, L->top, luaS_newlstr(L, s, len));
|
setsvalue2s(L, L->top, luaS_newlstr(L, s, len));
|
||||||
api_incr_top(L);
|
api_incr_top(L);
|
||||||
return;
|
return;
|
||||||
@ -624,7 +626,7 @@ void lua_pushstring(lua_State* L, const char* s)
|
|||||||
const char* lua_pushvfstring(lua_State* L, const char* fmt, va_list argp)
|
const char* lua_pushvfstring(lua_State* L, const char* fmt, va_list argp)
|
||||||
{
|
{
|
||||||
luaC_checkGC(L);
|
luaC_checkGC(L);
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
const char* ret = luaO_pushvfstring(L, fmt, argp);
|
const char* ret = luaO_pushvfstring(L, fmt, argp);
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
@ -632,7 +634,7 @@ const char* lua_pushvfstring(lua_State* L, const char* fmt, va_list argp)
|
|||||||
const char* lua_pushfstringL(lua_State* L, const char* fmt, ...)
|
const char* lua_pushfstringL(lua_State* L, const char* fmt, ...)
|
||||||
{
|
{
|
||||||
luaC_checkGC(L);
|
luaC_checkGC(L);
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
va_list argp;
|
va_list argp;
|
||||||
va_start(argp, fmt);
|
va_start(argp, fmt);
|
||||||
const char* ret = luaO_pushvfstring(L, fmt, argp);
|
const char* ret = luaO_pushvfstring(L, fmt, argp);
|
||||||
@ -643,7 +645,7 @@ const char* lua_pushfstringL(lua_State* L, const char* fmt, ...)
|
|||||||
void lua_pushcclosurek(lua_State* L, lua_CFunction fn, const char* debugname, int nup, lua_Continuation cont)
|
void lua_pushcclosurek(lua_State* L, lua_CFunction fn, const char* debugname, int nup, lua_Continuation cont)
|
||||||
{
|
{
|
||||||
luaC_checkGC(L);
|
luaC_checkGC(L);
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
api_checknelems(L, nup);
|
api_checknelems(L, nup);
|
||||||
Closure* cl = luaF_newCclosure(L, nup, getcurrenv(L));
|
Closure* cl = luaF_newCclosure(L, nup, getcurrenv(L));
|
||||||
cl->c.f = fn;
|
cl->c.f = fn;
|
||||||
@ -674,7 +676,7 @@ void lua_pushlightuserdata(lua_State* L, void* p)
|
|||||||
|
|
||||||
int lua_pushthread(lua_State* L)
|
int lua_pushthread(lua_State* L)
|
||||||
{
|
{
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
setthvalue(L, L->top, L);
|
setthvalue(L, L->top, L);
|
||||||
api_incr_top(L);
|
api_incr_top(L);
|
||||||
return L->global->mainthread == L;
|
return L->global->mainthread == L;
|
||||||
@ -686,7 +688,7 @@ int lua_pushthread(lua_State* L)
|
|||||||
|
|
||||||
int lua_gettable(lua_State* L, int idx)
|
int lua_gettable(lua_State* L, int idx)
|
||||||
{
|
{
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
StkId t = index2addr(L, idx);
|
StkId t = index2addr(L, idx);
|
||||||
api_checkvalidindex(L, t);
|
api_checkvalidindex(L, t);
|
||||||
luaV_gettable(L, t, L->top - 1, L->top - 1);
|
luaV_gettable(L, t, L->top - 1, L->top - 1);
|
||||||
@ -695,7 +697,7 @@ int lua_gettable(lua_State* L, int idx)
|
|||||||
|
|
||||||
int lua_getfield(lua_State* L, int idx, const char* k)
|
int lua_getfield(lua_State* L, int idx, const char* k)
|
||||||
{
|
{
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
StkId t = index2addr(L, idx);
|
StkId t = index2addr(L, idx);
|
||||||
api_checkvalidindex(L, t);
|
api_checkvalidindex(L, t);
|
||||||
TValue key;
|
TValue key;
|
||||||
@ -707,7 +709,7 @@ int lua_getfield(lua_State* L, int idx, const char* k)
|
|||||||
|
|
||||||
int lua_rawgetfield(lua_State* L, int idx, const char* k)
|
int lua_rawgetfield(lua_State* L, int idx, const char* k)
|
||||||
{
|
{
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
StkId t = index2addr(L, idx);
|
StkId t = index2addr(L, idx);
|
||||||
api_check(L, ttistable(t));
|
api_check(L, ttistable(t));
|
||||||
TValue key;
|
TValue key;
|
||||||
@ -719,7 +721,7 @@ int lua_rawgetfield(lua_State* L, int idx, const char* k)
|
|||||||
|
|
||||||
int lua_rawget(lua_State* L, int idx)
|
int lua_rawget(lua_State* L, int idx)
|
||||||
{
|
{
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
StkId t = index2addr(L, idx);
|
StkId t = index2addr(L, idx);
|
||||||
api_check(L, ttistable(t));
|
api_check(L, ttistable(t));
|
||||||
setobj2s(L, L->top - 1, luaH_get(hvalue(t), L->top - 1));
|
setobj2s(L, L->top - 1, luaH_get(hvalue(t), L->top - 1));
|
||||||
@ -728,7 +730,7 @@ int lua_rawget(lua_State* L, int idx)
|
|||||||
|
|
||||||
int lua_rawgeti(lua_State* L, int idx, int n)
|
int lua_rawgeti(lua_State* L, int idx, int n)
|
||||||
{
|
{
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
StkId t = index2addr(L, idx);
|
StkId t = index2addr(L, idx);
|
||||||
api_check(L, ttistable(t));
|
api_check(L, ttistable(t));
|
||||||
setobj2s(L, L->top, luaH_getnum(hvalue(t), n));
|
setobj2s(L, L->top, luaH_getnum(hvalue(t), n));
|
||||||
@ -739,7 +741,7 @@ int lua_rawgeti(lua_State* L, int idx, int n)
|
|||||||
void lua_createtable(lua_State* L, int narray, int nrec)
|
void lua_createtable(lua_State* L, int narray, int nrec)
|
||||||
{
|
{
|
||||||
luaC_checkGC(L);
|
luaC_checkGC(L);
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
sethvalue(L, L->top, luaH_new(L, narray, nrec));
|
sethvalue(L, L->top, luaH_new(L, narray, nrec));
|
||||||
api_incr_top(L);
|
api_incr_top(L);
|
||||||
return;
|
return;
|
||||||
@ -775,7 +777,7 @@ void lua_setsafeenv(lua_State* L, int objindex, int enabled)
|
|||||||
|
|
||||||
int lua_getmetatable(lua_State* L, int objindex)
|
int lua_getmetatable(lua_State* L, int objindex)
|
||||||
{
|
{
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
Table* mt = NULL;
|
Table* mt = NULL;
|
||||||
const TValue* obj = index2addr(L, objindex);
|
const TValue* obj = index2addr(L, objindex);
|
||||||
switch (ttype(obj))
|
switch (ttype(obj))
|
||||||
@ -800,7 +802,7 @@ int lua_getmetatable(lua_State* L, int objindex)
|
|||||||
|
|
||||||
void lua_getfenv(lua_State* L, int idx)
|
void lua_getfenv(lua_State* L, int idx)
|
||||||
{
|
{
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
StkId o = index2addr(L, idx);
|
StkId o = index2addr(L, idx);
|
||||||
api_checkvalidindex(L, o);
|
api_checkvalidindex(L, o);
|
||||||
switch (ttype(o))
|
switch (ttype(o))
|
||||||
@ -1161,7 +1163,7 @@ l_noret lua_error(lua_State* L)
|
|||||||
|
|
||||||
int lua_next(lua_State* L, int idx)
|
int lua_next(lua_State* L, int idx)
|
||||||
{
|
{
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
StkId t = index2addr(L, idx);
|
StkId t = index2addr(L, idx);
|
||||||
api_check(L, ttistable(t));
|
api_check(L, ttistable(t));
|
||||||
int more = luaH_next(L, hvalue(t), L->top - 1);
|
int more = luaH_next(L, hvalue(t), L->top - 1);
|
||||||
@ -1180,13 +1182,13 @@ void lua_concat(lua_State* L, int n)
|
|||||||
if (n >= 2)
|
if (n >= 2)
|
||||||
{
|
{
|
||||||
luaC_checkGC(L);
|
luaC_checkGC(L);
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
luaV_concat(L, n, cast_int(L->top - L->base) - 1);
|
luaV_concat(L, n, cast_int(L->top - L->base) - 1);
|
||||||
L->top -= (n - 1);
|
L->top -= (n - 1);
|
||||||
}
|
}
|
||||||
else if (n == 0)
|
else if (n == 0)
|
||||||
{ // push empty string
|
{ // push empty string
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
setsvalue2s(L, L->top, luaS_newlstr(L, "", 0));
|
setsvalue2s(L, L->top, luaS_newlstr(L, "", 0));
|
||||||
api_incr_top(L);
|
api_incr_top(L);
|
||||||
}
|
}
|
||||||
@ -1198,7 +1200,7 @@ void* lua_newuserdatatagged(lua_State* L, size_t sz, int tag)
|
|||||||
{
|
{
|
||||||
api_check(L, unsigned(tag) < LUA_UTAG_LIMIT || tag == UTAG_PROXY);
|
api_check(L, unsigned(tag) < LUA_UTAG_LIMIT || tag == UTAG_PROXY);
|
||||||
luaC_checkGC(L);
|
luaC_checkGC(L);
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
Udata* u = luaU_newudata(L, sz, tag);
|
Udata* u = luaU_newudata(L, sz, tag);
|
||||||
setuvalue(L, L->top, u);
|
setuvalue(L, L->top, u);
|
||||||
api_incr_top(L);
|
api_incr_top(L);
|
||||||
@ -1208,7 +1210,7 @@ void* lua_newuserdatatagged(lua_State* L, size_t sz, int tag)
|
|||||||
void* lua_newuserdatadtor(lua_State* L, size_t sz, void (*dtor)(void*))
|
void* lua_newuserdatadtor(lua_State* L, size_t sz, void (*dtor)(void*))
|
||||||
{
|
{
|
||||||
luaC_checkGC(L);
|
luaC_checkGC(L);
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
// make sure sz + sizeof(dtor) doesn't overflow; luaU_newdata will reject SIZE_MAX correctly
|
// make sure sz + sizeof(dtor) doesn't overflow; luaU_newdata will reject SIZE_MAX correctly
|
||||||
size_t as = sz < SIZE_MAX - sizeof(dtor) ? sz + sizeof(dtor) : SIZE_MAX;
|
size_t as = sz < SIZE_MAX - sizeof(dtor) ? sz + sizeof(dtor) : SIZE_MAX;
|
||||||
Udata* u = luaU_newudata(L, as, UTAG_IDTOR);
|
Udata* u = luaU_newudata(L, as, UTAG_IDTOR);
|
||||||
@ -1244,7 +1246,7 @@ static const char* aux_upvalue(StkId fi, int n, TValue** val)
|
|||||||
|
|
||||||
const char* lua_getupvalue(lua_State* L, int funcindex, int n)
|
const char* lua_getupvalue(lua_State* L, int funcindex, int n)
|
||||||
{
|
{
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
TValue* val;
|
TValue* val;
|
||||||
const char* name = aux_upvalue(index2addr(L, funcindex), n, &val);
|
const char* name = aux_upvalue(index2addr(L, funcindex), n, &val);
|
||||||
if (name)
|
if (name)
|
||||||
@ -1266,7 +1268,8 @@ const char* lua_setupvalue(lua_State* L, int funcindex, int n)
|
|||||||
L->top--;
|
L->top--;
|
||||||
setobj(L, val, L->top);
|
setobj(L, val, L->top);
|
||||||
luaC_barrier(L, clvalue(fi), L->top);
|
luaC_barrier(L, clvalue(fi), L->top);
|
||||||
luaC_upvalbarrier(L, cast_to(UpVal*, NULL), val);
|
if (!FFlag::LuauSimplerUpval)
|
||||||
|
luaC_upvalbarrier(L, cast_to(UpVal*, NULL), val);
|
||||||
}
|
}
|
||||||
return name;
|
return name;
|
||||||
}
|
}
|
||||||
@ -1336,7 +1339,7 @@ void lua_setuserdatadtor(lua_State* L, int tag, void (*dtor)(lua_State*, void*))
|
|||||||
void lua_clonefunction(lua_State* L, int idx)
|
void lua_clonefunction(lua_State* L, int idx)
|
||||||
{
|
{
|
||||||
luaC_checkGC(L);
|
luaC_checkGC(L);
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
StkId p = index2addr(L, idx);
|
StkId p = index2addr(L, idx);
|
||||||
api_check(L, isLfunction(p));
|
api_check(L, isLfunction(p));
|
||||||
Closure* cl = clvalue(p);
|
Closure* cl = clvalue(p);
|
||||||
|
@ -12,8 +12,6 @@
|
|||||||
#include <string.h>
|
#include <string.h>
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
|
|
||||||
LUAU_FASTFLAGVARIABLE(LuauDebuggerBreakpointHitOnNextBestLine, false);
|
|
||||||
|
|
||||||
static const char* getfuncname(Closure* f);
|
static const char* getfuncname(Closure* f);
|
||||||
|
|
||||||
static int currentpc(lua_State* L, CallInfo* ci)
|
static int currentpc(lua_State* L, CallInfo* ci)
|
||||||
@ -44,13 +42,13 @@ int lua_getargument(lua_State* L, int level, int n)
|
|||||||
{
|
{
|
||||||
if (n <= fp->numparams)
|
if (n <= fp->numparams)
|
||||||
{
|
{
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
luaA_pushobject(L, ci->base + (n - 1));
|
luaA_pushobject(L, ci->base + (n - 1));
|
||||||
res = 1;
|
res = 1;
|
||||||
}
|
}
|
||||||
else if (fp->is_vararg && n < ci->base - ci->func)
|
else if (fp->is_vararg && n < ci->base - ci->func)
|
||||||
{
|
{
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
luaA_pushobject(L, ci->func + n);
|
luaA_pushobject(L, ci->func + n);
|
||||||
res = 1;
|
res = 1;
|
||||||
}
|
}
|
||||||
@ -69,7 +67,7 @@ const char* lua_getlocal(lua_State* L, int level, int n)
|
|||||||
const LocVar* var = fp ? luaF_getlocal(fp, n, currentpc(L, ci)) : NULL;
|
const LocVar* var = fp ? luaF_getlocal(fp, n, currentpc(L, ci)) : NULL;
|
||||||
if (var)
|
if (var)
|
||||||
{
|
{
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
luaA_pushobject(L, ci->base + var->reg);
|
luaA_pushobject(L, ci->base + var->reg);
|
||||||
}
|
}
|
||||||
const char* name = var ? getstr(var->varname) : NULL;
|
const char* name = var ? getstr(var->varname) : NULL;
|
||||||
@ -185,7 +183,7 @@ int lua_getinfo(lua_State* L, int level, const char* what, lua_Debug* ar)
|
|||||||
status = auxgetinfo(L, what, ar, f, ci);
|
status = auxgetinfo(L, what, ar, f, ci);
|
||||||
if (strchr(what, 'f'))
|
if (strchr(what, 'f'))
|
||||||
{
|
{
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
setclvalue(L, L->top, f);
|
setclvalue(L, L->top, f);
|
||||||
incr_top(L);
|
incr_top(L);
|
||||||
}
|
}
|
||||||
@ -437,29 +435,17 @@ static int getnextline(Proto* p, int line)
|
|||||||
|
|
||||||
int lua_breakpoint(lua_State* L, int funcindex, int line, int enabled)
|
int lua_breakpoint(lua_State* L, int funcindex, int line, int enabled)
|
||||||
{
|
{
|
||||||
int target = -1;
|
const TValue* func = luaA_toobject(L, funcindex);
|
||||||
|
api_check(L, ttisfunction(func) && !clvalue(func)->isC);
|
||||||
|
|
||||||
if (FFlag::LuauDebuggerBreakpointHitOnNextBestLine)
|
Proto* p = clvalue(func)->l.p;
|
||||||
|
// Find line number to add the breakpoint to.
|
||||||
|
int target = getnextline(p, line);
|
||||||
|
|
||||||
|
if (target != -1)
|
||||||
{
|
{
|
||||||
const TValue* func = luaA_toobject(L, funcindex);
|
// Add breakpoint on the exact line
|
||||||
api_check(L, ttisfunction(func) && !clvalue(func)->isC);
|
luaG_breakpoint(L, p, target, bool(enabled));
|
||||||
|
|
||||||
Proto* p = clvalue(func)->l.p;
|
|
||||||
// Find line number to add the breakpoint to.
|
|
||||||
target = getnextline(p, line);
|
|
||||||
|
|
||||||
if (target != -1)
|
|
||||||
{
|
|
||||||
// Add breakpoint on the exact line
|
|
||||||
luaG_breakpoint(L, p, target, bool(enabled));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
const TValue* func = luaA_toobject(L, funcindex);
|
|
||||||
api_check(L, ttisfunction(func) && !clvalue(func)->isC);
|
|
||||||
|
|
||||||
luaG_breakpoint(L, clvalue(func)->l.p, line, bool(enabled));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return target;
|
return target;
|
||||||
|
@ -158,7 +158,7 @@ l_noret luaD_throw(lua_State* L, int errcode)
|
|||||||
static void correctstack(lua_State* L, TValue* oldstack)
|
static void correctstack(lua_State* L, TValue* oldstack)
|
||||||
{
|
{
|
||||||
L->top = (L->top - oldstack) + L->stack;
|
L->top = (L->top - oldstack) + L->stack;
|
||||||
for (UpVal* up = L->openupval; up != NULL; up = up->u.l.threadnext)
|
for (UpVal* up = L->openupval; up != NULL; up = up->u.open.threadnext)
|
||||||
up->v = (up->v - oldstack) + L->stack;
|
up->v = (up->v - oldstack) + L->stack;
|
||||||
for (CallInfo* ci = L->base_ci; ci <= L->ci; ci++)
|
for (CallInfo* ci = L->base_ci; ci <= L->ci; ci++)
|
||||||
{
|
{
|
||||||
@ -245,7 +245,7 @@ void luaD_call(lua_State* L, StkId func, int nResults)
|
|||||||
|
|
||||||
int oldactive = luaC_threadactive(L);
|
int oldactive = luaC_threadactive(L);
|
||||||
l_setbit(L->stackstate, THREAD_ACTIVEBIT);
|
l_setbit(L->stackstate, THREAD_ACTIVEBIT);
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
|
|
||||||
luau_execute(L); // call it
|
luau_execute(L); // call it
|
||||||
|
|
||||||
@ -454,7 +454,7 @@ int lua_resume(lua_State* L, lua_State* from, int nargs)
|
|||||||
L->baseCcalls = ++L->nCcalls;
|
L->baseCcalls = ++L->nCcalls;
|
||||||
l_setbit(L->stackstate, THREAD_ACTIVEBIT);
|
l_setbit(L->stackstate, THREAD_ACTIVEBIT);
|
||||||
|
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
|
|
||||||
status = luaD_rawrunprotected(L, resume, L->top - nargs);
|
status = luaD_rawrunprotected(L, resume, L->top - nargs);
|
||||||
|
|
||||||
@ -483,7 +483,7 @@ int lua_resumeerror(lua_State* L, lua_State* from)
|
|||||||
L->baseCcalls = ++L->nCcalls;
|
L->baseCcalls = ++L->nCcalls;
|
||||||
l_setbit(L->stackstate, THREAD_ACTIVEBIT);
|
l_setbit(L->stackstate, THREAD_ACTIVEBIT);
|
||||||
|
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
|
|
||||||
status = LUA_ERRRUN;
|
status = LUA_ERRRUN;
|
||||||
|
|
||||||
|
127
VM/src/lfunc.cpp
127
VM/src/lfunc.cpp
@ -6,6 +6,9 @@
|
|||||||
#include "lmem.h"
|
#include "lmem.h"
|
||||||
#include "lgc.h"
|
#include "lgc.h"
|
||||||
|
|
||||||
|
LUAU_FASTFLAG(LuauSimplerUpval)
|
||||||
|
LUAU_FASTFLAG(LuauNoSleepBit)
|
||||||
|
|
||||||
Proto* luaF_newproto(lua_State* L)
|
Proto* luaF_newproto(lua_State* L)
|
||||||
{
|
{
|
||||||
Proto* f = luaM_newgco(L, Proto, sizeof(Proto), L->activememcat);
|
Proto* f = luaM_newgco(L, Proto, sizeof(Proto), L->activememcat);
|
||||||
@ -71,59 +74,76 @@ UpVal* luaF_findupval(lua_State* L, StkId level)
|
|||||||
UpVal* p;
|
UpVal* p;
|
||||||
while (*pp != NULL && (p = *pp)->v >= level)
|
while (*pp != NULL && (p = *pp)->v >= level)
|
||||||
{
|
{
|
||||||
LUAU_ASSERT(p->v != &p->u.value);
|
LUAU_ASSERT(!FFlag::LuauSimplerUpval || !isdead(g, obj2gco(p)));
|
||||||
|
LUAU_ASSERT(upisopen(p));
|
||||||
if (p->v == level)
|
if (p->v == level)
|
||||||
{ // found a corresponding upvalue?
|
{ // found a corresponding upvalue?
|
||||||
if (isdead(g, obj2gco(p))) // is it dead?
|
if (!FFlag::LuauSimplerUpval && isdead(g, obj2gco(p))) // is it dead?
|
||||||
changewhite(obj2gco(p)); // resurrect it
|
changewhite(obj2gco(p)); // resurrect it
|
||||||
return p;
|
return p;
|
||||||
}
|
}
|
||||||
|
|
||||||
pp = &p->u.l.threadnext;
|
pp = &p->u.open.threadnext;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
LUAU_ASSERT(luaC_threadactive(L));
|
||||||
|
LUAU_ASSERT(!luaC_threadsleeping(L));
|
||||||
|
LUAU_ASSERT(!FFlag::LuauNoSleepBit || !isblack(obj2gco(L))); // we don't use luaC_threadbarrier because active threads never turn black
|
||||||
|
|
||||||
UpVal* uv = luaM_newgco(L, UpVal, sizeof(UpVal), L->activememcat); // not found: create a new one
|
UpVal* uv = luaM_newgco(L, UpVal, sizeof(UpVal), L->activememcat); // not found: create a new one
|
||||||
uv->tt = LUA_TUPVAL;
|
luaC_init(L, uv, LUA_TUPVAL);
|
||||||
uv->marked = luaC_white(g);
|
uv->markedopen = 0;
|
||||||
uv->memcat = L->activememcat;
|
|
||||||
uv->v = level; // current value lives in the stack
|
uv->v = level; // current value lives in the stack
|
||||||
|
|
||||||
// chain the upvalue in the threads open upvalue list at the proper position
|
// chain the upvalue in the threads open upvalue list at the proper position
|
||||||
UpVal* next = *pp;
|
if (FFlag::LuauSimplerUpval)
|
||||||
uv->u.l.threadnext = next;
|
{
|
||||||
uv->u.l.threadprev = pp;
|
uv->u.open.threadnext = *pp;
|
||||||
if (next)
|
*pp = uv;
|
||||||
next->u.l.threadprev = &uv->u.l.threadnext;
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
UpVal* next = *pp;
|
||||||
|
uv->u.open.threadnext = next;
|
||||||
|
|
||||||
*pp = uv;
|
uv->u.open.threadprev = pp;
|
||||||
|
if (next)
|
||||||
|
next->u.open.threadprev = &uv->u.open.threadnext;
|
||||||
|
|
||||||
|
*pp = uv;
|
||||||
|
}
|
||||||
|
|
||||||
// double link the upvalue in the global open upvalue list
|
// double link the upvalue in the global open upvalue list
|
||||||
uv->u.l.prev = &g->uvhead;
|
uv->u.open.prev = &g->uvhead;
|
||||||
uv->u.l.next = g->uvhead.u.l.next;
|
uv->u.open.next = g->uvhead.u.open.next;
|
||||||
uv->u.l.next->u.l.prev = uv;
|
uv->u.open.next->u.open.prev = uv;
|
||||||
g->uvhead.u.l.next = uv;
|
g->uvhead.u.open.next = uv;
|
||||||
LUAU_ASSERT(uv->u.l.next->u.l.prev == uv && uv->u.l.prev->u.l.next == uv);
|
LUAU_ASSERT(uv->u.open.next->u.open.prev == uv && uv->u.open.prev->u.open.next == uv);
|
||||||
|
|
||||||
return uv;
|
return uv;
|
||||||
}
|
}
|
||||||
|
|
||||||
void luaF_unlinkupval(UpVal* uv)
|
void luaF_unlinkupval(UpVal* uv)
|
||||||
{
|
{
|
||||||
|
LUAU_ASSERT(!FFlag::LuauSimplerUpval);
|
||||||
|
|
||||||
// unlink upvalue from the global open upvalue list
|
// unlink upvalue from the global open upvalue list
|
||||||
LUAU_ASSERT(uv->u.l.next->u.l.prev == uv && uv->u.l.prev->u.l.next == uv);
|
LUAU_ASSERT(uv->u.open.next->u.open.prev == uv && uv->u.open.prev->u.open.next == uv);
|
||||||
uv->u.l.next->u.l.prev = uv->u.l.prev;
|
uv->u.open.next->u.open.prev = uv->u.open.prev;
|
||||||
uv->u.l.prev->u.l.next = uv->u.l.next;
|
uv->u.open.prev->u.open.next = uv->u.open.next;
|
||||||
|
|
||||||
// unlink upvalue from the thread open upvalue list
|
// unlink upvalue from the thread open upvalue list
|
||||||
*uv->u.l.threadprev = uv->u.l.threadnext;
|
*uv->u.open.threadprev = uv->u.open.threadnext;
|
||||||
|
|
||||||
if (UpVal* next = uv->u.l.threadnext)
|
if (UpVal* next = uv->u.open.threadnext)
|
||||||
next->u.l.threadprev = uv->u.l.threadprev;
|
next->u.open.threadprev = uv->u.open.threadprev;
|
||||||
}
|
}
|
||||||
|
|
||||||
void luaF_freeupval(lua_State* L, UpVal* uv, lua_Page* page)
|
void luaF_freeupval(lua_State* L, UpVal* uv, lua_Page* page)
|
||||||
{
|
{
|
||||||
if (uv->v != &uv->u.value) // is it open?
|
if (!FFlag::LuauSimplerUpval && uv->v != &uv->u.value) // is it open?
|
||||||
luaF_unlinkupval(uv); // remove from open list
|
luaF_unlinkupval(uv); // remove from open list
|
||||||
luaM_freegco(L, uv, sizeof(UpVal), uv->memcat, page); // free upvalue
|
luaM_freegco(L, uv, sizeof(UpVal), uv->memcat, page); // free upvalue
|
||||||
}
|
}
|
||||||
|
|
||||||
void luaF_close(lua_State* L, StkId level)
|
void luaF_close(lua_State* L, StkId level)
|
||||||
@ -133,26 +153,55 @@ void luaF_close(lua_State* L, StkId level)
|
|||||||
while (L->openupval != NULL && (uv = L->openupval)->v >= level)
|
while (L->openupval != NULL && (uv = L->openupval)->v >= level)
|
||||||
{
|
{
|
||||||
GCObject* o = obj2gco(uv);
|
GCObject* o = obj2gco(uv);
|
||||||
LUAU_ASSERT(!isblack(o) && uv->v != &uv->u.value);
|
LUAU_ASSERT(!isblack(o) && upisopen(uv));
|
||||||
|
|
||||||
// by removing the upvalue from global/thread open upvalue lists, L->openupval will be pointing to the next upvalue
|
if (FFlag::LuauSimplerUpval)
|
||||||
luaF_unlinkupval(uv);
|
|
||||||
|
|
||||||
if (isdead(g, o))
|
|
||||||
{
|
{
|
||||||
// close the upvalue without copying the dead data so that luaF_freeupval will not unlink again
|
LUAU_ASSERT(!isdead(g, o));
|
||||||
uv->v = &uv->u.value;
|
|
||||||
|
// unlink value *before* closing it since value storage overlaps
|
||||||
|
L->openupval = uv->u.open.threadnext;
|
||||||
|
|
||||||
|
luaF_closeupval(L, uv, /* dead= */ false);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
setobj(L, &uv->u.value, uv->v);
|
// by removing the upvalue from global/thread open upvalue lists, L->openupval will be pointing to the next upvalue
|
||||||
uv->v = &uv->u.value;
|
luaF_unlinkupval(uv);
|
||||||
// GC state of a new closed upvalue has to be initialized
|
|
||||||
luaC_initupval(L, uv);
|
if (isdead(g, o))
|
||||||
|
{
|
||||||
|
// close the upvalue without copying the dead data so that luaF_freeupval will not unlink again
|
||||||
|
uv->v = &uv->u.value;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
setobj(L, &uv->u.value, uv->v);
|
||||||
|
uv->v = &uv->u.value;
|
||||||
|
// GC state of a new closed upvalue has to be initialized
|
||||||
|
luaC_upvalclosed(L, uv);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void luaF_closeupval(lua_State* L, UpVal* uv, bool dead)
|
||||||
|
{
|
||||||
|
LUAU_ASSERT(FFlag::LuauSimplerUpval);
|
||||||
|
|
||||||
|
// unlink value from all lists *before* closing it since value storage overlaps
|
||||||
|
LUAU_ASSERT(uv->u.open.next->u.open.prev == uv && uv->u.open.prev->u.open.next == uv);
|
||||||
|
uv->u.open.next->u.open.prev = uv->u.open.prev;
|
||||||
|
uv->u.open.prev->u.open.next = uv->u.open.next;
|
||||||
|
|
||||||
|
if (dead)
|
||||||
|
return;
|
||||||
|
|
||||||
|
setobj(L, &uv->u.value, uv->v);
|
||||||
|
uv->v = &uv->u.value;
|
||||||
|
luaC_upvalclosed(L, uv);
|
||||||
|
}
|
||||||
|
|
||||||
void luaF_freeproto(lua_State* L, Proto* f, lua_Page* page)
|
void luaF_freeproto(lua_State* L, Proto* f, lua_Page* page)
|
||||||
{
|
{
|
||||||
luaM_freearray(L, f->code, f->sizecode, Instruction, f->memcat);
|
luaM_freearray(L, f->code, f->sizecode, Instruction, f->memcat);
|
||||||
|
@ -12,6 +12,7 @@ LUAI_FUNC Closure* luaF_newLclosure(lua_State* L, int nelems, Table* e, Proto* p
|
|||||||
LUAI_FUNC Closure* luaF_newCclosure(lua_State* L, int nelems, Table* e);
|
LUAI_FUNC Closure* luaF_newCclosure(lua_State* L, int nelems, Table* e);
|
||||||
LUAI_FUNC UpVal* luaF_findupval(lua_State* L, StkId level);
|
LUAI_FUNC UpVal* luaF_findupval(lua_State* L, StkId level);
|
||||||
LUAI_FUNC void luaF_close(lua_State* L, StkId level);
|
LUAI_FUNC void luaF_close(lua_State* L, StkId level);
|
||||||
|
LUAI_FUNC void luaF_closeupval(lua_State* L, UpVal* uv, bool dead);
|
||||||
LUAI_FUNC void luaF_freeproto(lua_State* L, Proto* f, struct lua_Page* page);
|
LUAI_FUNC void luaF_freeproto(lua_State* L, Proto* f, struct lua_Page* page);
|
||||||
LUAI_FUNC void luaF_freeclosure(lua_State* L, Closure* c, struct lua_Page* page);
|
LUAI_FUNC void luaF_freeclosure(lua_State* L, Closure* c, struct lua_Page* page);
|
||||||
LUAI_FUNC void luaF_unlinkupval(UpVal* uv);
|
LUAI_FUNC void luaF_unlinkupval(UpVal* uv);
|
||||||
|
259
VM/src/lgc.cpp
259
VM/src/lgc.cpp
@ -13,6 +13,117 @@
|
|||||||
|
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Luau uses an incremental non-generational non-moving mark&sweep garbage collector.
|
||||||
|
*
|
||||||
|
* The collector runs in three stages: mark, atomic and sweep. Mark and sweep are incremental and try to do a limited amount
|
||||||
|
* of work every GC step; atomic is ran once per the GC cycle and is indivisible. In either case, the work happens during GC
|
||||||
|
* steps that are "scheduled" by the GC pacing algorithm - the steps happen either from explicit calls to lua_gc, or after
|
||||||
|
* the mutator (aka application) allocates some amount of memory, which is known as "GC assist". In either case, GC steps
|
||||||
|
* can't happen concurrently with other access to VM state.
|
||||||
|
*
|
||||||
|
* Current GC stage is stored in global_State::gcstate, and has two additional stages for pause and second-phase mark, explained below.
|
||||||
|
*
|
||||||
|
* GC pacer is an algorithm that tries to ensure that GC can always catch up to the application allocating garbage, but do this
|
||||||
|
* with minimal amount of effort. To configure the pacer Luau provides control over three variables: GC goal, defined as the
|
||||||
|
* target heap size during atomic phase in relation to live heap size (e.g. 200% goal means the heap's worst case size is double
|
||||||
|
* the total size of alive objects), step size (how many kilobytes should the application allocate for GC step to trigger), and
|
||||||
|
* GC multiplier (how much should the GC try to mark relative to how much the application allocated). It's critical that step
|
||||||
|
* multiplier is significantly above 1, as this is what allows the GC to catch up to the application's allocation rate, and
|
||||||
|
* GC goal and GC multiplier are linked in subtle ways, described in lua.h comments for LUA_GCSETGOAL.
|
||||||
|
*
|
||||||
|
* During mark, GC tries to identify all reachable objects and mark them as reachable, while keeping unreachable objects unmarked.
|
||||||
|
* During sweep, GC tries to sweep all objects that were not reachable at the end of mark. The atomic phase is needed to ensure
|
||||||
|
* that all pending marking has completed and all objects that are still marked as unreachable are, in fact, unreachable.
|
||||||
|
*
|
||||||
|
* Notably, during mark GC doesn't free any objects, and so the heap size constantly grows; during sweep, GC doesn't do any marking
|
||||||
|
* work, so it can't immediately free objects that became unreachable after sweeping started.
|
||||||
|
*
|
||||||
|
* Every collectable object has one of three colors at any given point in time: white, gray or black. This coloring scheme
|
||||||
|
* is necessary to implement incremental marking: white objects have not been marked and may be unreachable, black objects
|
||||||
|
* have been marked and will not be marked again if they stay black, and gray objects have been marked but may contain unmarked
|
||||||
|
* references.
|
||||||
|
*
|
||||||
|
* Objects are allocated as white; however, during sweep, we need to differentiate between objects that remained white in the mark
|
||||||
|
* phase (these are not reachable and can be freed) and objects that were allocated after the mark phase ended. Because of this, the
|
||||||
|
* colors are encoded using three bits inside GCheader::marked: white0, white1 and black (so technically we use a four-color scheme:
|
||||||
|
* any object can be white0, white1, gray or black). All bits are exclusive, and gray objects have all three bits unset. This allows
|
||||||
|
* us to have the "current" white bit, which is flipped during atomic stage - during sweeping, objects that have the white color from
|
||||||
|
* the previous mark may be deleted, and all other objects may or may not be reachable, and will be changed to the current white color,
|
||||||
|
* so that the next mark can start coloring objects from scratch again.
|
||||||
|
*
|
||||||
|
* Crucially, the coloring scheme comes with what's known as a tri-color invariant: a black object may never point to a white object.
|
||||||
|
*
|
||||||
|
* At the end of atomic stage, the expectation is that there are no gray objects anymore, which means all objects are either black
|
||||||
|
* (reachable) or white (unreachable = dead). Tri-color invariant is maintained throughout mark and atomic phase. To uphold this
|
||||||
|
* invariant, every modification of an object needs to check if the object is black and the new referent is white; if so, we
|
||||||
|
* need to either mark the referent, making it non-white (known as a forward barrier), or mark the object as gray and queue it
|
||||||
|
* for additional marking (known as a backward barrier).
|
||||||
|
*
|
||||||
|
* Luau uses both types of barriers. Forward barriers advance GC progress, since they don't create new outstanding work for GC,
|
||||||
|
* but they may be expensive when an object is modified many times in succession. Backward barriers are cheaper, as they defer
|
||||||
|
* most of the work until "later", but they require queueing the object for a rescan which isn't always possible. Table writes usually
|
||||||
|
* use backward barriers (but switch to forward barriers during second-phase mark), whereas upvalue writes and setmetatable use forward
|
||||||
|
* barriers.
|
||||||
|
*
|
||||||
|
* Since marking is incremental, it needs a way to track progress, which is implemented as a gray set: at any point, objects that
|
||||||
|
* are gray need to mark their white references, objects that are black have no pending work, and objects that are white have not yet
|
||||||
|
* been reached. Once the gray set is empty, the work completes; as such, incremental marking is as simple as removing an object from
|
||||||
|
* the gray set, and turning it to black (which requires turning all its white references to gray). The gray set is implemented as
|
||||||
|
* an intrusive singly linked list, using `gclist` field in multiple objects (functions, tables, threads and protos). When an object
|
||||||
|
* doesn't have gclist field, the marking of that object needs to be "immediate", changing the colors of all references in one go.
|
||||||
|
*
|
||||||
|
* When a black object is modified, it needs to become gray again. Objects like this are placed on a separate `grayagain` list by a
|
||||||
|
* barrier - this is important because it allows us to have a mark stage that terminates when the gray set is empty even if the mutator
|
||||||
|
* is constantly changing existing objects to gray. After mark stage finishes traversing `gray` list, we copy `grayagain` list to `gray`
|
||||||
|
* once and incrementally mark it again. During this phase of marking, we may get more objects marked as `grayagain`, so after we finish
|
||||||
|
* emptying out the `gray` list the second time, we finish the mark stage and do final marking of `grayagain` during atomic phase.
|
||||||
|
* GC works correctly without this second-phase mark (called GCSpropagateagain), but it reduces the time spent during atomic phase.
|
||||||
|
*
|
||||||
|
* Sweeping is also incremental, but instead of working at a granularity of an object, it works at a granularity of a page: all GC
|
||||||
|
* objects are allocated in special pages (see lmem.cpp for details), and sweeper traverses all objects in one page in one incremental
|
||||||
|
* step, freeing objects that aren't reachable (old white), and recoloring all other objects with the new white to prepare them for next
|
||||||
|
* mark. During sweeping we don't need to maintain the GC invariant, because our goal is to paint all objects with current white -
|
||||||
|
* however, some barriers will still trigger (because some reachable objects are still black as sweeping didn't get to them yet), and
|
||||||
|
* some barriers will proactively mark black objects as white to avoid extra barriers from triggering excessively.
|
||||||
|
*
|
||||||
|
* Most references that GC deals with are strong, and as such they fit neatly into the incremental marking scheme. Some, however, are
|
||||||
|
* weak - notably, tables can be marked as having weak keys/values (using __mode metafield). During incremental marking, we don't know
|
||||||
|
* for certain if a given object is alive - if it's marked as black, it definitely was reachable during marking, but if it's marked as
|
||||||
|
* white, we don't know if it's actually unreachable. Because of this, we need to defer weak table handling to the atomic phase; after
|
||||||
|
* all objects are marked, we traverse all weak tables (that are linked into special weak table lists using `gclist` during marking),
|
||||||
|
* and remove all entries that have white keys or values. If keys or values are strong, they are marked normally.
|
||||||
|
*
|
||||||
|
* The simplified scheme described above isn't fully accurate because of threads, upvalues and strings.
|
||||||
|
*
|
||||||
|
* Strings are semantically black (they are initially white, and when the mark stage reaches a string, it changes its color and never
|
||||||
|
* touches the object again), but they are technically marked as gray - the black bit is never set on a string object. This behavior
|
||||||
|
* is inherited from Lua 5.1 GC, but doesn't have a clear rationale - effectively, strings are marked as gray but are never part of
|
||||||
|
* a gray list.
|
||||||
|
*
|
||||||
|
* Threads are hard to deal with because for them to fit into the white-gray-black scheme, writes to thread stacks need to have barriers
|
||||||
|
* that turn the thread from black (already scanned) to gray - but this is very expensive because stack writes are very common. To
|
||||||
|
* get around this problem, threads have an "active" state which means that a thread is actively executing code. When GC reaches an active
|
||||||
|
* thread, it keeps it as gray, and rescans it during atomic phase. When a thread is inactive, GC instead paints the thread black. All
|
||||||
|
* API calls that can write to thread stacks outside of execution (which implies active) uses a thread barrier that checks if the thread is
|
||||||
|
* black, and if it is it marks it as gray and puts it on a gray list to be rescanned during atomic phase.
|
||||||
|
*
|
||||||
|
* NOTE: The above is only true when LuauNoSleepBit is enabled.
|
||||||
|
*
|
||||||
|
* Upvalues are special objects that can be closed, in which case they contain the value (acting as a reference cell) and can be dealt
|
||||||
|
* with using the regular algorithm, or open, in which case they refer to a stack slot in some other thread. These are difficult to deal
|
||||||
|
* with because the stack writes are not monitored. Because of this open upvalues are treated in a somewhat special way: they are never marked
|
||||||
|
* as black (doing so would violate the GC invariant), and they are kept in a special global list (global_State::uvhead) which is traversed
|
||||||
|
* during atomic phase. This is needed because an open upvalue might point to a stack location in a dead thread that never marked the stack
|
||||||
|
* slot - upvalues like this are identified since they don't have `markedopen` bit set during thread traversal and closed in `clearupvals`.
|
||||||
|
*
|
||||||
|
* NOTE: The above is only true when LuauSimplerUpval is enabled.
|
||||||
|
*/
|
||||||
|
|
||||||
|
LUAU_FASTFLAGVARIABLE(LuauSimplerUpval, false)
|
||||||
|
LUAU_FASTFLAGVARIABLE(LuauNoSleepBit, false)
|
||||||
|
LUAU_FASTFLAGVARIABLE(LuauEagerShrink, false)
|
||||||
|
|
||||||
#define GC_SWEEPPAGESTEPCOST 16
|
#define GC_SWEEPPAGESTEPCOST 16
|
||||||
|
|
||||||
#define GC_INTERRUPT(state) \
|
#define GC_INTERRUPT(state) \
|
||||||
@ -150,8 +261,8 @@ static void reallymarkobject(global_State* g, GCObject* o)
|
|||||||
{
|
{
|
||||||
UpVal* uv = gco2uv(o);
|
UpVal* uv = gco2uv(o);
|
||||||
markvalue(g, uv->v);
|
markvalue(g, uv->v);
|
||||||
if (uv->v == &uv->u.value) // closed?
|
if (!upisopen(uv)) // closed?
|
||||||
gray2black(o); // open upvalues are never black
|
gray2black(o); // open upvalues are never black
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
case LUA_TFUNCTION:
|
case LUA_TFUNCTION:
|
||||||
@ -289,22 +400,34 @@ static void traverseclosure(global_State* g, Closure* cl)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static void traversestack(global_State* g, lua_State* l, bool clearstack)
|
static void traversestack(global_State* g, lua_State* l)
|
||||||
{
|
{
|
||||||
markobject(g, l->gt);
|
markobject(g, l->gt);
|
||||||
if (l->namecall)
|
if (l->namecall)
|
||||||
stringmark(l->namecall);
|
stringmark(l->namecall);
|
||||||
for (StkId o = l->stack; o < l->top; o++)
|
for (StkId o = l->stack; o < l->top; o++)
|
||||||
markvalue(g, o);
|
markvalue(g, o);
|
||||||
// final traversal?
|
if (FFlag::LuauSimplerUpval)
|
||||||
if (g->gcstate == GCSatomic || clearstack)
|
|
||||||
{
|
{
|
||||||
StkId stack_end = l->stack + l->stacksize;
|
for (UpVal* uv = l->openupval; uv; uv = uv->u.open.threadnext)
|
||||||
for (StkId o = l->top; o < stack_end; o++) // clear not-marked stack slice
|
{
|
||||||
setnilvalue(o);
|
LUAU_ASSERT(upisopen(uv));
|
||||||
|
uv->markedopen = 1;
|
||||||
|
markobject(g, uv);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static void clearstack(lua_State* l)
|
||||||
|
{
|
||||||
|
StkId stack_end = l->stack + l->stacksize;
|
||||||
|
for (StkId o = l->top; o < stack_end; o++) // clear not-marked stack slice
|
||||||
|
setnilvalue(o);
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: pull function definition here when FFlag::LuauEagerShrink is removed
|
||||||
|
static void shrinkstack(lua_State* L);
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** traverse one gray object, turning it to black.
|
** traverse one gray object, turning it to black.
|
||||||
** Returns `quantity' traversed.
|
** Returns `quantity' traversed.
|
||||||
@ -338,14 +461,17 @@ static size_t propagatemark(global_State* g)
|
|||||||
|
|
||||||
LUAU_ASSERT(!luaC_threadsleeping(th));
|
LUAU_ASSERT(!luaC_threadsleeping(th));
|
||||||
|
|
||||||
// threads that are executing and the main thread are not deactivated
|
// threads that are executing and the main thread remain gray
|
||||||
bool active = luaC_threadactive(th) || th == th->global->mainthread;
|
bool active = luaC_threadactive(th) || th == th->global->mainthread;
|
||||||
|
|
||||||
|
// TODO: Refactor this logic after LuauNoSleepBit is removed
|
||||||
if (!active && g->gcstate == GCSpropagate)
|
if (!active && g->gcstate == GCSpropagate)
|
||||||
{
|
{
|
||||||
traversestack(g, th, /* clearstack= */ true);
|
traversestack(g, th);
|
||||||
|
clearstack(th);
|
||||||
|
|
||||||
l_setbit(th->stackstate, THREAD_SLEEPINGBIT);
|
if (!FFlag::LuauNoSleepBit)
|
||||||
|
l_setbit(th->stackstate, THREAD_SLEEPINGBIT);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
@ -354,9 +480,17 @@ static size_t propagatemark(global_State* g)
|
|||||||
|
|
||||||
black2gray(o);
|
black2gray(o);
|
||||||
|
|
||||||
traversestack(g, th, /* clearstack= */ false);
|
traversestack(g, th);
|
||||||
|
|
||||||
|
// final traversal?
|
||||||
|
if (g->gcstate == GCSatomic)
|
||||||
|
clearstack(th);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// we could shrink stack at any time but we opt to skip it during atomic since it's redundant to do that more than once per cycle
|
||||||
|
if (FFlag::LuauEagerShrink && g->gcstate != GCSatomic)
|
||||||
|
shrinkstack(th);
|
||||||
|
|
||||||
return sizeof(lua_State) + sizeof(TValue) * th->stacksize + sizeof(CallInfo) * th->size_ci;
|
return sizeof(lua_State) + sizeof(TValue) * th->stacksize + sizeof(CallInfo) * th->size_ci;
|
||||||
}
|
}
|
||||||
case LUA_TPROTO:
|
case LUA_TPROTO:
|
||||||
@ -537,7 +671,7 @@ static bool deletegco(void* context, lua_Page* page, GCObject* gco)
|
|||||||
// we are in the process of deleting everything
|
// we are in the process of deleting everything
|
||||||
// threads with open upvalues will attempt to close them all on removal
|
// threads with open upvalues will attempt to close them all on removal
|
||||||
// but those upvalues might point to stack values that were already deleted
|
// but those upvalues might point to stack values that were already deleted
|
||||||
if (gco->gch.tt == LUA_TTHREAD)
|
if (!FFlag::LuauSimplerUpval && gco->gch.tt == LUA_TTHREAD)
|
||||||
{
|
{
|
||||||
lua_State* th = gco2th(gco);
|
lua_State* th = gco2th(gco);
|
||||||
|
|
||||||
@ -595,13 +729,53 @@ static void markroot(lua_State* L)
|
|||||||
static size_t remarkupvals(global_State* g)
|
static size_t remarkupvals(global_State* g)
|
||||||
{
|
{
|
||||||
size_t work = 0;
|
size_t work = 0;
|
||||||
for (UpVal* uv = g->uvhead.u.l.next; uv != &g->uvhead; uv = uv->u.l.next)
|
|
||||||
|
for (UpVal* uv = g->uvhead.u.open.next; uv != &g->uvhead; uv = uv->u.open.next)
|
||||||
{
|
{
|
||||||
work += sizeof(UpVal);
|
work += sizeof(UpVal);
|
||||||
LUAU_ASSERT(uv->u.l.next->u.l.prev == uv && uv->u.l.prev->u.l.next == uv);
|
|
||||||
|
LUAU_ASSERT(upisopen(uv));
|
||||||
|
LUAU_ASSERT(uv->u.open.next->u.open.prev == uv && uv->u.open.prev->u.open.next == uv);
|
||||||
|
LUAU_ASSERT(!isblack(obj2gco(uv))); // open upvalues are never black
|
||||||
|
|
||||||
if (isgray(obj2gco(uv)))
|
if (isgray(obj2gco(uv)))
|
||||||
markvalue(g, uv->v);
|
markvalue(g, uv->v);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return work;
|
||||||
|
}
|
||||||
|
|
||||||
|
static size_t clearupvals(lua_State* L)
|
||||||
|
{
|
||||||
|
global_State* g = L->global;
|
||||||
|
|
||||||
|
size_t work = 0;
|
||||||
|
|
||||||
|
for (UpVal* uv = g->uvhead.u.open.next; uv != &g->uvhead;)
|
||||||
|
{
|
||||||
|
work += sizeof(UpVal);
|
||||||
|
|
||||||
|
LUAU_ASSERT(upisopen(uv));
|
||||||
|
LUAU_ASSERT(uv->u.open.next->u.open.prev == uv && uv->u.open.prev->u.open.next == uv);
|
||||||
|
LUAU_ASSERT(!isblack(obj2gco(uv))); // open upvalues are never black
|
||||||
|
LUAU_ASSERT(iswhite(obj2gco(uv)) || !iscollectable(uv->v) || !iswhite(gcvalue(uv->v)));
|
||||||
|
|
||||||
|
if (uv->markedopen)
|
||||||
|
{
|
||||||
|
// upvalue is still open (belongs to alive thread)
|
||||||
|
LUAU_ASSERT(isgray(obj2gco(uv)));
|
||||||
|
uv->markedopen = 0; // for next cycle
|
||||||
|
uv = uv->u.open.next;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// upvalue is either dead, or alive but the thread is dead; unlink and close
|
||||||
|
UpVal* next = uv->u.open.next;
|
||||||
|
luaF_closeupval(L, uv, /* dead= */ iswhite(obj2gco(uv)));
|
||||||
|
uv = next;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return work;
|
return work;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -654,6 +828,16 @@ static size_t atomic(lua_State* L)
|
|||||||
g->gcmetrics.currcycle.atomictimeclear += recordGcDeltaTime(currts);
|
g->gcmetrics.currcycle.atomictimeclear += recordGcDeltaTime(currts);
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
if (FFlag::LuauSimplerUpval)
|
||||||
|
{
|
||||||
|
// close orphaned live upvalues of dead threads and clear dead upvalues
|
||||||
|
work += clearupvals(L);
|
||||||
|
|
||||||
|
#ifdef LUAI_GCMETRICS
|
||||||
|
g->gcmetrics.currcycle.atomictimeupval += recordGcDeltaTime(currts);
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
// flip current white
|
// flip current white
|
||||||
g->currentwhite = cast_byte(otherwhite(g));
|
g->currentwhite = cast_byte(otherwhite(g));
|
||||||
g->sweepgcopage = g->allgcopages;
|
g->sweepgcopage = g->allgcopages;
|
||||||
@ -677,8 +861,11 @@ static bool sweepgco(lua_State* L, lua_Page* page, GCObject* gco)
|
|||||||
|
|
||||||
if (alive)
|
if (alive)
|
||||||
{
|
{
|
||||||
resetbit(th->stackstate, THREAD_SLEEPINGBIT);
|
if (!FFlag::LuauNoSleepBit)
|
||||||
shrinkstack(th);
|
resetbit(th->stackstate, THREAD_SLEEPINGBIT);
|
||||||
|
|
||||||
|
if (!FFlag::LuauEagerShrink)
|
||||||
|
shrinkstack(th);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -945,7 +1132,7 @@ void luaC_fullgc(lua_State* L)
|
|||||||
startGcCycleMetrics(g);
|
startGcCycleMetrics(g);
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
if (g->gcstate <= GCSatomic)
|
if (FFlag::LuauSimplerUpval ? keepinvariant(g) : g->gcstate <= GCSatomic)
|
||||||
{
|
{
|
||||||
// reset sweep marks to sweep all elements (returning them to white)
|
// reset sweep marks to sweep all elements (returning them to white)
|
||||||
g->sweepgcopage = g->allgcopages;
|
g->sweepgcopage = g->allgcopages;
|
||||||
@ -955,7 +1142,7 @@ void luaC_fullgc(lua_State* L)
|
|||||||
g->weak = NULL;
|
g->weak = NULL;
|
||||||
g->gcstate = GCSsweep;
|
g->gcstate = GCSsweep;
|
||||||
}
|
}
|
||||||
LUAU_ASSERT(g->gcstate == GCSsweep);
|
LUAU_ASSERT(g->gcstate == GCSpause || g->gcstate == GCSsweep);
|
||||||
// finish any pending sweep phase
|
// finish any pending sweep phase
|
||||||
while (g->gcstate != GCSpause)
|
while (g->gcstate != GCSpause)
|
||||||
{
|
{
|
||||||
@ -963,6 +1150,16 @@ void luaC_fullgc(lua_State* L)
|
|||||||
gcstep(L, SIZE_MAX);
|
gcstep(L, SIZE_MAX);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (FFlag::LuauSimplerUpval)
|
||||||
|
{
|
||||||
|
// clear markedopen bits for all open upvalues; these might be stuck from half-finished mark prior to full gc
|
||||||
|
for (UpVal* uv = g->uvhead.u.open.next; uv != &g->uvhead; uv = uv->u.open.next)
|
||||||
|
{
|
||||||
|
LUAU_ASSERT(upisopen(uv));
|
||||||
|
uv->markedopen = 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#ifdef LUAI_GCMETRICS
|
#ifdef LUAI_GCMETRICS
|
||||||
finishGcCycleMetrics(g);
|
finishGcCycleMetrics(g);
|
||||||
startGcCycleMetrics(g);
|
startGcCycleMetrics(g);
|
||||||
@ -999,6 +1196,7 @@ void luaC_fullgc(lua_State* L)
|
|||||||
|
|
||||||
void luaC_barrierupval(lua_State* L, GCObject* v)
|
void luaC_barrierupval(lua_State* L, GCObject* v)
|
||||||
{
|
{
|
||||||
|
LUAU_ASSERT(!FFlag::LuauSimplerUpval);
|
||||||
global_State* g = L->global;
|
global_State* g = L->global;
|
||||||
LUAU_ASSERT(iswhite(v) && !isdead(g, v));
|
LUAU_ASSERT(iswhite(v) && !isdead(g, v));
|
||||||
|
|
||||||
@ -1038,30 +1236,24 @@ void luaC_barriertable(lua_State* L, Table* t, GCObject* v)
|
|||||||
g->grayagain = o;
|
g->grayagain = o;
|
||||||
}
|
}
|
||||||
|
|
||||||
void luaC_barrierback(lua_State* L, Table* t)
|
void luaC_barrierback(lua_State* L, GCObject* o, GCObject** gclist)
|
||||||
{
|
{
|
||||||
global_State* g = L->global;
|
global_State* g = L->global;
|
||||||
GCObject* o = obj2gco(t);
|
|
||||||
LUAU_ASSERT(isblack(o) && !isdead(g, o));
|
LUAU_ASSERT(isblack(o) && !isdead(g, o));
|
||||||
LUAU_ASSERT(g->gcstate != GCSpause);
|
LUAU_ASSERT(g->gcstate != GCSpause);
|
||||||
black2gray(o); // make table gray (again)
|
|
||||||
t->gclist = g->grayagain;
|
black2gray(o); // make object gray (again)
|
||||||
|
*gclist = g->grayagain;
|
||||||
g->grayagain = o;
|
g->grayagain = o;
|
||||||
}
|
}
|
||||||
|
|
||||||
void luaC_initobj(lua_State* L, GCObject* o, uint8_t tt)
|
void luaC_upvalclosed(lua_State* L, UpVal* uv)
|
||||||
{
|
|
||||||
global_State* g = L->global;
|
|
||||||
o->gch.marked = luaC_white(g);
|
|
||||||
o->gch.tt = tt;
|
|
||||||
o->gch.memcat = L->activememcat;
|
|
||||||
}
|
|
||||||
|
|
||||||
void luaC_initupval(lua_State* L, UpVal* uv)
|
|
||||||
{
|
{
|
||||||
global_State* g = L->global;
|
global_State* g = L->global;
|
||||||
GCObject* o = obj2gco(uv);
|
GCObject* o = obj2gco(uv);
|
||||||
|
|
||||||
|
LUAU_ASSERT(!upisopen(uv)); // upvalue was closed but needs GC state fixup
|
||||||
|
|
||||||
if (isgray(o))
|
if (isgray(o))
|
||||||
{
|
{
|
||||||
if (keepinvariant(g))
|
if (keepinvariant(g))
|
||||||
@ -1105,6 +1297,7 @@ int64_t luaC_allocationrate(lua_State* L)
|
|||||||
|
|
||||||
void luaC_wakethread(lua_State* L)
|
void luaC_wakethread(lua_State* L)
|
||||||
{
|
{
|
||||||
|
LUAU_ASSERT(!FFlag::LuauNoSleepBit);
|
||||||
if (!luaC_threadsleeping(L))
|
if (!luaC_threadsleeping(L))
|
||||||
return;
|
return;
|
||||||
|
|
||||||
@ -1116,6 +1309,8 @@ void luaC_wakethread(lua_State* L)
|
|||||||
{
|
{
|
||||||
GCObject* o = obj2gco(L);
|
GCObject* o = obj2gco(L);
|
||||||
|
|
||||||
|
LUAU_ASSERT(isblack(o));
|
||||||
|
|
||||||
L->gclist = g->grayagain;
|
L->gclist = g->grayagain;
|
||||||
g->grayagain = o;
|
g->grayagain = o;
|
||||||
|
|
||||||
|
31
VM/src/lgc.h
31
VM/src/lgc.h
@ -6,6 +6,8 @@
|
|||||||
#include "lobject.h"
|
#include "lobject.h"
|
||||||
#include "lstate.h"
|
#include "lstate.h"
|
||||||
|
|
||||||
|
LUAU_FASTFLAG(LuauNoSleepBit)
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** Default settings for GC tunables (settable via lua_gc)
|
** Default settings for GC tunables (settable via lua_gc)
|
||||||
*/
|
*/
|
||||||
@ -74,6 +76,7 @@
|
|||||||
#define luaC_white(g) cast_to(uint8_t, ((g)->currentwhite) & WHITEBITS)
|
#define luaC_white(g) cast_to(uint8_t, ((g)->currentwhite) & WHITEBITS)
|
||||||
|
|
||||||
// Thread stack states
|
// Thread stack states
|
||||||
|
// TODO: Remove with FFlag::LuauNoSleepBit and replace with lua_State::threadactive
|
||||||
#define THREAD_ACTIVEBIT 0 // thread is currently active
|
#define THREAD_ACTIVEBIT 0 // thread is currently active
|
||||||
#define THREAD_SLEEPINGBIT 1 // thread is not executing and stack should not be modified
|
#define THREAD_SLEEPINGBIT 1 // thread is not executing and stack should not be modified
|
||||||
|
|
||||||
@ -109,7 +112,7 @@
|
|||||||
#define luaC_barrierfast(L, t) \
|
#define luaC_barrierfast(L, t) \
|
||||||
{ \
|
{ \
|
||||||
if (isblack(obj2gco(t))) \
|
if (isblack(obj2gco(t))) \
|
||||||
luaC_barrierback(L, t); \
|
luaC_barrierback(L, obj2gco(t), &t->gclist); \
|
||||||
}
|
}
|
||||||
|
|
||||||
#define luaC_objbarrier(L, p, o) \
|
#define luaC_objbarrier(L, p, o) \
|
||||||
@ -118,29 +121,43 @@
|
|||||||
luaC_barrierf(L, obj2gco(p), obj2gco(o)); \
|
luaC_barrierf(L, obj2gco(p), obj2gco(o)); \
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TODO: Remove with FFlag::LuauSimplerUpval
|
||||||
#define luaC_upvalbarrier(L, uv, tv) \
|
#define luaC_upvalbarrier(L, uv, tv) \
|
||||||
{ \
|
{ \
|
||||||
if (iscollectable(tv) && iswhite(gcvalue(tv)) && (!(uv) || (uv)->v != &(uv)->u.value)) \
|
if (iscollectable(tv) && iswhite(gcvalue(tv)) && (!(uv) || (uv)->v != &(uv)->u.value)) \
|
||||||
luaC_barrierupval(L, gcvalue(tv)); \
|
luaC_barrierupval(L, gcvalue(tv)); \
|
||||||
}
|
}
|
||||||
|
|
||||||
#define luaC_checkthreadsleep(L) \
|
#define luaC_threadbarrier(L) \
|
||||||
{ \
|
{ \
|
||||||
if (luaC_threadsleeping(L)) \
|
if (FFlag::LuauNoSleepBit) \
|
||||||
luaC_wakethread(L); \
|
{ \
|
||||||
|
if (isblack(obj2gco(L))) \
|
||||||
|
luaC_barrierback(L, obj2gco(L), &L->gclist); \
|
||||||
|
} \
|
||||||
|
else \
|
||||||
|
{ \
|
||||||
|
if (luaC_threadsleeping(L)) \
|
||||||
|
luaC_wakethread(L); \
|
||||||
|
} \
|
||||||
}
|
}
|
||||||
|
|
||||||
#define luaC_init(L, o, tt) luaC_initobj(L, cast_to(GCObject*, (o)), tt)
|
#define luaC_init(L, o, tt_) \
|
||||||
|
{ \
|
||||||
|
o->marked = luaC_white(L->global); \
|
||||||
|
o->tt = tt_; \
|
||||||
|
o->memcat = L->activememcat; \
|
||||||
|
}
|
||||||
|
|
||||||
LUAI_FUNC void luaC_freeall(lua_State* L);
|
LUAI_FUNC void luaC_freeall(lua_State* L);
|
||||||
LUAI_FUNC size_t luaC_step(lua_State* L, bool assist);
|
LUAI_FUNC size_t luaC_step(lua_State* L, bool assist);
|
||||||
LUAI_FUNC void luaC_fullgc(lua_State* L);
|
LUAI_FUNC void luaC_fullgc(lua_State* L);
|
||||||
LUAI_FUNC void luaC_initobj(lua_State* L, GCObject* o, uint8_t tt);
|
LUAI_FUNC void luaC_initobj(lua_State* L, GCObject* o, uint8_t tt);
|
||||||
LUAI_FUNC void luaC_initupval(lua_State* L, UpVal* uv);
|
LUAI_FUNC void luaC_upvalclosed(lua_State* L, UpVal* uv);
|
||||||
LUAI_FUNC void luaC_barrierupval(lua_State* L, GCObject* v);
|
LUAI_FUNC void luaC_barrierupval(lua_State* L, GCObject* v);
|
||||||
LUAI_FUNC void luaC_barrierf(lua_State* L, GCObject* o, GCObject* v);
|
LUAI_FUNC void luaC_barrierf(lua_State* L, GCObject* o, GCObject* v);
|
||||||
LUAI_FUNC void luaC_barriertable(lua_State* L, Table* t, GCObject* v);
|
LUAI_FUNC void luaC_barriertable(lua_State* L, Table* t, GCObject* v);
|
||||||
LUAI_FUNC void luaC_barrierback(lua_State* L, Table* t);
|
LUAI_FUNC void luaC_barrierback(lua_State* L, GCObject* o, GCObject** gclist);
|
||||||
LUAI_FUNC void luaC_validate(lua_State* L);
|
LUAI_FUNC void luaC_validate(lua_State* L);
|
||||||
LUAI_FUNC void luaC_dump(lua_State* L, void* file, const char* (*categoryName)(lua_State* L, uint8_t memcat));
|
LUAI_FUNC void luaC_dump(lua_State* L, void* file, const char* (*categoryName)(lua_State* L, uint8_t memcat));
|
||||||
LUAI_FUNC int64_t luaC_allocationrate(lua_State* L);
|
LUAI_FUNC int64_t luaC_allocationrate(lua_State* L);
|
||||||
|
@ -102,10 +102,12 @@ static void validatestack(global_State* g, lua_State* l)
|
|||||||
if (l->namecall)
|
if (l->namecall)
|
||||||
validateobjref(g, obj2gco(l), obj2gco(l->namecall));
|
validateobjref(g, obj2gco(l), obj2gco(l->namecall));
|
||||||
|
|
||||||
for (UpVal* uv = l->openupval; uv; uv = uv->u.l.threadnext)
|
for (UpVal* uv = l->openupval; uv; uv = uv->u.open.threadnext)
|
||||||
{
|
{
|
||||||
LUAU_ASSERT(uv->tt == LUA_TUPVAL);
|
LUAU_ASSERT(uv->tt == LUA_TUPVAL);
|
||||||
LUAU_ASSERT(uv->v != &uv->u.value);
|
LUAU_ASSERT(upisopen(uv));
|
||||||
|
LUAU_ASSERT(uv->u.open.next->u.open.prev == uv && uv->u.open.prev->u.open.next == uv);
|
||||||
|
LUAU_ASSERT(!isblack(obj2gco(uv))); // open upvalues are never black
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -235,11 +237,12 @@ void luaC_validate(lua_State* L)
|
|||||||
|
|
||||||
luaM_visitgco(L, L, validategco);
|
luaM_visitgco(L, L, validategco);
|
||||||
|
|
||||||
for (UpVal* uv = g->uvhead.u.l.next; uv != &g->uvhead; uv = uv->u.l.next)
|
for (UpVal* uv = g->uvhead.u.open.next; uv != &g->uvhead; uv = uv->u.open.next)
|
||||||
{
|
{
|
||||||
LUAU_ASSERT(uv->tt == LUA_TUPVAL);
|
LUAU_ASSERT(uv->tt == LUA_TUPVAL);
|
||||||
LUAU_ASSERT(uv->v != &uv->u.value);
|
LUAU_ASSERT(upisopen(uv));
|
||||||
LUAU_ASSERT(uv->u.l.next->u.l.prev == uv && uv->u.l.prev->u.l.next == uv);
|
LUAU_ASSERT(uv->u.open.next->u.open.prev == uv && uv->u.open.prev->u.open.next == uv);
|
||||||
|
LUAU_ASSERT(!isblack(obj2gco(uv))); // open upvalues are never black
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -508,13 +511,14 @@ static void dumpproto(FILE* f, Proto* p)
|
|||||||
|
|
||||||
static void dumpupval(FILE* f, UpVal* uv)
|
static void dumpupval(FILE* f, UpVal* uv)
|
||||||
{
|
{
|
||||||
fprintf(f, "{\"type\":\"upvalue\",\"cat\":%d,\"size\":%d", uv->memcat, int(sizeof(UpVal)));
|
fprintf(f, "{\"type\":\"upvalue\",\"cat\":%d,\"size\":%d,\"open\":%s", uv->memcat, int(sizeof(UpVal)), upisopen(uv) ? "true" : "false");
|
||||||
|
|
||||||
if (iscollectable(uv->v))
|
if (iscollectable(uv->v))
|
||||||
{
|
{
|
||||||
fprintf(f, ",\"object\":");
|
fprintf(f, ",\"object\":");
|
||||||
dumpref(f, gcvalue(uv->v));
|
dumpref(f, gcvalue(uv->v));
|
||||||
}
|
}
|
||||||
|
|
||||||
fprintf(f, "}");
|
fprintf(f, "}");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -232,7 +232,7 @@ typedef struct TString
|
|||||||
int16_t atom;
|
int16_t atom;
|
||||||
// 2 byte padding
|
// 2 byte padding
|
||||||
|
|
||||||
TString* next; // next string in the hash table bucket or the string buffer linked list
|
TString* next; // next string in the hash table bucket
|
||||||
|
|
||||||
unsigned int hash;
|
unsigned int hash;
|
||||||
unsigned int len;
|
unsigned int len;
|
||||||
@ -316,7 +316,10 @@ typedef struct LocVar
|
|||||||
typedef struct UpVal
|
typedef struct UpVal
|
||||||
{
|
{
|
||||||
CommonHeader;
|
CommonHeader;
|
||||||
// 1 (x86) or 5 (x64) byte padding
|
uint8_t markedopen; // set if reachable from an alive thread (only valid during atomic)
|
||||||
|
|
||||||
|
// 4 byte padding (x64)
|
||||||
|
|
||||||
TValue* v; // points to stack or to its own value
|
TValue* v; // points to stack or to its own value
|
||||||
union
|
union
|
||||||
{
|
{
|
||||||
@ -327,14 +330,16 @@ typedef struct UpVal
|
|||||||
struct UpVal* prev;
|
struct UpVal* prev;
|
||||||
struct UpVal* next;
|
struct UpVal* next;
|
||||||
|
|
||||||
// thread double linked list (when open)
|
// thread linked list (when open)
|
||||||
struct UpVal* threadnext;
|
struct UpVal* threadnext;
|
||||||
// note: this is the location of a pointer to this upvalue in the previous element that can be either an UpVal or a lua_State
|
// note: this is the location of a pointer to this upvalue in the previous element that can be either an UpVal or a lua_State
|
||||||
struct UpVal** threadprev;
|
struct UpVal** threadprev; // TODO: remove with FFlag::LuauSimplerUpval
|
||||||
} l;
|
} open;
|
||||||
} u;
|
} u;
|
||||||
} UpVal;
|
} UpVal;
|
||||||
|
|
||||||
|
#define upisopen(up) ((up)->v != &(up)->u.value)
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** Closures
|
** Closures
|
||||||
*/
|
*/
|
||||||
|
@ -10,6 +10,8 @@
|
|||||||
#include "ldo.h"
|
#include "ldo.h"
|
||||||
#include "ldebug.h"
|
#include "ldebug.h"
|
||||||
|
|
||||||
|
LUAU_FASTFLAG(LuauSimplerUpval)
|
||||||
|
|
||||||
/*
|
/*
|
||||||
** Main thread combines a thread state and the global state
|
** Main thread combines a thread state and the global state
|
||||||
*/
|
*/
|
||||||
@ -119,8 +121,11 @@ lua_State* luaE_newthread(lua_State* L)
|
|||||||
|
|
||||||
void luaE_freethread(lua_State* L, lua_State* L1, lua_Page* page)
|
void luaE_freethread(lua_State* L, lua_State* L1, lua_Page* page)
|
||||||
{
|
{
|
||||||
luaF_close(L1, L1->stack); // close all upvalues for this thread
|
if (!FFlag::LuauSimplerUpval)
|
||||||
LUAU_ASSERT(L1->openupval == NULL);
|
{
|
||||||
|
luaF_close(L1, L1->stack); // close all upvalues for this thread
|
||||||
|
LUAU_ASSERT(L1->openupval == NULL);
|
||||||
|
}
|
||||||
global_State* g = L->global;
|
global_State* g = L->global;
|
||||||
if (g->cb.userthread)
|
if (g->cb.userthread)
|
||||||
g->cb.userthread(NULL, L1);
|
g->cb.userthread(NULL, L1);
|
||||||
@ -175,8 +180,8 @@ lua_State* lua_newstate(lua_Alloc f, void* ud)
|
|||||||
g->frealloc = f;
|
g->frealloc = f;
|
||||||
g->ud = ud;
|
g->ud = ud;
|
||||||
g->mainthread = L;
|
g->mainthread = L;
|
||||||
g->uvhead.u.l.prev = &g->uvhead;
|
g->uvhead.u.open.prev = &g->uvhead;
|
||||||
g->uvhead.u.l.next = &g->uvhead;
|
g->uvhead.u.open.next = &g->uvhead;
|
||||||
g->GCthreshold = 0; // mark it as unfinished state
|
g->GCthreshold = 0; // mark it as unfinished state
|
||||||
g->registryfree = 0;
|
g->registryfree = 0;
|
||||||
g->errorjmp = NULL;
|
g->errorjmp = NULL;
|
||||||
|
@ -167,7 +167,7 @@ typedef struct global_State
|
|||||||
GCObject* grayagain; // list of objects to be traversed atomically
|
GCObject* grayagain; // list of objects to be traversed atomically
|
||||||
GCObject* weak; // list of weak tables (to be cleared)
|
GCObject* weak; // list of weak tables (to be cleared)
|
||||||
|
|
||||||
TString* strbufgc; // list of all string buffer objects
|
TString* strbufgc; // list of all string buffer objects; TODO: remove with LuauNoStrbufLink
|
||||||
|
|
||||||
|
|
||||||
size_t GCthreshold; // when totalbytes > GCthreshold, run GC step
|
size_t GCthreshold; // when totalbytes > GCthreshold, run GC step
|
||||||
|
@ -7,6 +7,8 @@
|
|||||||
|
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
|
||||||
|
LUAU_FASTFLAGVARIABLE(LuauNoStrbufLink, false)
|
||||||
|
|
||||||
unsigned int luaS_hash(const char* str, size_t len)
|
unsigned int luaS_hash(const char* str, size_t len)
|
||||||
{
|
{
|
||||||
// Note that this hashing algorithm is replicated in BytecodeBuilder.cpp, BytecodeBuilder::getStringHash
|
// Note that this hashing algorithm is replicated in BytecodeBuilder.cpp, BytecodeBuilder::getStringHash
|
||||||
@ -70,40 +72,33 @@ void luaS_resize(lua_State* L, int newsize)
|
|||||||
|
|
||||||
static TString* newlstr(lua_State* L, const char* str, size_t l, unsigned int h)
|
static TString* newlstr(lua_State* L, const char* str, size_t l, unsigned int h)
|
||||||
{
|
{
|
||||||
TString* ts;
|
|
||||||
stringtable* tb;
|
|
||||||
if (l > MAXSSIZE)
|
if (l > MAXSSIZE)
|
||||||
luaM_toobig(L);
|
luaM_toobig(L);
|
||||||
ts = luaM_newgco(L, TString, sizestring(l), L->activememcat);
|
|
||||||
ts->len = unsigned(l);
|
TString* ts = luaM_newgco(L, TString, sizestring(l), L->activememcat);
|
||||||
|
luaC_init(L, ts, LUA_TSTRING);
|
||||||
|
ts->atom = ATOM_UNDEF;
|
||||||
ts->hash = h;
|
ts->hash = h;
|
||||||
ts->marked = luaC_white(L->global);
|
ts->len = unsigned(l);
|
||||||
ts->tt = LUA_TSTRING;
|
|
||||||
ts->memcat = L->activememcat;
|
|
||||||
memcpy(ts->data, str, l);
|
memcpy(ts->data, str, l);
|
||||||
ts->data[l] = '\0'; // ending 0
|
ts->data[l] = '\0'; // ending 0
|
||||||
ts->atom = ATOM_UNDEF;
|
|
||||||
tb = &L->global->strt;
|
stringtable* tb = &L->global->strt;
|
||||||
h = lmod(h, tb->size);
|
h = lmod(h, tb->size);
|
||||||
ts->next = tb->hash[h]; // chain new entry
|
ts->next = tb->hash[h]; // chain new entry
|
||||||
tb->hash[h] = ts;
|
tb->hash[h] = ts;
|
||||||
|
|
||||||
tb->nuse++;
|
tb->nuse++;
|
||||||
if (tb->nuse > cast_to(uint32_t, tb->size) && tb->size <= INT_MAX / 2)
|
if (tb->nuse > cast_to(uint32_t, tb->size) && tb->size <= INT_MAX / 2)
|
||||||
luaS_resize(L, tb->size * 2); // too crowded
|
luaS_resize(L, tb->size * 2); // too crowded
|
||||||
|
|
||||||
return ts;
|
return ts;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void linkstrbuf(lua_State* L, TString* ts)
|
|
||||||
{
|
|
||||||
global_State* g = L->global;
|
|
||||||
|
|
||||||
ts->next = g->strbufgc;
|
|
||||||
g->strbufgc = ts;
|
|
||||||
ts->marked = luaC_white(g);
|
|
||||||
}
|
|
||||||
|
|
||||||
static void unlinkstrbuf(lua_State* L, TString* ts)
|
static void unlinkstrbuf(lua_State* L, TString* ts)
|
||||||
{
|
{
|
||||||
|
LUAU_ASSERT(!FFlag::LuauNoStrbufLink);
|
||||||
global_State* g = L->global;
|
global_State* g = L->global;
|
||||||
|
|
||||||
TString** p = &g->strbufgc;
|
TString** p = &g->strbufgc;
|
||||||
@ -129,14 +124,24 @@ TString* luaS_bufstart(lua_State* L, size_t size)
|
|||||||
if (size > MAXSSIZE)
|
if (size > MAXSSIZE)
|
||||||
luaM_toobig(L);
|
luaM_toobig(L);
|
||||||
|
|
||||||
|
global_State* g = L->global;
|
||||||
|
|
||||||
TString* ts = luaM_newgco(L, TString, sizestring(size), L->activememcat);
|
TString* ts = luaM_newgco(L, TString, sizestring(size), L->activememcat);
|
||||||
|
luaC_init(L, ts, LUA_TSTRING);
|
||||||
ts->tt = LUA_TSTRING;
|
ts->atom = ATOM_UNDEF;
|
||||||
ts->memcat = L->activememcat;
|
ts->hash = 0; // computed in luaS_buffinish
|
||||||
linkstrbuf(L, ts);
|
|
||||||
|
|
||||||
ts->len = unsigned(size);
|
ts->len = unsigned(size);
|
||||||
|
|
||||||
|
if (FFlag::LuauNoStrbufLink)
|
||||||
|
{
|
||||||
|
ts->next = NULL;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
ts->next = g->strbufgc;
|
||||||
|
g->strbufgc = ts;
|
||||||
|
}
|
||||||
|
|
||||||
return ts;
|
return ts;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -159,7 +164,10 @@ TString* luaS_buffinish(lua_State* L, TString* ts)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
unlinkstrbuf(L, ts);
|
if (FFlag::LuauNoStrbufLink)
|
||||||
|
LUAU_ASSERT(ts->next == NULL);
|
||||||
|
else
|
||||||
|
unlinkstrbuf(L, ts);
|
||||||
|
|
||||||
ts->hash = h;
|
ts->hash = h;
|
||||||
ts->data[ts->len] = '\0'; // ending 0
|
ts->data[ts->len] = '\0'; // ending 0
|
||||||
@ -214,11 +222,21 @@ static bool unlinkstr(lua_State* L, TString* ts)
|
|||||||
|
|
||||||
void luaS_free(lua_State* L, TString* ts, lua_Page* page)
|
void luaS_free(lua_State* L, TString* ts, lua_Page* page)
|
||||||
{
|
{
|
||||||
// Unchain from the string table
|
if (FFlag::LuauNoStrbufLink)
|
||||||
if (!unlinkstr(L, ts))
|
{
|
||||||
unlinkstrbuf(L, ts); // An unlikely scenario when we have a string buffer on our hands
|
if (unlinkstr(L, ts))
|
||||||
|
L->global->strt.nuse--;
|
||||||
|
else
|
||||||
|
LUAU_ASSERT(ts->next == NULL); // orphaned string buffer
|
||||||
|
}
|
||||||
else
|
else
|
||||||
L->global->strt.nuse--;
|
{
|
||||||
|
// Unchain from the string table
|
||||||
|
if (!unlinkstr(L, ts))
|
||||||
|
unlinkstrbuf(L, ts); // An unlikely scenario when we have a string buffer on our hands
|
||||||
|
else
|
||||||
|
L->global->strt.nuse--;
|
||||||
|
}
|
||||||
|
|
||||||
luaM_freegco(L, ts, sizestring(ts->len), ts->memcat, page);
|
luaM_freegco(L, ts, sizestring(ts->len), ts->memcat, page);
|
||||||
}
|
}
|
||||||
|
@ -16,7 +16,8 @@
|
|||||||
|
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
|
||||||
LUAU_FASTFLAGVARIABLE(LuauNicerMethodErrors, false)
|
LUAU_FASTFLAG(LuauSimplerUpval)
|
||||||
|
LUAU_FASTFLAG(LuauNoSleepBit)
|
||||||
|
|
||||||
// Disable c99-designator to avoid the warning in CGOTO dispatch table
|
// Disable c99-designator to avoid the warning in CGOTO dispatch table
|
||||||
#ifdef __clang__
|
#ifdef __clang__
|
||||||
@ -111,7 +112,7 @@ LUAU_FASTFLAGVARIABLE(LuauNicerMethodErrors, false)
|
|||||||
VM_DISPATCH_OP(LOP_LOADKX), VM_DISPATCH_OP(LOP_JUMPX), VM_DISPATCH_OP(LOP_FASTCALL), VM_DISPATCH_OP(LOP_COVERAGE), \
|
VM_DISPATCH_OP(LOP_LOADKX), VM_DISPATCH_OP(LOP_JUMPX), VM_DISPATCH_OP(LOP_FASTCALL), VM_DISPATCH_OP(LOP_COVERAGE), \
|
||||||
VM_DISPATCH_OP(LOP_CAPTURE), VM_DISPATCH_OP(LOP_JUMPIFEQK), VM_DISPATCH_OP(LOP_JUMPIFNOTEQK), VM_DISPATCH_OP(LOP_FASTCALL1), \
|
VM_DISPATCH_OP(LOP_CAPTURE), VM_DISPATCH_OP(LOP_JUMPIFEQK), VM_DISPATCH_OP(LOP_JUMPIFNOTEQK), VM_DISPATCH_OP(LOP_FASTCALL1), \
|
||||||
VM_DISPATCH_OP(LOP_FASTCALL2), VM_DISPATCH_OP(LOP_FASTCALL2K), VM_DISPATCH_OP(LOP_FORGPREP), VM_DISPATCH_OP(LOP_JUMPXEQKNIL), \
|
VM_DISPATCH_OP(LOP_FASTCALL2), VM_DISPATCH_OP(LOP_FASTCALL2K), VM_DISPATCH_OP(LOP_FORGPREP), VM_DISPATCH_OP(LOP_JUMPXEQKNIL), \
|
||||||
VM_DISPATCH_OP(LOP_JUMPXEQKB), VM_DISPATCH_OP(LOP_JUMPXEQKN), VM_DISPATCH_OP(LOP_JUMPXEQKS), \
|
VM_DISPATCH_OP(LOP_JUMPXEQKB), VM_DISPATCH_OP(LOP_JUMPXEQKN), VM_DISPATCH_OP(LOP_JUMPXEQKS),
|
||||||
|
|
||||||
#if defined(__GNUC__) || defined(__clang__)
|
#if defined(__GNUC__) || defined(__clang__)
|
||||||
#define VM_USE_CGOTO 1
|
#define VM_USE_CGOTO 1
|
||||||
@ -317,6 +318,7 @@ static void luau_execute(lua_State* L)
|
|||||||
LUAU_ASSERT(isLua(L->ci));
|
LUAU_ASSERT(isLua(L->ci));
|
||||||
LUAU_ASSERT(luaC_threadactive(L));
|
LUAU_ASSERT(luaC_threadactive(L));
|
||||||
LUAU_ASSERT(!luaC_threadsleeping(L));
|
LUAU_ASSERT(!luaC_threadsleeping(L));
|
||||||
|
LUAU_ASSERT(!FFlag::LuauNoSleepBit || !isblack(obj2gco(L))); // we don't use luaC_threadbarrier because active threads never turn black
|
||||||
|
|
||||||
pc = L->ci->savedpc;
|
pc = L->ci->savedpc;
|
||||||
cl = clvalue(L->ci->func);
|
cl = clvalue(L->ci->func);
|
||||||
@ -496,7 +498,8 @@ static void luau_execute(lua_State* L)
|
|||||||
|
|
||||||
setobj(L, uv->v, ra);
|
setobj(L, uv->v, ra);
|
||||||
luaC_barrier(L, uv, ra);
|
luaC_barrier(L, uv, ra);
|
||||||
luaC_upvalbarrier(L, uv, uv->v);
|
if (!FFlag::LuauSimplerUpval)
|
||||||
|
luaC_upvalbarrier(L, uv, uv->v);
|
||||||
VM_NEXT();
|
VM_NEXT();
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -932,7 +935,7 @@ static void luau_execute(lua_State* L)
|
|||||||
VM_PATCH_C(pc - 2, L->cachedslot);
|
VM_PATCH_C(pc - 2, L->cachedslot);
|
||||||
// recompute ra since stack might have been reallocated
|
// recompute ra since stack might have been reallocated
|
||||||
ra = VM_REG(LUAU_INSN_A(insn));
|
ra = VM_REG(LUAU_INSN_A(insn));
|
||||||
if (FFlag::LuauNicerMethodErrors && ttisnil(ra))
|
if (ttisnil(ra))
|
||||||
luaG_methoderror(L, ra + 1, tsvalue(kv));
|
luaG_methoderror(L, ra + 1, tsvalue(kv));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -973,7 +976,7 @@ static void luau_execute(lua_State* L)
|
|||||||
VM_PATCH_C(pc - 2, L->cachedslot);
|
VM_PATCH_C(pc - 2, L->cachedslot);
|
||||||
// recompute ra since stack might have been reallocated
|
// recompute ra since stack might have been reallocated
|
||||||
ra = VM_REG(LUAU_INSN_A(insn));
|
ra = VM_REG(LUAU_INSN_A(insn));
|
||||||
if (FFlag::LuauNicerMethodErrors && ttisnil(ra))
|
if (ttisnil(ra))
|
||||||
luaG_methoderror(L, ra + 1, tsvalue(kv));
|
luaG_methoderror(L, ra + 1, tsvalue(kv));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -984,7 +987,7 @@ static void luau_execute(lua_State* L)
|
|||||||
VM_PROTECT(luaV_gettable(L, rb, kv, ra));
|
VM_PROTECT(luaV_gettable(L, rb, kv, ra));
|
||||||
// recompute ra since stack might have been reallocated
|
// recompute ra since stack might have been reallocated
|
||||||
ra = VM_REG(LUAU_INSN_A(insn));
|
ra = VM_REG(LUAU_INSN_A(insn));
|
||||||
if (FFlag::LuauNicerMethodErrors && ttisnil(ra))
|
if (ttisnil(ra))
|
||||||
luaG_methoderror(L, ra + 1, tsvalue(kv));
|
luaG_methoderror(L, ra + 1, tsvalue(kv));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -351,7 +351,7 @@ int luau_load(lua_State* L, const char* chunkname, const char* data, size_t size
|
|||||||
uint32_t mainid = readVarInt(data, size, offset);
|
uint32_t mainid = readVarInt(data, size, offset);
|
||||||
Proto* main = protos[mainid];
|
Proto* main = protos[mainid];
|
||||||
|
|
||||||
luaC_checkthreadsleep(L);
|
luaC_threadbarrier(L);
|
||||||
|
|
||||||
Closure* cl = luaF_newLclosure(L, 0, envt, main);
|
Closure* cl = luaF_newLclosure(L, 0, envt, main);
|
||||||
setclvalue(L, L->top, cl);
|
setclvalue(L, L->top, cl);
|
||||||
|
@ -10,9 +10,6 @@
|
|||||||
#include "lnumutils.h"
|
#include "lnumutils.h"
|
||||||
|
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
#include <stdio.h>
|
|
||||||
|
|
||||||
LUAU_FASTFLAGVARIABLE(LuauBetterNewindex, false)
|
|
||||||
|
|
||||||
// limit for table tag-method chains (to avoid loops)
|
// limit for table tag-method chains (to avoid loops)
|
||||||
#define MAXTAGLOOP 100
|
#define MAXTAGLOOP 100
|
||||||
@ -142,46 +139,25 @@ void luaV_settable(lua_State* L, const TValue* t, TValue* key, StkId val)
|
|||||||
{ // `t' is a table?
|
{ // `t' is a table?
|
||||||
Table* h = hvalue(t);
|
Table* h = hvalue(t);
|
||||||
|
|
||||||
if (FFlag::LuauBetterNewindex)
|
const TValue* oldval = luaH_get(h, key);
|
||||||
{
|
|
||||||
const TValue* oldval = luaH_get(h, key);
|
|
||||||
|
|
||||||
// should we assign the key? (if key is valid or __newindex is not set)
|
// should we assign the key? (if key is valid or __newindex is not set)
|
||||||
if (!ttisnil(oldval) || (tm = fasttm(L, h->metatable, TM_NEWINDEX)) == NULL)
|
if (!ttisnil(oldval) || (tm = fasttm(L, h->metatable, TM_NEWINDEX)) == NULL)
|
||||||
{
|
|
||||||
if (h->readonly)
|
|
||||||
luaG_readonlyerror(L);
|
|
||||||
|
|
||||||
// luaH_set would work but would repeat the lookup so we use luaH_setslot that can reuse oldval if it's safe
|
|
||||||
TValue* newval = luaH_setslot(L, h, oldval, key);
|
|
||||||
|
|
||||||
L->cachedslot = gval2slot(h, newval); // remember slot to accelerate future lookups
|
|
||||||
|
|
||||||
setobj2t(L, newval, val);
|
|
||||||
luaC_barriert(L, h, val);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// fallthrough to metamethod
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
{
|
||||||
if (h->readonly)
|
if (h->readonly)
|
||||||
luaG_readonlyerror(L);
|
luaG_readonlyerror(L);
|
||||||
|
|
||||||
TValue* oldval = luaH_set(L, h, key); // do a primitive set
|
// luaH_set would work but would repeat the lookup so we use luaH_setslot that can reuse oldval if it's safe
|
||||||
|
TValue* newval = luaH_setslot(L, h, oldval, key);
|
||||||
|
|
||||||
L->cachedslot = gval2slot(h, oldval); // remember slot to accelerate future lookups
|
L->cachedslot = gval2slot(h, newval); // remember slot to accelerate future lookups
|
||||||
|
|
||||||
if (!ttisnil(oldval) || // result is no nil?
|
setobj2t(L, newval, val);
|
||||||
(tm = fasttm(L, h->metatable, TM_NEWINDEX)) == NULL)
|
luaC_barriert(L, h, val);
|
||||||
{ // or no TM?
|
return;
|
||||||
setobj2t(L, oldval, val);
|
|
||||||
luaC_barriert(L, h, val);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
// else will try the tag method
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// fallthrough to metamethod
|
||||||
}
|
}
|
||||||
else if (ttisnil(tm = luaT_gettmbyobj(L, t, TM_NEWINDEX)))
|
else if (ttisnil(tm = luaT_gettmbyobj(L, t, TM_NEWINDEX)))
|
||||||
luaG_indexerror(L, t, key);
|
luaG_indexerror(L, t, key);
|
||||||
|
@ -38,6 +38,7 @@ argumentParser.add_argument('--run-test', action='store', default=None, help='Re
|
|||||||
argumentParser.add_argument('--extra-loops', action='store',type=int,default=0, help='Amount of times to loop over one test (one test already performs multiple runs)')
|
argumentParser.add_argument('--extra-loops', action='store',type=int,default=0, help='Amount of times to loop over one test (one test already performs multiple runs)')
|
||||||
argumentParser.add_argument('--filename', action='store',type=str,default='bench', help='File name for graph and results file')
|
argumentParser.add_argument('--filename', action='store',type=str,default='bench', help='File name for graph and results file')
|
||||||
argumentParser.add_argument('--callgrind', dest='callgrind',action='store_const',const=1,default=0,help='Use callgrind to run benchmarks')
|
argumentParser.add_argument('--callgrind', dest='callgrind',action='store_const',const=1,default=0,help='Use callgrind to run benchmarks')
|
||||||
|
argumentParser.add_argument('--show-commands', dest='show_commands',action='store_const',const=1,default=0,help='Show the command line used to launch the VM and tests')
|
||||||
|
|
||||||
if matplotlib != None:
|
if matplotlib != None:
|
||||||
argumentParser.add_argument('--absolute', dest='absolute',action='store_const',const=1,default=0,help='Display absolute values instead of relative (enabled by default when benchmarking a single VM)')
|
argumentParser.add_argument('--absolute', dest='absolute',action='store_const',const=1,default=0,help='Display absolute values instead of relative (enabled by default when benchmarking a single VM)')
|
||||||
@ -87,17 +88,25 @@ def getCallgrindOutput(lines):
|
|||||||
|
|
||||||
return "".join(result)
|
return "".join(result)
|
||||||
|
|
||||||
|
def conditionallyShowCommand(cmd):
|
||||||
|
if arguments.show_commands:
|
||||||
|
print(f'{colored(Color.BLUE, "EXECUTING")}: {cmd}')
|
||||||
|
|
||||||
def getVmOutput(cmd):
|
def getVmOutput(cmd):
|
||||||
if os.name == "nt":
|
if os.name == "nt":
|
||||||
try:
|
try:
|
||||||
return subprocess.check_output("start /realtime /affinity 1 /b /wait cmd /C \"" + cmd + "\"", shell=True, cwd=scriptdir).decode()
|
fullCmd = "start /realtime /affinity 1 /b /wait cmd /C \"" + cmd + "\""
|
||||||
|
conditionallyShowCommand(fullCmd)
|
||||||
|
return subprocess.check_output(fullCmd, shell=True, cwd=scriptdir).decode()
|
||||||
except KeyboardInterrupt:
|
except KeyboardInterrupt:
|
||||||
exit(1)
|
exit(1)
|
||||||
except:
|
except:
|
||||||
return ""
|
return ""
|
||||||
elif arguments.callgrind:
|
elif arguments.callgrind:
|
||||||
try:
|
try:
|
||||||
subprocess.check_call("valgrind --tool=callgrind --callgrind-out-file=callgrind.out --combine-dumps=yes --dump-line=no " + cmd, shell=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL, cwd=scriptdir)
|
fullCmd = "valgrind --tool=callgrind --callgrind-out-file=callgrind.out --combine-dumps=yes --dump-line=no " + cmd
|
||||||
|
conditionallyShowCommand(fullCmd)
|
||||||
|
subprocess.check_call(fullCmd, shell=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL, cwd=scriptdir)
|
||||||
path = os.path.join(scriptdir, "callgrind.out")
|
path = os.path.join(scriptdir, "callgrind.out")
|
||||||
with open(path, "r") as file:
|
with open(path, "r") as file:
|
||||||
lines = file.readlines()
|
lines = file.readlines()
|
||||||
@ -106,6 +115,7 @@ def getVmOutput(cmd):
|
|||||||
except:
|
except:
|
||||||
return ""
|
return ""
|
||||||
else:
|
else:
|
||||||
|
conditionallyShowCommand(cmd)
|
||||||
with subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, cwd=scriptdir) as p:
|
with subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, cwd=scriptdir) as p:
|
||||||
# Try to lock to a single processor
|
# Try to lock to a single processor
|
||||||
if sys.platform != "darwin":
|
if sys.platform != "darwin":
|
||||||
|
@ -138,4 +138,80 @@ print(workspace:)
|
|||||||
REQUIRE(ancestry.back()->is<AstExprIndexName>());
|
REQUIRE(ancestry.back()->is<AstExprIndexName>());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
TEST_CASE_FIXTURE(Fixture, "Luau_query")
|
||||||
|
{
|
||||||
|
AstStatBlock* block = parse(R"(
|
||||||
|
if true then
|
||||||
|
end
|
||||||
|
)");
|
||||||
|
|
||||||
|
AstStatIf* if_ = Luau::query<AstStatIf>(block);
|
||||||
|
CHECK(if_);
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST_CASE_FIXTURE(Fixture, "Luau_query_for_2nd_if_stat_which_doesnt_exist")
|
||||||
|
{
|
||||||
|
AstStatBlock* block = parse(R"(
|
||||||
|
if true then
|
||||||
|
end
|
||||||
|
)");
|
||||||
|
|
||||||
|
AstStatIf* if_ = Luau::query<AstStatIf, 2>(block);
|
||||||
|
CHECK(!if_);
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST_CASE_FIXTURE(Fixture, "Luau_nested_query")
|
||||||
|
{
|
||||||
|
AstStatBlock* block = parse(R"(
|
||||||
|
if true then
|
||||||
|
end
|
||||||
|
)");
|
||||||
|
|
||||||
|
AstStatIf* if_ = Luau::query<AstStatIf>(block);
|
||||||
|
REQUIRE(if_);
|
||||||
|
AstExprConstantBool* bool_ = Luau::query<AstExprConstantBool>(if_);
|
||||||
|
REQUIRE(bool_);
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST_CASE_FIXTURE(Fixture, "Luau_nested_query_but_first_query_failed")
|
||||||
|
{
|
||||||
|
AstStatBlock* block = parse(R"(
|
||||||
|
if true then
|
||||||
|
end
|
||||||
|
)");
|
||||||
|
|
||||||
|
AstStatIf* if_ = Luau::query<AstStatIf, 2>(block);
|
||||||
|
REQUIRE(!if_);
|
||||||
|
AstExprConstantBool* bool_ = Luau::query<AstExprConstantBool>(if_); // ensure it doesn't crash
|
||||||
|
REQUIRE(!bool_);
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST_CASE_FIXTURE(Fixture, "Luau_selectively_query_for_a_different_boolean")
|
||||||
|
{
|
||||||
|
AstStatBlock* block = parse(R"(
|
||||||
|
local x = false and true
|
||||||
|
local y = true and false
|
||||||
|
)");
|
||||||
|
|
||||||
|
AstExprConstantBool* fst = Luau::query<AstExprConstantBool>(block, {nth<AstStatLocal>(), nth<AstExprConstantBool>(2)});
|
||||||
|
REQUIRE(fst);
|
||||||
|
REQUIRE(fst->value == true);
|
||||||
|
|
||||||
|
AstExprConstantBool* snd = Luau::query<AstExprConstantBool>(block, {nth<AstStatLocal>(2), nth<AstExprConstantBool>(2)});
|
||||||
|
REQUIRE(snd);
|
||||||
|
REQUIRE(snd->value == false);
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST_CASE_FIXTURE(Fixture, "Luau_selectively_query_for_a_different_boolean_2")
|
||||||
|
{
|
||||||
|
AstStatBlock* block = parse(R"(
|
||||||
|
local x = false and true
|
||||||
|
local y = true and false
|
||||||
|
)");
|
||||||
|
|
||||||
|
AstExprConstantBool* snd = Luau::query<AstExprConstantBool>(block, {nth<AstStatLocal>(2), nth<AstExprConstantBool>()});
|
||||||
|
REQUIRE(snd);
|
||||||
|
REQUIRE(snd->value == true);
|
||||||
|
}
|
||||||
|
|
||||||
TEST_SUITE_END();
|
TEST_SUITE_END();
|
||||||
|
@ -2154,8 +2154,6 @@ TEST_CASE("RecursionParse")
|
|||||||
CHECK_EQ(std::string(e.what()), "Exceeded allowed recursion depth; simplify your expression to make the code compile");
|
CHECK_EQ(std::string(e.what()), "Exceeded allowed recursion depth; simplify your expression to make the code compile");
|
||||||
}
|
}
|
||||||
|
|
||||||
#if 0
|
|
||||||
// This currently requires too much stack space on MSVC/x64 and crashes with stack overflow at recursion depth 935
|
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
Luau::compileOrThrow(bcb, rep("function a() ", 1500) + "print()" + rep(" end", 1500));
|
Luau::compileOrThrow(bcb, rep("function a() ", 1500) + "print()" + rep(" end", 1500));
|
||||||
@ -2175,7 +2173,6 @@ TEST_CASE("RecursionParse")
|
|||||||
{
|
{
|
||||||
CHECK_EQ(std::string(e.what()), "Exceeded allowed recursion depth; simplify your block to make the code compile");
|
CHECK_EQ(std::string(e.what()), "Exceeded allowed recursion depth; simplify your block to make the code compile");
|
||||||
}
|
}
|
||||||
#endif
|
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_CASE("ArrayIndexLiteral")
|
TEST_CASE("ArrayIndexLiteral")
|
||||||
|
@ -319,15 +319,11 @@ TEST_CASE("Literals")
|
|||||||
|
|
||||||
TEST_CASE("Errors")
|
TEST_CASE("Errors")
|
||||||
{
|
{
|
||||||
ScopedFastFlag sff("LuauNicerMethodErrors", true);
|
|
||||||
|
|
||||||
runConformance("errors.lua");
|
runConformance("errors.lua");
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_CASE("Events")
|
TEST_CASE("Events")
|
||||||
{
|
{
|
||||||
ScopedFastFlag sff("LuauBetterNewindex", true);
|
|
||||||
|
|
||||||
runConformance("events.lua");
|
runConformance("events.lua");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -512,4 +512,41 @@ void dump(const std::vector<Constraint>& constraints)
|
|||||||
printf("%s\n", toString(c, opts).c_str());
|
printf("%s\n", toString(c, opts).c_str());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
FindNthOccurenceOf::FindNthOccurenceOf(Nth nth)
|
||||||
|
: requestedNth(nth)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
bool FindNthOccurenceOf::checkIt(AstNode* n)
|
||||||
|
{
|
||||||
|
if (theNode)
|
||||||
|
return false;
|
||||||
|
|
||||||
|
if (n->classIndex == requestedNth.classIndex)
|
||||||
|
{
|
||||||
|
// Human factor: the requestedNth starts from 1 because of the term `nth`.
|
||||||
|
if (currentOccurrence + 1 != requestedNth.nth)
|
||||||
|
++currentOccurrence;
|
||||||
|
else
|
||||||
|
theNode = n;
|
||||||
|
}
|
||||||
|
|
||||||
|
return !theNode; // once found, returns false and stops traversal
|
||||||
|
}
|
||||||
|
|
||||||
|
bool FindNthOccurenceOf::visit(AstNode* n)
|
||||||
|
{
|
||||||
|
return checkIt(n);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool FindNthOccurenceOf::visit(AstType* t)
|
||||||
|
{
|
||||||
|
return checkIt(t);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool FindNthOccurenceOf::visit(AstTypePack* t)
|
||||||
|
{
|
||||||
|
return checkIt(t);
|
||||||
|
}
|
||||||
|
|
||||||
} // namespace Luau
|
} // namespace Luau
|
||||||
|
@ -195,6 +195,76 @@ std::optional<TypeId> lookupName(ScopePtr scope, const std::string& name); // Wa
|
|||||||
|
|
||||||
std::optional<TypeId> linearSearchForBinding(Scope* scope, const char* name);
|
std::optional<TypeId> linearSearchForBinding(Scope* scope, const char* name);
|
||||||
|
|
||||||
|
struct Nth
|
||||||
|
{
|
||||||
|
int classIndex;
|
||||||
|
int nth;
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename T>
|
||||||
|
Nth nth(int nth = 1)
|
||||||
|
{
|
||||||
|
static_assert(std::is_base_of_v<AstNode, T>, "T must be a derived class of AstNode");
|
||||||
|
LUAU_ASSERT(nth > 0); // Did you mean to use `nth<T>(1)`?
|
||||||
|
|
||||||
|
return Nth{T::ClassIndex(), nth};
|
||||||
|
}
|
||||||
|
|
||||||
|
struct FindNthOccurenceOf : public AstVisitor
|
||||||
|
{
|
||||||
|
Nth requestedNth;
|
||||||
|
int currentOccurrence = 0;
|
||||||
|
AstNode* theNode = nullptr;
|
||||||
|
|
||||||
|
FindNthOccurenceOf(Nth nth);
|
||||||
|
|
||||||
|
bool checkIt(AstNode* n);
|
||||||
|
|
||||||
|
bool visit(AstNode* n) override;
|
||||||
|
bool visit(AstType* n) override;
|
||||||
|
bool visit(AstTypePack* n) override;
|
||||||
|
};
|
||||||
|
|
||||||
|
/** DSL querying of the AST.
|
||||||
|
*
|
||||||
|
* Given an AST, one can query for a particular node directly without having to manually unwrap the tree, for example:
|
||||||
|
*
|
||||||
|
* ```
|
||||||
|
* if a and b then
|
||||||
|
* print(a + b)
|
||||||
|
* end
|
||||||
|
*
|
||||||
|
* function f(x, y)
|
||||||
|
* return x + y
|
||||||
|
* end
|
||||||
|
* ```
|
||||||
|
*
|
||||||
|
* There are numerous ways to access the second AstExprBinary.
|
||||||
|
* 1. Luau::query<AstExprBinary>(block, {nth<AstStatFunction>(), nth<AstExprBinary>()})
|
||||||
|
* 2. Luau::query<AstExprBinary>(Luau::query<AstStatFunction>(block))
|
||||||
|
* 3. Luau::query<AstExprBinary>(block, {nth<AstExprBinary>(2)})
|
||||||
|
*/
|
||||||
|
template<typename T, size_t N = 1>
|
||||||
|
T* query(AstNode* node, const std::vector<Nth>& nths = {nth<T>(N)})
|
||||||
|
{
|
||||||
|
static_assert(std::is_base_of_v<AstNode, T>, "T must be a derived class of AstNode");
|
||||||
|
|
||||||
|
// If a nested query call fails to find the node in question, subsequent calls can propagate rather than trying to do more.
|
||||||
|
// This supports `query(query(...))`
|
||||||
|
|
||||||
|
for (Nth nth : nths)
|
||||||
|
{
|
||||||
|
if (!node)
|
||||||
|
return nullptr;
|
||||||
|
|
||||||
|
FindNthOccurenceOf finder{nth};
|
||||||
|
node->visit(&finder);
|
||||||
|
node = finder.theNode;
|
||||||
|
}
|
||||||
|
|
||||||
|
return node ? node->as<T>() : nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
} // namespace Luau
|
} // namespace Luau
|
||||||
|
|
||||||
#define LUAU_REQUIRE_ERRORS(result) \
|
#define LUAU_REQUIRE_ERRORS(result) \
|
||||||
|
@ -43,6 +43,20 @@ TEST_CASE_FIXTURE(Fixture, "DeprecatedGlobal")
|
|||||||
CHECK_EQ(result.warnings[0].text, "Global 'Wait' is deprecated, use 'wait' instead");
|
CHECK_EQ(result.warnings[0].text, "Global 'Wait' is deprecated, use 'wait' instead");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
TEST_CASE_FIXTURE(Fixture, "DeprecatedGlobalNoReplacement")
|
||||||
|
{
|
||||||
|
ScopedFastFlag sff{"LuauLintFixDeprecationMessage", true};
|
||||||
|
|
||||||
|
// Normally this would be defined externally, so hack it in for testing
|
||||||
|
const char* deprecationReplacementString = "";
|
||||||
|
addGlobalBinding(typeChecker, "Version", Binding{typeChecker.anyType, {}, true, deprecationReplacementString});
|
||||||
|
|
||||||
|
LintResult result = lintTyped("Version()");
|
||||||
|
|
||||||
|
REQUIRE_EQ(result.warnings.size(), 1);
|
||||||
|
CHECK_EQ(result.warnings[0].text, "Global 'Version' is deprecated");
|
||||||
|
}
|
||||||
|
|
||||||
TEST_CASE_FIXTURE(Fixture, "PlaceholderRead")
|
TEST_CASE_FIXTURE(Fixture, "PlaceholderRead")
|
||||||
{
|
{
|
||||||
LintResult result = lint(R"(
|
LintResult result = lint(R"(
|
||||||
|
@ -11,6 +11,7 @@ using namespace Luau;
|
|||||||
|
|
||||||
LUAU_FASTFLAG(LuauRecursiveTypeParameterRestriction);
|
LUAU_FASTFLAG(LuauRecursiveTypeParameterRestriction);
|
||||||
LUAU_FASTFLAG(LuauSpecialTypesAsterisked);
|
LUAU_FASTFLAG(LuauSpecialTypesAsterisked);
|
||||||
|
LUAU_FASTFLAG(LuauFixNameMaps);
|
||||||
|
|
||||||
TEST_SUITE_BEGIN("ToString");
|
TEST_SUITE_BEGIN("ToString");
|
||||||
|
|
||||||
@ -433,29 +434,40 @@ TEST_CASE_FIXTURE(Fixture, "toStringDetailed")
|
|||||||
|
|
||||||
LUAU_REQUIRE_NO_ERRORS(result);
|
LUAU_REQUIRE_NO_ERRORS(result);
|
||||||
|
|
||||||
TypeId id3Type = requireType("id3");
|
ToStringOptions opts;
|
||||||
ToStringResult nameData = toStringDetailed(id3Type);
|
|
||||||
|
TypeId id3Type = requireType("id3");
|
||||||
|
ToStringResult nameData = toStringDetailed(id3Type, opts);
|
||||||
|
|
||||||
|
if (FFlag::LuauFixNameMaps)
|
||||||
|
REQUIRE(3 == opts.nameMap.typeVars.size());
|
||||||
|
else
|
||||||
|
REQUIRE_EQ(3, nameData.DEPRECATED_nameMap.typeVars.size());
|
||||||
|
|
||||||
REQUIRE_EQ(3, nameData.nameMap.typeVars.size());
|
|
||||||
REQUIRE_EQ("<a, b, c>(a, b, c) -> (a, b, c)", nameData.name);
|
REQUIRE_EQ("<a, b, c>(a, b, c) -> (a, b, c)", nameData.name);
|
||||||
|
|
||||||
ToStringOptions opts;
|
ToStringOptions opts2; // TODO: delete opts2 when clipping FFlag::LuauFixNameMaps
|
||||||
opts.nameMap = std::move(nameData.nameMap);
|
if (FFlag::LuauFixNameMaps)
|
||||||
|
opts2.nameMap = std::move(opts.nameMap);
|
||||||
|
else
|
||||||
|
opts2.DEPRECATED_nameMap = std::move(nameData.DEPRECATED_nameMap);
|
||||||
|
|
||||||
const FunctionTypeVar* ftv = get<FunctionTypeVar>(follow(id3Type));
|
const FunctionTypeVar* ftv = get<FunctionTypeVar>(follow(id3Type));
|
||||||
REQUIRE(ftv != nullptr);
|
REQUIRE(ftv != nullptr);
|
||||||
|
|
||||||
auto params = flatten(ftv->argTypes).first;
|
auto params = flatten(ftv->argTypes).first;
|
||||||
REQUIRE_EQ(3, params.size());
|
REQUIRE(3 == params.size());
|
||||||
|
|
||||||
REQUIRE_EQ("a", toString(params[0], opts));
|
CHECK("a" == toString(params[0], opts2));
|
||||||
REQUIRE_EQ("b", toString(params[1], opts));
|
CHECK("b" == toString(params[1], opts2));
|
||||||
REQUIRE_EQ("c", toString(params[2], opts));
|
CHECK("c" == toString(params[2], opts2));
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_CASE_FIXTURE(BuiltinsFixture, "toStringDetailed2")
|
TEST_CASE_FIXTURE(BuiltinsFixture, "toStringDetailed2")
|
||||||
{
|
{
|
||||||
ScopedFastFlag sff2{"DebugLuauSharedSelf", true};
|
ScopedFastFlag sff[] = {
|
||||||
|
{"DebugLuauSharedSelf", true},
|
||||||
|
};
|
||||||
|
|
||||||
CheckResult result = check(R"(
|
CheckResult result = check(R"(
|
||||||
local base = {}
|
local base = {}
|
||||||
@ -470,13 +482,18 @@ TEST_CASE_FIXTURE(BuiltinsFixture, "toStringDetailed2")
|
|||||||
)");
|
)");
|
||||||
LUAU_REQUIRE_NO_ERRORS(result);
|
LUAU_REQUIRE_NO_ERRORS(result);
|
||||||
|
|
||||||
TypeId tType = requireType("inst");
|
|
||||||
ToStringResult r = toStringDetailed(tType);
|
|
||||||
CHECK_EQ("{ @metatable { __index: { @metatable {| __index: base |}, child } }, inst }", r.name);
|
|
||||||
CHECK_EQ(0, r.nameMap.typeVars.size());
|
|
||||||
|
|
||||||
ToStringOptions opts;
|
ToStringOptions opts;
|
||||||
opts.nameMap = r.nameMap;
|
|
||||||
|
TypeId tType = requireType("inst");
|
||||||
|
ToStringResult r = toStringDetailed(tType, opts);
|
||||||
|
CHECK_EQ("{ @metatable { __index: { @metatable {| __index: base |}, child } }, inst }", r.name);
|
||||||
|
if (FFlag::LuauFixNameMaps)
|
||||||
|
CHECK(0 == opts.nameMap.typeVars.size());
|
||||||
|
else
|
||||||
|
CHECK_EQ(0, r.DEPRECATED_nameMap.typeVars.size());
|
||||||
|
|
||||||
|
if (!FFlag::LuauFixNameMaps)
|
||||||
|
opts.DEPRECATED_nameMap = r.DEPRECATED_nameMap;
|
||||||
|
|
||||||
const MetatableTypeVar* tMeta = get<MetatableTypeVar>(tType);
|
const MetatableTypeVar* tMeta = get<MetatableTypeVar>(tType);
|
||||||
REQUIRE(tMeta);
|
REQUIRE(tMeta);
|
||||||
@ -499,7 +516,8 @@ TEST_CASE_FIXTURE(BuiltinsFixture, "toStringDetailed2")
|
|||||||
REQUIRE(tMeta6);
|
REQUIRE(tMeta6);
|
||||||
|
|
||||||
ToStringResult oneResult = toStringDetailed(tMeta5->props["one"].type, opts);
|
ToStringResult oneResult = toStringDetailed(tMeta5->props["one"].type, opts);
|
||||||
opts.nameMap = oneResult.nameMap;
|
if (!FFlag::LuauFixNameMaps)
|
||||||
|
opts.DEPRECATED_nameMap = oneResult.DEPRECATED_nameMap;
|
||||||
|
|
||||||
std::string twoResult = toString(tMeta6->props["two"].type, opts);
|
std::string twoResult = toString(tMeta6->props["two"].type, opts);
|
||||||
|
|
||||||
|
@ -10,6 +10,7 @@ using namespace Luau;
|
|||||||
|
|
||||||
LUAU_FASTFLAG(LuauLowerBoundsCalculation);
|
LUAU_FASTFLAG(LuauLowerBoundsCalculation);
|
||||||
LUAU_FASTFLAG(LuauSpecialTypesAsterisked);
|
LUAU_FASTFLAG(LuauSpecialTypesAsterisked);
|
||||||
|
LUAU_FASTFLAG(LuauStringFormatArgumentErrorFix)
|
||||||
|
|
||||||
TEST_SUITE_BEGIN("BuiltinTests");
|
TEST_SUITE_BEGIN("BuiltinTests");
|
||||||
|
|
||||||
@ -721,7 +722,14 @@ TEST_CASE_FIXTURE(Fixture, "string_format_use_correct_argument")
|
|||||||
|
|
||||||
LUAU_REQUIRE_ERROR_COUNT(1, result);
|
LUAU_REQUIRE_ERROR_COUNT(1, result);
|
||||||
|
|
||||||
CHECK_EQ("Argument count mismatch. Function expects 1 argument, but 2 are specified", toString(result.errors[0]));
|
if (FFlag::LuauStringFormatArgumentErrorFix)
|
||||||
|
{
|
||||||
|
CHECK_EQ("Argument count mismatch. Function expects 2 arguments, but 3 are specified", toString(result.errors[0]));
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
CHECK_EQ("Argument count mismatch. Function expects 1 argument, but 2 are specified", toString(result.errors[0]));
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_CASE_FIXTURE(Fixture, "string_format_use_correct_argument2")
|
TEST_CASE_FIXTURE(Fixture, "string_format_use_correct_argument2")
|
||||||
@ -736,6 +744,22 @@ TEST_CASE_FIXTURE(Fixture, "string_format_use_correct_argument2")
|
|||||||
CHECK_EQ("Type 'number' could not be converted into 'string'", toString(result.errors[1]));
|
CHECK_EQ("Type 'number' could not be converted into 'string'", toString(result.errors[1]));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
TEST_CASE_FIXTURE(BuiltinsFixture, "string_format_use_correct_argument3")
|
||||||
|
{
|
||||||
|
ScopedFastFlag LuauStringFormatArgumentErrorFix{"LuauStringFormatArgumentErrorFix", true};
|
||||||
|
|
||||||
|
CheckResult result = check(R"(
|
||||||
|
local s1 = string.format("%d")
|
||||||
|
local s2 = string.format("%d", 1)
|
||||||
|
local s3 = string.format("%d", 1, 2)
|
||||||
|
)");
|
||||||
|
|
||||||
|
LUAU_REQUIRE_ERROR_COUNT(2, result);
|
||||||
|
|
||||||
|
CHECK_EQ("Argument count mismatch. Function expects 2 arguments, but only 1 is specified", toString(result.errors[0]));
|
||||||
|
CHECK_EQ("Argument count mismatch. Function expects 2 arguments, but 3 are specified", toString(result.errors[1]));
|
||||||
|
}
|
||||||
|
|
||||||
TEST_CASE_FIXTURE(BuiltinsFixture, "debug_traceback_is_crazy")
|
TEST_CASE_FIXTURE(BuiltinsFixture, "debug_traceback_is_crazy")
|
||||||
{
|
{
|
||||||
CheckResult result = check(R"(
|
CheckResult result = check(R"(
|
||||||
|
@ -544,4 +544,69 @@ TEST_CASE_FIXTURE(BuiltinsFixture, "greedy_inference_with_shared_self_triggers_f
|
|||||||
CHECK_EQ("Not all codepaths in this function return 'self, a...'.", toString(result.errors[0]));
|
CHECK_EQ("Not all codepaths in this function return 'self, a...'.", toString(result.errors[0]));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
TEST_CASE_FIXTURE(Fixture, "dcr_cant_partially_dispatch_a_constraint")
|
||||||
|
{
|
||||||
|
ScopedFastFlag sff[] = {
|
||||||
|
{"DebugLuauDeferredConstraintResolution", true},
|
||||||
|
{"LuauSpecialTypesAsterisked", true},
|
||||||
|
};
|
||||||
|
|
||||||
|
CheckResult result = check(R"(
|
||||||
|
local function hasDivisors(value: number)
|
||||||
|
end
|
||||||
|
|
||||||
|
function prime_iter(state, index)
|
||||||
|
hasDivisors(index)
|
||||||
|
index += 1
|
||||||
|
end
|
||||||
|
)");
|
||||||
|
|
||||||
|
LUAU_REQUIRE_NO_ERRORS(result);
|
||||||
|
|
||||||
|
// We should be able to resolve this to number, but we're not there yet.
|
||||||
|
// Solving this requires recognizing that we can partially solve the
|
||||||
|
// following constraint:
|
||||||
|
//
|
||||||
|
// (*blocked*) -> () <: (number) -> (b...)
|
||||||
|
//
|
||||||
|
// The correct thing for us to do is to consider the constraint dispatched,
|
||||||
|
// but we need to also record a new constraint number <: *blocked* to finish
|
||||||
|
// the job later.
|
||||||
|
CHECK("<a>(a, *error-type*) -> ()" == toString(requireType("prime_iter")));
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST_CASE_FIXTURE(Fixture, "free_options_cannot_be_unified_together")
|
||||||
|
{
|
||||||
|
ScopedFastFlag sff[] = {
|
||||||
|
{"LuauFixNameMaps", true},
|
||||||
|
};
|
||||||
|
|
||||||
|
TypeArena arena;
|
||||||
|
TypeId nilType = getSingletonTypes().nilType;
|
||||||
|
|
||||||
|
std::unique_ptr scope = std::make_unique<Scope>(getSingletonTypes().anyTypePack);
|
||||||
|
|
||||||
|
TypeId free1 = arena.addType(FreeTypePack{scope.get()});
|
||||||
|
TypeId option1 = arena.addType(UnionTypeVar{{nilType, free1}});
|
||||||
|
|
||||||
|
TypeId free2 = arena.addType(FreeTypePack{scope.get()});
|
||||||
|
TypeId option2 = arena.addType(UnionTypeVar{{nilType, free2}});
|
||||||
|
|
||||||
|
InternalErrorReporter iceHandler;
|
||||||
|
UnifierSharedState sharedState{&iceHandler};
|
||||||
|
Unifier u{&arena, Mode::Strict, NotNull{scope.get()}, Location{}, Variance::Covariant, sharedState};
|
||||||
|
|
||||||
|
u.tryUnify(option1, option2);
|
||||||
|
|
||||||
|
CHECK(u.errors.empty());
|
||||||
|
|
||||||
|
u.log.commit();
|
||||||
|
|
||||||
|
ToStringOptions opts;
|
||||||
|
CHECK("a?" == toString(option1, opts));
|
||||||
|
|
||||||
|
// CHECK("a?" == toString(option2, opts)); // This should hold, but does not.
|
||||||
|
CHECK("b?" == toString(option2, opts)); // This should not hold.
|
||||||
|
}
|
||||||
|
|
||||||
TEST_SUITE_END();
|
TEST_SUITE_END();
|
||||||
|
@ -252,25 +252,30 @@ if not rawget(_G, "_soft") then
|
|||||||
end
|
end
|
||||||
|
|
||||||
-- create many threads with self-references and open upvalues
|
-- create many threads with self-references and open upvalues
|
||||||
local thread_id = 0
|
do
|
||||||
local threads = {}
|
local thread_id = 0
|
||||||
|
local threads = {}
|
||||||
|
|
||||||
function fn(thread)
|
function fn(thread)
|
||||||
local x = {}
|
local x = {}
|
||||||
threads[thread_id] = function()
|
threads[thread_id] = function()
|
||||||
thread = x
|
thread = x
|
||||||
end
|
end
|
||||||
coroutine.yield()
|
coroutine.yield()
|
||||||
|
end
|
||||||
|
|
||||||
|
while thread_id < 1000 do
|
||||||
|
local thread = coroutine.create(fn)
|
||||||
|
coroutine.resume(thread, thread)
|
||||||
|
thread_id = thread_id + 1
|
||||||
|
end
|
||||||
|
|
||||||
|
collectgarbage()
|
||||||
|
|
||||||
|
-- ensure that we no longer have a lot of reachable threads for subsequent tests
|
||||||
|
threads = {}
|
||||||
end
|
end
|
||||||
|
|
||||||
while thread_id < 1000 do
|
|
||||||
local thread = coroutine.create(fn)
|
|
||||||
coroutine.resume(thread, thread)
|
|
||||||
thread_id = thread_id + 1
|
|
||||||
end
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
-- create a userdata to be collected when state is closed
|
-- create a userdata to be collected when state is closed
|
||||||
do
|
do
|
||||||
local newproxy,assert,type,print,getmetatable =
|
local newproxy,assert,type,print,getmetatable =
|
||||||
@ -322,4 +327,27 @@ do
|
|||||||
collectgarbage()
|
collectgarbage()
|
||||||
end
|
end
|
||||||
|
|
||||||
|
-- create a lot of threads with upvalues to force a case where full gc happens after we've marked some upvalues
|
||||||
|
do
|
||||||
|
local t = {}
|
||||||
|
for i = 1,100 do
|
||||||
|
local c = coroutine.wrap(function()
|
||||||
|
local uv = {i + 1}
|
||||||
|
local function f()
|
||||||
|
return uv[1] * 10
|
||||||
|
end
|
||||||
|
coroutine.yield(uv[1])
|
||||||
|
uv = {i + 2}
|
||||||
|
coroutine.yield(f())
|
||||||
|
end)
|
||||||
|
|
||||||
|
assert(c() == i + 1)
|
||||||
|
table.insert(t, c)
|
||||||
|
end
|
||||||
|
|
||||||
|
t = {}
|
||||||
|
|
||||||
|
collectgarbage()
|
||||||
|
end
|
||||||
|
|
||||||
return('OK')
|
return('OK')
|
||||||
|
@ -53,6 +53,7 @@ AutocompleteTest.generic_types
|
|||||||
AutocompleteTest.get_suggestions_for_the_very_start_of_the_script
|
AutocompleteTest.get_suggestions_for_the_very_start_of_the_script
|
||||||
AutocompleteTest.global_function_params
|
AutocompleteTest.global_function_params
|
||||||
AutocompleteTest.global_functions_are_not_scoped_lexically
|
AutocompleteTest.global_functions_are_not_scoped_lexically
|
||||||
|
AutocompleteTest.globals_are_order_independent
|
||||||
AutocompleteTest.if_then_else_elseif_completions
|
AutocompleteTest.if_then_else_elseif_completions
|
||||||
AutocompleteTest.keyword_methods
|
AutocompleteTest.keyword_methods
|
||||||
AutocompleteTest.keyword_types
|
AutocompleteTest.keyword_types
|
||||||
@ -588,7 +589,6 @@ TypeInferFunctions.another_recursive_local_function
|
|||||||
TypeInferFunctions.calling_function_with_anytypepack_doesnt_leak_free_types
|
TypeInferFunctions.calling_function_with_anytypepack_doesnt_leak_free_types
|
||||||
TypeInferFunctions.calling_function_with_incorrect_argument_type_yields_errors_spanning_argument
|
TypeInferFunctions.calling_function_with_incorrect_argument_type_yields_errors_spanning_argument
|
||||||
TypeInferFunctions.complicated_return_types_require_an_explicit_annotation
|
TypeInferFunctions.complicated_return_types_require_an_explicit_annotation
|
||||||
TypeInferFunctions.cyclic_function_type_in_args
|
|
||||||
TypeInferFunctions.dont_give_other_overloads_message_if_only_one_argument_matching_overload_exists
|
TypeInferFunctions.dont_give_other_overloads_message_if_only_one_argument_matching_overload_exists
|
||||||
TypeInferFunctions.dont_infer_parameter_types_for_functions_from_their_call_site
|
TypeInferFunctions.dont_infer_parameter_types_for_functions_from_their_call_site
|
||||||
TypeInferFunctions.duplicate_functions_with_different_signatures_not_allowed_in_nonstrict
|
TypeInferFunctions.duplicate_functions_with_different_signatures_not_allowed_in_nonstrict
|
||||||
@ -744,7 +744,6 @@ TypeInferUnknownNever.math_operators_and_never
|
|||||||
TypeInferUnknownNever.type_packs_containing_never_is_itself_uninhabitable
|
TypeInferUnknownNever.type_packs_containing_never_is_itself_uninhabitable
|
||||||
TypeInferUnknownNever.type_packs_containing_never_is_itself_uninhabitable2
|
TypeInferUnknownNever.type_packs_containing_never_is_itself_uninhabitable2
|
||||||
TypeInferUnknownNever.unary_minus_of_never
|
TypeInferUnknownNever.unary_minus_of_never
|
||||||
TypeInferUnknownNever.unknown_is_reflexive
|
|
||||||
TypePackTests.higher_order_function
|
TypePackTests.higher_order_function
|
||||||
TypePackTests.multiple_varargs_inference_are_not_confused
|
TypePackTests.multiple_varargs_inference_are_not_confused
|
||||||
TypePackTests.no_return_size_should_be_zero
|
TypePackTests.no_return_size_should_be_zero
|
||||||
|
@ -195,7 +195,7 @@
|
|||||||
<Expand>
|
<Expand>
|
||||||
<LinkedListItems>
|
<LinkedListItems>
|
||||||
<HeadPointer>openupval</HeadPointer>
|
<HeadPointer>openupval</HeadPointer>
|
||||||
<NextPointer>u.l.threadnext</NextPointer>
|
<NextPointer>u.open.threadnext</NextPointer>
|
||||||
<ValueNode>this</ValueNode>
|
<ValueNode>this</ValueNode>
|
||||||
</LinkedListItems>
|
</LinkedListItems>
|
||||||
</Expand>
|
</Expand>
|
||||||
|
@ -56,7 +56,63 @@ def nodeFromCallstackListFile(source_file):
|
|||||||
|
|
||||||
return root
|
return root
|
||||||
|
|
||||||
def getDuration(obj):
|
|
||||||
|
def getDuration(nodes, nid):
|
||||||
|
node = nodes[nid - 1]
|
||||||
|
total = node['TotalDuration']
|
||||||
|
|
||||||
|
for cid in node['NodeIds']:
|
||||||
|
total -= nodes[cid - 1]['TotalDuration']
|
||||||
|
|
||||||
|
return total
|
||||||
|
|
||||||
|
def getFunctionKey(fn):
|
||||||
|
return fn['Source'] + "," + fn['Name'] + "," + str(fn['Line'])
|
||||||
|
|
||||||
|
def recursivelyBuildNodeTree(nodes, functions, parent, fid, nid):
|
||||||
|
ninfo = nodes[nid - 1]
|
||||||
|
finfo = functions[fid - 1]
|
||||||
|
|
||||||
|
child = parent.child(getFunctionKey(finfo))
|
||||||
|
child.source = finfo['Source']
|
||||||
|
child.function = finfo['Name']
|
||||||
|
child.line = int(finfo['Line']) if finfo['Line'] > 0 else 0
|
||||||
|
|
||||||
|
child.ticks = getDuration(nodes, nid)
|
||||||
|
|
||||||
|
assert(len(ninfo['FunctionIds']) == len(ninfo['NodeIds']))
|
||||||
|
|
||||||
|
for i in range(0, len(ninfo['FunctionIds'])):
|
||||||
|
recursivelyBuildNodeTree(nodes, functions, child, ninfo['FunctionIds'][i], ninfo['NodeIds'][i])
|
||||||
|
|
||||||
|
return
|
||||||
|
|
||||||
|
def nodeFromJSONV2(dump):
|
||||||
|
assert(dump['Version'] == 2)
|
||||||
|
|
||||||
|
nodes = dump['Nodes']
|
||||||
|
functions = dump['Functions']
|
||||||
|
categories = dump['Categories']
|
||||||
|
|
||||||
|
root = Node()
|
||||||
|
|
||||||
|
for category in categories:
|
||||||
|
nid = category['NodeId']
|
||||||
|
node = nodes[nid - 1]
|
||||||
|
name = category['Name']
|
||||||
|
|
||||||
|
child = root.child(name)
|
||||||
|
child.function = name
|
||||||
|
child.ticks = getDuration(nodes, nid)
|
||||||
|
|
||||||
|
assert(len(node['FunctionIds']) == len(node['NodeIds']))
|
||||||
|
|
||||||
|
for i in range(0, len(node['FunctionIds'])):
|
||||||
|
recursivelyBuildNodeTree(nodes, functions, child, node['FunctionIds'][i], node['NodeIds'][i])
|
||||||
|
|
||||||
|
return root
|
||||||
|
|
||||||
|
def getDurationV1(obj):
|
||||||
total = obj['TotalDuration']
|
total = obj['TotalDuration']
|
||||||
|
|
||||||
if 'Children' in obj:
|
if 'Children' in obj:
|
||||||
@ -73,7 +129,7 @@ def nodeFromJSONObject(node, key, obj):
|
|||||||
node.source = source
|
node.source = source
|
||||||
node.line = int(line) if len(line) > 0 else 0
|
node.line = int(line) if len(line) > 0 else 0
|
||||||
|
|
||||||
node.ticks = getDuration(obj)
|
node.ticks = getDurationV1(obj)
|
||||||
|
|
||||||
if 'Children' in obj:
|
if 'Children' in obj:
|
||||||
for key, obj in obj['Children'].items():
|
for key, obj in obj['Children'].items():
|
||||||
@ -81,10 +137,8 @@ def nodeFromJSONObject(node, key, obj):
|
|||||||
|
|
||||||
return node
|
return node
|
||||||
|
|
||||||
|
def nodeFromJSONV1(dump):
|
||||||
def nodeFromJSONFile(source_file):
|
assert(dump['Version'] == 1)
|
||||||
dump = json.load(source_file)
|
|
||||||
|
|
||||||
root = Node()
|
root = Node()
|
||||||
|
|
||||||
if 'Children' in dump:
|
if 'Children' in dump:
|
||||||
@ -93,6 +147,16 @@ def nodeFromJSONFile(source_file):
|
|||||||
|
|
||||||
return root
|
return root
|
||||||
|
|
||||||
|
def nodeFromJSONFile(source_file):
|
||||||
|
dump = json.load(source_file)
|
||||||
|
|
||||||
|
if dump['Version'] == 2:
|
||||||
|
return nodeFromJSONV2(dump)
|
||||||
|
elif dump['Version'] == 1:
|
||||||
|
return nodeFromJSONV1(dump)
|
||||||
|
|
||||||
|
return Node()
|
||||||
|
|
||||||
|
|
||||||
arguments = argumentParser.parse_args()
|
arguments = argumentParser.parse_args()
|
||||||
|
|
||||||
|
@ -14,12 +14,14 @@ def loadFailList():
|
|||||||
with open(FAIL_LIST_PATH) as f:
|
with open(FAIL_LIST_PATH) as f:
|
||||||
return set(map(str.strip, f.readlines()))
|
return set(map(str.strip, f.readlines()))
|
||||||
|
|
||||||
|
|
||||||
def safeParseInt(i, default=0):
|
def safeParseInt(i, default=0):
|
||||||
try:
|
try:
|
||||||
return int(i)
|
return int(i)
|
||||||
except ValueError:
|
except ValueError:
|
||||||
return default
|
return default
|
||||||
|
|
||||||
|
|
||||||
class Handler(x.ContentHandler):
|
class Handler(x.ContentHandler):
|
||||||
def __init__(self, failList):
|
def __init__(self, failList):
|
||||||
self.currentTest = []
|
self.currentTest = []
|
||||||
@ -47,7 +49,7 @@ class Handler(x.ContentHandler):
|
|||||||
r = self.results.get(dottedName, True)
|
r = self.results.get(dottedName, True)
|
||||||
self.results[dottedName] = r and passed
|
self.results[dottedName] = r and passed
|
||||||
|
|
||||||
elif name == 'OverallResultsTestCases':
|
elif name == "OverallResultsTestCases":
|
||||||
self.numSkippedTests = safeParseInt(attrs.get("skipped", 0))
|
self.numSkippedTests = safeParseInt(attrs.get("skipped", 0))
|
||||||
|
|
||||||
def endElement(self, name):
|
def endElement(self, name):
|
||||||
@ -104,9 +106,9 @@ def main():
|
|||||||
|
|
||||||
for testName, passed in handler.results.items():
|
for testName, passed in handler.results.items():
|
||||||
if passed and testName in failList:
|
if passed and testName in failList:
|
||||||
print('UNEXPECTED: {} should have failed'.format(testName))
|
print("UNEXPECTED: {} should have failed".format(testName))
|
||||||
elif not passed and testName not in failList:
|
elif not passed and testName not in failList:
|
||||||
print('UNEXPECTED: {} should have passed'.format(testName))
|
print("UNEXPECTED: {} should have passed".format(testName))
|
||||||
|
|
||||||
if args.write:
|
if args.write:
|
||||||
newFailList = sorted(
|
newFailList = sorted(
|
||||||
@ -123,17 +125,24 @@ def main():
|
|||||||
print("Updated faillist.txt")
|
print("Updated faillist.txt")
|
||||||
|
|
||||||
if handler.numSkippedTests > 0:
|
if handler.numSkippedTests > 0:
|
||||||
print('{} test(s) were skipped! That probably means that a test segfaulted!'.format(handler.numSkippedTests), file=sys.stderr)
|
print(
|
||||||
|
"{} test(s) were skipped! That probably means that a test segfaulted!".format(
|
||||||
|
handler.numSkippedTests
|
||||||
|
),
|
||||||
|
file=sys.stderr,
|
||||||
|
)
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
|
||||||
sys.exit(
|
ok = all(
|
||||||
0
|
not passed == (dottedName in failList)
|
||||||
if all(
|
for dottedName, passed in handler.results.items()
|
||||||
not passed == (dottedName in failList)
|
|
||||||
for dottedName, passed in handler.results.items()
|
|
||||||
)
|
|
||||||
else 1
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
if ok:
|
||||||
|
print("Everything in order!", file=sys.stderr)
|
||||||
|
|
||||||
|
sys.exit(0 if ok else 1)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
Loading…
Reference in New Issue
Block a user