myvtm/Module/mod_systemcustomization/systemcustomization_server_g.h

#ifndef __SYSTEMCUSTOMIZATION_SERVER_G_H
#define __SYSTEMCUSTOMIZATION_SERVER_G_H

#pragma once

// This code is generated by spgen tool!

#include "systemcustomization_def_g.h"

namespace systemcustomization {
class systemcustomizationInterface_ServerSessionBase : public CServerSessionBase
{
public:
	systemcustomizationInterface_ServerSessionBase() { }

	virtual ~systemcustomizationInterface_ServerSessionBase() { }

	virtual bool IsExclusive() { return false; }

	virtual bool IsSessionOverlap() { return true; }

	virtual ErrorCodeEnum GetMessageAttr(DWORD dwMessageID, DWORD dwSignature, bool &bOverlap)
	{
		ErrorCodeEnum Error = Error_Succeed;
		switch (dwMessageID) {
		case systemcustomizationInterface_Method_UpdateMD5List:
			if (dwSignature == systemcustomizationInterface_MethodSignature_UpdateMD5List) {
				bOverlap = true;
			} else {
				Error = Error_MethodSignatureFailed;
			}
			break;
		case systemcustomizationInterface_Method_VerifyCodeSign:
			if (dwSignature == systemcustomizationInterface_MethodSignature_VerifyCodeSign) {
				bOverlap = true;
			} else {
				Error = Error_MethodSignatureFailed;
			}
			break;
		case systemcustomizationInterface_Method_InstallPatch:
			if (dwSignature == systemcustomizationInterface_MethodSignature_InstallPatch) {
				bOverlap = true;
			} else {
				Error = Error_MethodSignatureFailed;
			}
			break;
		default:
			Error = Error_MethodNotFound;
			break;
		}
		return Error;
	}

	int CheckMessageSignature(DWORD dwMessageID, DWORD dwSignature)
	{
		ErrorCodeEnum Error = Error_Succeed;
		switch (dwMessageID) {
		case systemcustomizationInterface_Method_UpdateMD5List:
			if (dwSignature != systemcustomizationInterface_MethodSignature_UpdateMD5List) {
				Error = Error_MethodSignatureFailed;
			}
			break;
		case systemcustomizationInterface_Method_VerifyCodeSign:
			if (dwSignature != systemcustomizationInterface_MethodSignature_VerifyCodeSign) {
				Error = Error_MethodSignatureFailed;
			}
			break;
		case systemcustomizationInterface_Method_InstallPatch:
			if (dwSignature != systemcustomizationInterface_MethodSignature_InstallPatch) {
				Error = Error_MethodSignatureFailed;
			}
			break;
		default:
			Error = Error_MethodNotFound;
			break;
		}
		return Error;
	}

	virtual void Handle_UpdateMD5List(SpReqAnsContext<systemcustomizationInterface_UpdateMD5List_Req, systemcustomizationInterface_UpdateMD5List_Ans>::Pointer ctx)
	{
	/// override by user
	}

	virtual void Handle_VerifyCodeSign(SpReqAnsContext<systemcustomizationInterface_VerifyCodeSign_Req, systemcustomizationInterface_VerifyCodeSign_Ans>::Pointer ctx)
	{
	/// override by user
	}

	virtual void Handle_InstallPatch(SpReqAnsContext<systemcustomizationInterface_InstallPatch_Req, systemcustomizationInterface_InstallPatch_Ans>::Pointer ctx)
	{
	/// override by user
	}

	virtual void OnRequest(CSmartPointer<ITransactionContext> pTransactionContext)
	{
		CAutoBuffer Buf;
		DWORD dwMessageID;
		DWORD dwMessageSignature;
		ErrorCodeEnum Error = pTransactionContext->GetReceiveBuffer(dwMessageID, dwMessageSignature, Buf);
		if (Error == Error_Succeed) {
#ifdef DEBUG
			assert(CheckMessageSignature(dwMessageID, dwMessageSignature) == Error_Succeed);
#else
			if (CheckMessageSignature(dwMessageID, dwMessageSignature) != Error_Succeed) {
				pTransactionContext->SendAnswer(Error_MethodSignatureFailed);
				return;
			}
#endif
			switch (dwMessageID) {
				case systemcustomizationInterface_Method_UpdateMD5List:
					{
						SpReqAnsContext<systemcustomizationInterface_UpdateMD5List_Req,systemcustomizationInterface_UpdateMD5List_Ans>::Pointer ctx;
						ctx.Attach(new SpReqAnsContext<systemcustomizationInterface_UpdateMD5List_Req,systemcustomizationInterface_UpdateMD5List_Ans>(pTransactionContext));
						SpBuffer2Object(Buf, ctx->Req);
						Handle_UpdateMD5List(ctx);
					}
					break;
				case systemcustomizationInterface_Method_VerifyCodeSign:
					{
						SpReqAnsContext<systemcustomizationInterface_VerifyCodeSign_Req,systemcustomizationInterface_VerifyCodeSign_Ans>::Pointer ctx;
						ctx.Attach(new SpReqAnsContext<systemcustomizationInterface_VerifyCodeSign_Req,systemcustomizationInterface_VerifyCodeSign_Ans>(pTransactionContext));
						SpBuffer2Object(Buf, ctx->Req);
						Handle_VerifyCodeSign(ctx);
					}
					break;
				case systemcustomizationInterface_Method_InstallPatch:
					{
						SpReqAnsContext<systemcustomizationInterface_InstallPatch_Req,systemcustomizationInterface_InstallPatch_Ans>::Pointer ctx;
						ctx.Attach(new SpReqAnsContext<systemcustomizationInterface_InstallPatch_Req,systemcustomizationInterface_InstallPatch_Ans>(pTransactionContext));
						SpBuffer2Object(Buf, ctx->Req);
						Handle_InstallPatch(ctx);
					}
					break;
				default:
					assert(0);
					break;
			}
			
		} else {
			pTransactionContext->SendAnswer(Error);
		}
	}

};


///////////////////////////

} // namespace systemcustomization
#endif // __SYSTEMCUSTOMIZATION_SERVER_G_H