Method AddFunc

source/script.cpp:6827–6935

UserFunc *Script::AddFunc(LPCTSTR aFuncName, size_t aFuncNameLength, Object *aClassObject)
// Returns the address of the new function or NULL on failure.
// The caller must already have verified that this isn't a duplicate function.
{
	if (aFuncNameLength == -1) // Caller didn't specify, so use the entire string.
		aFuncNameLength = _tcslen(aFuncName);

	if (aFuncNameLength > MAX_VAR_NAME_LENGTH) // FindFunc(), BIF_OnMessage() and perhaps others rely on this limit being enforced.
	{
		ScriptError(_T("Function name too long."), aFuncName);
		return nullptr;
	}

	// ValidateName requires that the name be null-terminated, but it isn't in this case.
	// Doing this first saves doing tcslcpy() into a temporary buffer, and won't leak memory
	// since the script currently always exits if an error occurs anywhere below:
	LPTSTR new_name = SimpleHeap::Alloc(aFuncName, aFuncNameLength);

	if (!aClassObject && *new_name && !Var::ValidateName(new_name, DISPLAY_FUNC_ERROR))  // Variable and function names are both validated the same way.
		return nullptr; // Above already displayed the error for us.

	auto the_new_func = new UserFunc(new_name);
	if (!the_new_func)
	{
		ScriptError(ERR_OUTOFMEM);
		return nullptr;
	}

	if (aClassObject)
	{
		LPTSTR key = _tcsrchr(new_name, '.'); // DefineFunc() always passes "ClassName.MethodName".
		++key;
		if (!Var::ValidateName(key, DISPLAY_METHOD_ERROR))
			return nullptr;
		if (mClassProperty)
		{
			bool is_getter = ctoupper(*key) == 'G';
			if (is_getter ? mClassProperty->Getter() : mClassProperty->Setter())
			{
				ScriptError(ERR_DUPLICATE_DECLARATION, new_name);
				return nullptr;
			}
			if (is_getter)
				mClassProperty->SetGetter(the_new_func);
			else
				mClassProperty->SetSetter(the_new_func);
		}
		else
		{
			switch (aClassObject->GetOwnPropType(key))
			{
			case Object::PropType::Object:
			case Object::PropType::DynamicMethod: // call
			case Object::PropType::DynamicMixed: // get/set and call
				ScriptError(ERR_DUPLICATE_DECLARATION, new_name);
				return nullptr;
			}
			if (!aClassObject->DefineMethod(key, the_new_func))