1 files changed, 71 insertions, 0 deletions
diff --git a/lib/backupstore/BackupProtocol.h b/lib/backupstore/BackupProtocol.h
new file mode 100644
index 00000000..d9070c73
--- /dev/null
+++ b/lib/backupstore/BackupProtocol.h
@@ -0,0 +1,71 @@
+// --------------------------------------------------------------------------
+//
+// File
+//		Name:    BackupProtocol.h
+//		Purpose: A thin wrapper around autogen_BackupProtocol.h
+//		Created: 2014/01/05
+//
+// --------------------------------------------------------------------------
+
+#ifndef BACKUPPROTOCOL__H
+#define BACKUPPROTOCOL__H
+
+#include <autogen_BackupProtocol.h>
+#include <BackupStoreConstants.h>
+#include <BackupStoreContext.h>
+
+// --------------------------------------------------------------------------
+//
+// Class
+//		Name:    BackupProtocolLocal2
+//		Purpose: BackupProtocolLocal with a few more IQ points
+//		Created: 2014/09/20
+//
+// --------------------------------------------------------------------------
+class BackupProtocolLocal2 : public BackupProtocolLocal
+{
+private:
+	BackupStoreContext mContext;
+	int32_t mAccountNumber;
+	bool mReadOnly;
+
+protected:
+	BackupStoreContext& GetContext() { return mContext; }
+
+public:
+	BackupProtocolLocal2(int32_t AccountNumber,
+		const std::string& ConnectionDetails,
+		const std::string& AccountRootDir, int DiscSetNumber,
+		bool ReadOnly)
+	// This is rather ugly: the BackupProtocolLocal constructor must not
+	// touch the Context, because it's not initialised yet!
+	: BackupProtocolLocal(mContext),
+	  mContext(AccountNumber, (HousekeepingInterface *)NULL,
+		ConnectionDetails),
+	  mAccountNumber(AccountNumber),
+	  mReadOnly(ReadOnly)
+	{
+		mContext.SetClientHasAccount(AccountRootDir, DiscSetNumber);
+		QueryVersion(BACKUP_STORE_SERVER_VERSION);
+		QueryLogin(AccountNumber,
+			ReadOnly ? BackupProtocolLogin::Flags_ReadOnly : 0);
+	}
+	virtual ~BackupProtocolLocal2() { }
+
+	std::auto_ptr<BackupProtocolFinished> Query(const BackupProtocolFinished &rQuery)
+	{
+		std::auto_ptr<BackupProtocolFinished> finished =
+			BackupProtocolLocal::Query(rQuery);
+		mContext.ReleaseWriteLock();
+		return finished;
+	}
+
+	void Reopen()
+	{
+		QueryVersion(BACKUP_STORE_SERVER_VERSION);
+		QueryLogin(mAccountNumber,
+			mReadOnly ? BackupProtocolLogin::Flags_ReadOnly : 0);
+	}
+};
+
+#endif // BACKUPPROTOCOL__H