Example #1
0
tsc_result TSC_CALLTYPE
tscGetVersion( TSC_VERSION_DATA* pData )
{
    tsc_func_init();

    CModule* pM = CModule::GetInstance();
    if (!pM)
        return Error_Internal_NullModulePtr();
    if (!pData)
        return Error_ParamWasNull();
    if (pData->cbSize != sizeof(TSC_VERSION_DATA))
        return Error_StructSizeInvalid();

    CVersionData d(*pData);

    tsc_result r = TSC_E_FAIL;
    r = pM->GetVersion(d);

    if (TSC_SUCCEEDED(r))
    {
        *pData << d;
    }

    return r;
}
Example #2
0
IModule * CreateModule(const IRepository *rep, const std::_tstring &label, const ECLModule * data)
{
    CModule * mod = CreateModuleRaw(rep, label.c_str(), false);
    ATLASSERT(mod);
    mod->Update(data);
    return mod;
}
Example #3
0
bool CUser::UpdateModule(const CString &sModule) {
	const map<CString,CUser*>& Users = CZNC::Get().GetUserMap();
	map<CString,CUser*>::const_iterator it;
	map<CUser*, CString> Affected;
	map<CUser*, CString>::iterator it2;
	bool error = false;

	for (it = Users.begin(); it != Users.end(); ++it) {
		CModule *pMod = it->second->GetModules().FindModule(sModule);
		if (pMod) {
			Affected[it->second] = pMod->GetArgs();
			it->second->GetModules().UnloadModule(pMod->GetModName());
		}
	}

	CString sErr;
	for (it2 = Affected.begin(); it2 != Affected.end(); ++it2) {
		if (!it2->first->GetModules().LoadModule(sModule, it2->second, it2->first, sErr)) {
			error = true;
			DEBUG("Failed to reload [" << sModule << "] for [" << it2->first->GetUserName()
					<< "]: " << sErr);
		}
	}

	return !error;
}
Example #4
0
const char *WINAPI MGetMsg(int PluginNumber,int MsgId)
{
  CModule *module = (CModule *)PluginNumber;
  if (!module)
    return "";
  return module->GetMsg(MsgId);
}
Example #5
0
bool CModules::UnloadModule(const CString& sModule, CString& sRetMsg) {
    CString sMod = sModule;  // Make a copy incase the reference passed in is from CModule::GetModName()
    CModule* pModule = FindModule(sMod);
    sRetMsg = "";

    if (!pModule) {
        sRetMsg = "Module [" + sMod + "] not loaded.";
        return false;
    }

    bool bSuccess;
    GLOBALMODULECALL(OnModuleUnloading(pModule, bSuccess, sRetMsg), pModule->GetUser(), NULL, return bSuccess);

    ModHandle p = pModule->GetDLL();

    if (p) {
        delete pModule;

        for (iterator it = begin(); it != end(); ++it) {
            if (*it == pModule) {
                erase(it);
                break;
            }
        }

        dlclose(p);
        sRetMsg = "Module [" + sMod + "] unloaded";

        return true;
    }

    sRetMsg = "Unable to unload module [" + sMod + "]";
    return false;
}
Example #6
0
void CWell::HandleCollision(IEntity* other)
{	
	SGD::Vector dir;
	if ( tier >= 3 && (duration - timer) < 0.1f)
	{
		dir = other->GetPosition() - position;
		dir.Normalize();
		other->AddGravity(dir * strength * 2);
		EntityType type = (EntityType)other->GetType();
		if (type == EntityType::Asteroid)
		{
			CAsteroid* ast = dynamic_cast<CAsteroid*>(other);
			ast->TakeDamage(int(strength * 0.075f));
		}
		else if (type >= EntityType::Player && type <= EntityType::Coordinator)
		{
			CShip* ship = dynamic_cast<CShip*>(other);
			ship->TakeDamage(int(strength * 0.075f));
		}
		else if (type >= EntityType::BaseModule && type <= EntityType::WarpModule)
		{
			CModule* mod = dynamic_cast<CModule*>(other);
			mod->TakeDamage(int(strength * 0.075f));
		}
		return;
	}
	dir = position - other->GetPosition();
	if (dir.ComputeLength() < 1)
		dir.y += 1;
	dir.Normalize();
	//float mass = other->GetSize().width * other->GetSize().height;
	other->AddGravity(dir * strength);
}
Example #7
0
int main(int argc, char *argv[])
{
  int i = 1;

  while (i < argc) {
    // scanning, parsing & semantical analysis
    CScanner *s = new CScanner(new ifstream(argv[i]));
    CParser *p = new CParser(s);

    cout << "parsing '" << argv[i] << "'..." << endl;
    CAstNode *ast = p->Parse();

    if (p->HasError()) {
      const CToken *error = p->GetErrorToken();
      cout << "parse error : at " << error->GetLineNumber() << ":"
           << error->GetCharPosition() << " : "
           << p->GetErrorMessage() << endl;
    } else {
      // AST to TAC conversion
      cout << "converting to TAC..." << endl;
      CModule *m = new CModule(ast);

      // print TAC to console
      cout << m << endl;
      cout << endl;

      // output TAC as .dot and generate a PDF file from it
      ofstream out(string(argv[i]) + ".dot");
      out << "digraph IR {" << endl
          << "  graph [fontname=\"Times New Roman\",fontsize=10];" << endl
          << "  node  [fontname=\"Courier New\",fontsize=10];" << endl
          << "  edge  [fontname=\"Times New Roman\",fontsize=10];" << endl
          << endl;
      m->toDot(out, 2);
      const vector<CScope*> &proc = m->GetSubscopes();
      for (size_t p=0; p<proc.size(); p++) {
        proc[p]->toDot(out, 2);
      }
      out << "}" << endl;
      out.flush();

      ostringstream cmd;
      cmd << "dot -Tpdf -o" << argv[i] << ".pdf " << argv[i] << ".dot";
      cout << "run the following command to convert the .dot file into a PDF:" << endl
           << "  " << cmd.str() << 
           endl;

      delete m;
    }

    cout << endl << endl;

    i++;
  }

  cout << "Done." << endl;

  return EXIT_SUCCESS;
}
Example #8
0
void tsc_func_init()
{
    s_szLastError = s_szErrNoErr;

    CModule* pM = CModule::GetInstance();
    if (pM)
        pM->ClearLastError();
}
Example #9
0
void CAppSignal::OnWorkspace(class CUIWorkspace* pUIWorkspace, const POINT& pt)
{
	EnterCriticalSection(&m_csControls);
	{
		if ( pUIWorkspace == m_pUIWorkspace )
		{
			POINT ptScreen = { pt.x, pt.y };
			ClientToScreen(pUIWorkspace->GetHwnd(), &ptScreen);

			HMENU hMenu = CreatePopupMenu();

			map<string, CFactory<CModule>::FACTORYSTRUCT*>& mapFactory = CFactory<CModule>::GetFactoryMap();
			for ( map<string, CFactory<CModule>::FACTORYSTRUCT*>::iterator iter = mapFactory.begin() ; iter != mapFactory.end() ; ++iter )
			{
				MODULEFACTORYDATA* pmfd = ((MODULEFACTORYDATA*)iter->second->pvData);
				if ( !pmfd ) continue;
				AppendMenu(hMenu, MF_STRING, (uint32)iter->second, pmfd->m_sCategory.c_str());
			}

			uint32 nResult = TrackPopupMenu(hMenu, TPM_RETURNCMD|TPM_LEFTALIGN|TPM_LEFTBUTTON, ptScreen.x, ptScreen.y, 0, GetHwnd(), NULL);
			if ( nResult != 0 )
			{
				CFactory<CModule>::FACTORYSTRUCT* pfs = (CFactory<CModule>::FACTORYSTRUCT*)nResult;
				MODULEFACTORYDATA* pmfd = ((MODULEFACTORYDATA*)pfs->pvData);
				
				CUIModule* pUIModule = (CUIModule*)CFactory<CUIWindow>::Create("CUIModule");
				CModule* pModule = (CModule*)pfs->fnCreate();

				pUIModule->SetModule(pModule);
				pModule->SetUIModule(pUIModule);

				pUIModule->SetParent(m_pUIWorkspace);
				m_pUIWorkspace->AddChild(pUIModule);

				MODULEINITSTRUCT mis;
				mis.pApp = this;

				pModule->Init(mis);
				pModule->PostInit();
				
				CUIWindow::CREATESTRUCT cs;
				cs.x = pt.x;
				cs.y = pt.y;
				cs.hWndParent = m_pUIWorkspace->GetHwnd();
				cs.pUIHost = pModule;
				cs.dict.SetString("title", pmfd->m_sCategory);
				pUIModule->Create(cs);

				m_listModules.push_back(pModule);
				m_listUIModules.push_back(pUIModule);
			}

			DestroyMenu(hMenu);
		}
	}
	LeaveCriticalSection(&m_csControls);
}
Example #10
0
IModule * CreateModule(const IRepository *rep, const ns2__ECLModule * data, bool noBroadcast)
{
    ATLASSERT(data->Name);
    ATLASSERT(!data->Name->empty());
    CModule * mod = CreateModuleRaw(rep, data->Name->c_str(), false);
    ATLASSERT(mod);
    mod->Update(data, noBroadcast);
    return mod;
}
Example #11
0
// Why MODHALTCHK works only with functions returning EModRet ? :(
bool CModules::OnServerCapAvailable(const CString& sCap) {
	bool bResult = false;
	for (unsigned int a = 0; a < size(); ++a) {
		try {
			CModule* pMod = (*this)[a];
			CClient* pOldClient = pMod->GetClient();
			pMod->SetClient(m_pClient);
			if (m_pUser) {
				CUser* pOldUser = pMod->GetUser();
				pMod->SetUser(m_pUser);
				bResult |= pMod->OnServerCapAvailable(sCap);
				pMod->SetUser(pOldUser);
			} else {
				// WTF? Is that possible?
				bResult |= pMod->OnServerCapAvailable(sCap);
			}
			pMod->SetClient(pOldClient);
		} catch (CModule::EModException e) {
			if (CModule::UNLOAD == e) {
				UnloadModule((*this)[a]->GetModName());
			}
		}
	}
	return bResult;
}
Example #12
0
File: Modules.cpp Project: SjB/znc
// Maybe create new macro for this?
bool CModules::IsClientCapSupported(CClient* pClient, const CString& sCap, bool bState) {
	bool bResult = false;
	for (unsigned int a = 0; a < size(); ++a) {
		try {
			CModule* pMod = (CModule*) (*this)[a];
			CClient* pOldClient = pMod->GetClient();
			pMod->SetClient(m_pClient);
			if (m_pUser) {
				CUser* pOldUser = pMod->GetUser();
				pMod->SetUser(m_pUser);
				bResult |= pMod->IsClientCapSupported(pClient, sCap, bState);
				pMod->SetUser(pOldUser);
			} else {
				// WTF? Is that possible?
				bResult |= pMod->IsClientCapSupported(pClient, sCap, bState);
			}
			pMod->SetClient(pOldClient);
		} catch (CModule::EModException e) {
			if (CModule::UNLOAD == e) {
				UnloadModule((*this)[a]->GetModName());
			}
		}
	}
	return bResult;
}
Example #13
0
bool CModules::UnloadModule(const CString& sModule, CString& sRetMsg) {
	CString sMod = sModule;	// Make a copy incase the reference passed in is from CModule::GetModName()
	CModule* pModule = FindModule(sMod);
	sRetMsg = "";

	if (!pModule) {
		sRetMsg = "Module [" + sMod + "] not loaded.";
		return false;
	}

	if (pModule->IsFake()) {
		for (iterator it = begin(); it != end(); it++) {
			if (*it == pModule) {
				erase(it);
				sRetMsg = "Fake module [" + sMod + "] unloaded";
				return true;
			}
		}

		sRetMsg = "Fake module [" + sMod + "] not loaded.";
		return false;
	}

	ModHandle p = pModule->GetDLL();

	if (p) {
		typedef void (*fp)(CModule*);
		fp Unload = (fp)dlsym(p, "ZNCModUnload");

		if (Unload) {
			Unload(pModule);

			for (iterator it = begin(); it != end(); it++) {
				if (*it == pModule) {
					erase(it);
					break;
				}
			}

			dlclose(p);
			sRetMsg = "Module [" + sMod + "] unloaded";

			return true;
		} else {
			sRetMsg = "Unable to unload module [" + sMod + "] could not find ZNCModUnload()";
			return false;
		}
	}

	sRetMsg = "Unable to unload module [" + sMod + "]";
	return false;
}
Example #14
0
void CSession::FreeCurrentDictionary()
{
    if (m_pEnchantDict)
    {
        CModule* pMod = CModule::GetInstance();

        if (pMod)
        {
            pMod->FreeDictionary(m_pEnchantDict);
            m_pEnchantDict = NULL;
        }
    }
}
Example #15
0
tsc_result TSC_CALLTYPE
tscDestroySession( tsc_cookie SessionID )
{
    tsc_func_init();

    CModule* pM = CModule::GetInstance();
    if (!pM)
        return Error_Internal_NullModulePtr();

    tsc_result r = TSC_E_FAIL;
    r = pM->DestroySession(SessionID);

    return r;
}
Example #16
0
const char* TSC_CALLTYPE
tscGetLastError()
{
    if (s_szLastError != s_szErrNoErr)
        return s_szLastError;
    else
    {
        CModule* pM = CModule::GetInstance();
        if (pM)
            return pM->GetLastError();
        else
            return s_szErrErr; // catch-all error text
    }
}
Example #17
0
void CHijackChecker::check(CSysConfig& conf, CModule& mod)
{
	WStrContainer libs;
	WStrContainer::iterator it;

	mod.get_import_libs(libs);
	mod.get_module_path(m_mod_dir);
	conf.get_search_dirs(m_search_dirs);

	for (it = libs.begin(); it != libs.end(); it++) {
		wstring& lib = *it;
		check_module(conf, lib, true);
	}
}
Example #18
0
	CUser* GetUser(const CString& username) {
		if (username.Equals("$me"))
			return m_pUser;
		CUser *pUser = CZNC::Get().FindUser(username);
		if (!pUser) {
			PutModule("Error: User not found: " + username);
			return NULL;
		}
		if (pUser != m_pUser && !m_pUser->IsAdmin()) {
			PutModule("Error: You need to have admin rights to modify other users!");
			return NULL;
		}
		return pUser;
	}
Example #19
0
// rp_get_symbol(lib, const symbol[])
static cell AMX_NATIVE_CALL rp_get_symbol(AMX *amx, cell *params)
{
	CModule* mod = (CModule *)params[1];
	int len;
	const char* symbol = g_amxxapi.GetAmxString(amx, params[2], 0, &len);

	if (!mod)
	{
		MF_LogError(amx, AMX_ERR_NATIVE, "Invalid library handle provided in rp_get_symbol(0, \"%s\")", symbol);
		return NULL;
	}

	return (cell)mod->getSymbolAddress(symbol);
}
Example #20
0
// rp_find_signature(lib, const symbol[])
static cell AMX_NATIVE_CALL rp_find_signature(AMX *amx, cell *params)
{
	CModule* mod = (CModule *)params[1];
	int len;
	const char* sig = g_amxxapi.GetAmxString(amx, params[2], 0, &len);

	if (!mod)
	{
		MF_LogError(amx, AMX_ERR_NATIVE, "Invalid library handle provided in rp_find_signature(0, \"%s\")", sig);
		return NULL;
	}

	return (cell)mod->findPattern(sig, PARAMS_COUNT == 3 && params[3]);
}
Example #21
0
tsc_result TSC_CALLTYPE
tscUninit( void )
{
    tsc_func_init();

    CModule* pM = CModule::GetInstance();
    if (!pM)
        return Error_Internal_NullModulePtr();

    tsc_result r = TSC_E_FAIL;
    r = pM->Uninit();

    return r;
}
Example #22
0
CModule * CDsnInfo::AddModule(string name, EClkRate clkRate)
{
	m_modList.push_back(new CModule(name, clkRate, false));

	CModule * pModule = m_modList.back();

	if (g_appArgs.GetIqModName() == name)
		pModule->AddQueIntf("hif", "hifInQue", Pop, "5");

	if (g_appArgs.GetOqModName() == name)
		pModule->AddQueIntf("hif", "hifOutQue", Push, "5");

	return pModule;
}
Example #23
0
	void SendNotification(CModule& module, const CString& sSender, const CString& sNotification, const CChan *pChannel) {
		// todo parse from m_sPushEndpoint
		bool bUseTLS = true;
		CString sHostname = "api.palaverapp.com";
		unsigned short uPort = 443;
		CString sPath = "/1/push";

		CString sJSON = "{";
		sJSON += "\"message\": \"" + sNotification.Replace_n("\"", "\\\"") + "\"";
		sJSON += ",\"sender\": \"" + sSender.Replace_n("\"", "\\\"") + "\"";
		if (pChannel) {
			sJSON += ",\"channel\": \"" + pChannel->GetName().Replace_n("\"", "\\\"") + "\"";
		}
		sJSON += "}";

		CSocket *pSocket = new CSocket(&module);
		pSocket->Connect(sHostname, uPort, bUseTLS);
		pSocket->Write("POST " + sPath + " HTTP/1.1\r\n");
		pSocket->Write("Host: " + sHostname + "\r\n");
		pSocket->Write("Authorization: Bearer " + GetToken() + "\r\n");
		pSocket->Write("Connection: close\r\n");
		pSocket->Write("User-Agent: ZNC\r\n");
		pSocket->Write("Content-Type: application/json\r\n");
		pSocket->Write("Content-Length: " + CString(sJSON.length()) + "\r\n");
		pSocket->Write("\r\n");
		pSocket->Write(sJSON);
		pSocket->Close(Csock::CLT_AFTERWRITE);
		module.AddSocket(pSocket);
	}
Example #24
0
	void ListUsers(const CString&) {
		if (!m_pUser->IsAdmin())
			return;

		const map<CString, CUser*>& msUsers = CZNC::Get().GetUserMap();
		CTable Table;
		Table.AddColumn("Username");
		Table.AddColumn("Realname");
		Table.AddColumn("IsAdmin");
		Table.AddColumn("Nick");
		Table.AddColumn("AltNick");
		Table.AddColumn("Ident");
		Table.AddColumn("VHost");

		for (map<CString, CUser*>::const_iterator it = msUsers.begin(); it != msUsers.end(); it++) {
			Table.AddRow();
			Table.SetCell("Username", it->first);
			Table.SetCell("Realname", it->second->GetRealName());
			if (!it->second->IsAdmin())
				Table.SetCell("IsAdmin", "No");
			else
				Table.SetCell("IsAdmin", "Yes");
			Table.SetCell("Nick", it->second->GetNick());
			Table.SetCell("AltNick", it->second->GetAltNick());
			Table.SetCell("Ident", it->second->GetIdent());
			Table.SetCell("VHost", it->second->GetVHost());
		}

		PutModule(Table);
	}
Example #25
0
	void AddServer(const CString& sLine) {
		CString username = sLine.Token(1);
		CString server   = sLine.Token(2, true);

		if (server.empty()) {
			PutModule("Usage: addserver <username> <server>");
			return;
		}

		CUser* user = GetUser(username);
		if (!user)
			return;

		user->AddServer(server);
		PutModule("Added IRC Server: " + server);
	}
Example #26
0
File: alias.cpp Project: 54lman/znc
	// read an IRC line and do token substitution
	// throws an exception if a required parameter is missing, and might also throw if you manage to make it bork
	CString Imprint(CString line) const
	{
		CString output;
		CString alias_data = GetCommands();
		alias_data = parent->ExpandString(alias_data);
		size_t lastfound = 0, skip = 0;

		// it would be very inefficient to attempt to blindly replace every possible token
		// so let's just parse the line and replace when we find them
		// token syntax:
		// %[?]n[+]%
		// adding ? makes the substitution optional (you'll get "" if there are insufficient tokens, otherwise the alias will fail)
		// adding + makes the substitution contain all tokens from the nth to the end of the line
		while (true)
		{
			// if (found >= (int) alias_data.length()) break; 		// shouldn't be possible.
			size_t found = alias_data.find("%", lastfound + skip);
			if (found == CString::npos) break; 				// if we found nothing, break
			output.append(alias_data.substr(lastfound, found - lastfound));	// capture everything between the last stopping point and here
			ParseToken(alias_data, line, output, found, skip);				// attempt to read a token, updates indices based on success/failure
			lastfound = found;
		}

		output += alias_data.substr(lastfound); // append from the final 
		return output;
	}
Example #27
0
tsc_result TSC_CALLTYPE
tscCheckSpelling( tsc_cookie SessionID,
                  TSC_CHECKSPELLING_DATA* pData )
{
    tsc_func_init();

    CModule* pM = CModule::GetInstance();
    if (!pM)
        return Error_Internal_NullModulePtr();
    if (!pData)
        return Error_ParamWasNull();

    tsc_result r = TSC_E_FAIL;
    r = pM->CheckSpelling(SessionID, pData);

    return r;
}
Example #28
0
void CDsnInfo::AddHostData(void * pHandle, HtdFile::EHostMsgDir msgDir, bool bMaxBw)
{
	CModule * pModule = (CModule *)pHandle;

	switch (msgDir) {
	case HtdFile::Inbound:
		m_modList[0]->AddQueIntf(pModule->m_modName.AsStr(), "ihd", Push, "5", HtdFile::eDistRam, bMaxBw);
		pModule->AddQueIntf("hif", "ihd", Pop, "5", HtdFile::eDistRam, bMaxBw);
		break;
	case HtdFile::Outbound:
		m_modList[0]->AddQueIntf(pModule->m_modName.AsStr(), "ohd", Pop, "5");
		pModule->AddQueIntf("hif", "ohd", Push, "5");
		break;
	default:
		break;
	}
}
Example #29
0
	CString GetModArgs(CUser* pUser, const CString& sModName, bool bGlobal = false) {
		if (!bGlobal && !pUser) {
			return "";
		}

		CModules& Modules = (bGlobal) ? CZNC::Get().GetModules() : pUser->GetModules();

		for (unsigned int a = 0; a < Modules.size(); a++) {
			CModule* pModule = Modules[a];

			if (pModule->GetModName() == sModName) {
				return pModule->GetArgs();
			}
		}

		return "";
	}
Example #30
0
tsc_result TSC_CALLTYPE
tscCustomDic( tsc_cookie SessionID,
              TSC_CUSTOMDIC_DATA* pData )
{
    tsc_func_init();

    CModule* pM = CModule::GetInstance();
    if (!pM)
        return Error_Internal_NullModulePtr();
    if (!pData)
        return Error_ParamWasNull();

    tsc_result r = TSC_E_FAIL;
    r = pM->CustomDic(SessionID, pData);

    return r;
}