mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-11-05 08:35:26 +00:00
119 lines
3.4 KiB
C++
119 lines
3.4 KiB
C++
//* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#ifndef ProtocolParser_h__
|
|
#define ProtocolParser_h__
|
|
|
|
#include "HashStore.h"
|
|
#include "nsICryptoHMAC.h"
|
|
|
|
namespace mozilla {
|
|
namespace safebrowsing {
|
|
|
|
/**
|
|
* Some helpers for parsing the safe
|
|
*/
|
|
class ProtocolParser {
|
|
public:
|
|
struct ForwardedUpdate {
|
|
nsCString table;
|
|
nsCString url;
|
|
nsCString mac;
|
|
};
|
|
|
|
ProtocolParser(uint32_t aHashKey);
|
|
~ProtocolParser();
|
|
|
|
nsresult Status() const { return mUpdateStatus; }
|
|
|
|
nsresult Init(nsICryptoHash* aHasher, bool mPerClientRandomize);
|
|
|
|
nsresult InitHMAC(const nsACString& aClientKey,
|
|
const nsACString& aServerMAC);
|
|
nsresult FinishHMAC();
|
|
|
|
void SetCurrentTable(const nsACString& aTable);
|
|
|
|
nsresult Begin();
|
|
nsresult AppendStream(const nsACString& aData);
|
|
|
|
// Forget the table updates that were created by this pass. It
|
|
// becomes the caller's responsibility to free them. This is shitty.
|
|
TableUpdate *GetTableUpdate(const nsACString& aTable);
|
|
void ForgetTableUpdates() { mTableUpdates.Clear(); }
|
|
nsTArray<TableUpdate*> &GetTableUpdates() { return mTableUpdates; }
|
|
|
|
// Update information.
|
|
const nsTArray<ForwardedUpdate> &Forwards() const { return mForwards; }
|
|
int32_t UpdateWait() { return mUpdateWait; }
|
|
bool ResetRequested() { return mResetRequested; }
|
|
bool RekeyRequested() { return mRekeyRequested; }
|
|
|
|
private:
|
|
nsresult ProcessControl(bool* aDone);
|
|
nsresult ProcessMAC(const nsCString& aLine);
|
|
nsresult ProcessExpirations(const nsCString& aLine);
|
|
nsresult ProcessChunkControl(const nsCString& aLine);
|
|
nsresult ProcessForward(const nsCString& aLine);
|
|
nsresult AddForward(const nsACString& aUrl, const nsACString& aMac);
|
|
nsresult ProcessChunk(bool* done);
|
|
nsresult ProcessPlaintextChunk(const nsACString& aChunk);
|
|
nsresult ProcessShaChunk(const nsACString& aChunk);
|
|
nsresult ProcessHostAdd(const Prefix& aDomain, uint8_t aNumEntries,
|
|
const nsACString& aChunk, uint32_t* aStart);
|
|
nsresult ProcessHostSub(const Prefix& aDomain, uint8_t aNumEntries,
|
|
const nsACString& aChunk, uint32_t* aStart);
|
|
nsresult ProcessHostAddComplete(uint8_t aNumEntries, const nsACString& aChunk,
|
|
uint32_t *aStart);
|
|
nsresult ProcessHostSubComplete(uint8_t numEntries, const nsACString& aChunk,
|
|
uint32_t* start);
|
|
bool NextLine(nsACString& aLine);
|
|
|
|
void CleanupUpdates();
|
|
|
|
enum ParserState {
|
|
PROTOCOL_STATE_CONTROL,
|
|
PROTOCOL_STATE_CHUNK
|
|
};
|
|
ParserState mState;
|
|
|
|
enum ChunkType {
|
|
CHUNK_ADD,
|
|
CHUNK_SUB
|
|
};
|
|
|
|
struct ChunkState {
|
|
ChunkType type;
|
|
uint32_t num;
|
|
uint32_t hashSize;
|
|
uint32_t length;
|
|
void Clear() { num = 0; hashSize = 0; length = 0; }
|
|
};
|
|
ChunkState mChunkState;
|
|
|
|
uint32_t mHashKey;
|
|
bool mPerClientRandomize;
|
|
nsCOMPtr<nsICryptoHash> mCryptoHash;
|
|
|
|
nsresult mUpdateStatus;
|
|
nsCString mPending;
|
|
|
|
nsCOMPtr<nsICryptoHMAC> mHMAC;
|
|
nsCString mServerMAC;
|
|
|
|
uint32_t mUpdateWait;
|
|
bool mResetRequested;
|
|
bool mRekeyRequested;
|
|
|
|
nsTArray<ForwardedUpdate> mForwards;
|
|
nsTArray<TableUpdate*> mTableUpdates;
|
|
TableUpdate *mTableUpdate;
|
|
};
|
|
|
|
}
|
|
}
|
|
|
|
#endif
|