Rawrite32Dlg.h

/*-
 * Copyright (c) 2000-2003,2010-2018 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * Copyright (c) 2000-2003,2010-2018 Martin Husemann <martin@duskware.de>.
 * All rights reserved.
 * 
 * This code was developed by Martin Husemann for the benefit of
 * all NetBSD users and The NetBSD Foundation.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. The name of the author may not be used to endorse or promote products
 *    derived from this software withough specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#if !defined(AFX_RAWRITE32DLG_H__84A6A733_0EA9_40E6_AEC1_353C157AA5C8__INCLUDED_)
#define AFX_RAWRITE32DLG_H__84A6A733_0EA9_40E6_AEC1_353C157AA5C8__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

/////////////////////////////////////////////////////////////////////////////
// CRawrite32Dlg dialog

class CRawrite32Dlg : public CDialog
{
// Construction
public:
  CRawrite32Dlg(LPCTSTR imageFileName);	// standard constructor
  ~CRawrite32Dlg();

// Dialog Data
  //{{AFX_DATA(CRawrite32Dlg)
	enum { IDD = IDD_RAWRITE32_DIALOG };
	CComboBox	m_drives;
  CProgressCtrl m_progress;
	CString	m_output;
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CRawrite32Dlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);	// DDX/DDV support
	//}}AFX_VIRTUAL

public:
  void UpdateMenu(CMenu*);

// Implementation
protected:
	HICON m_hIcon;          // program icon
  HICON m_hSmallIcon;     // and a small variant of it
  CFont m_outWinFont;     // font used for the output window
  CMenu m_mainMenu;       // our menu, usually hidden

  // dialog data
  CString m_imageName;    // file path of the input image
  DWORD m_sectorSkip;     // don't touch this many sectors on the output
                          // drive (i.e. the image is written starting at the sector given here)
  // target drive selection
  struct DriveSelectionEntry {
    CString deviceName;       // product/manufacturer/version hardware description (if available)
    CString internalFileName; // unused on DOS
    CString size;
    DWORD driveNumber;        // internal drive number (physical disk index) or DOS drive number
    vector<CString> volumes;  // list of drive letters found on this disk
    bool hidden;              // do not display this entry, as i.e. C:\Windows points to it
    DriveSelectionEntry() : driveNumber(~0U), hidden(false) {}
  };
  vector<DriveSelectionEntry> m_driveData; // raw data of combo box contentents, ItemData is the index in this vector

  // sliding view on mapped input file and output statistics
  HANDLE m_inputFile;     // input file handle
  HANDLE m_inputMapping;  // file mapping handle
  DWORD64 m_inputFileSize;// file size of input image
  DWORD64 m_fileOffset;   // current offset of mapping view
  DWORD64 m_sizeWritten;  // bytes written to output device
  const BYTE *m_fsImage;  // input image
  size_t m_fsImageSize;   // size of input image
  void *m_startBuf;       // saved "start of disk" part

  // during decompression:
  const BYTE *m_curInput; // pointer to next input data
  size_t m_sizeRemaining; // unread data past m_curInput
  HANDLE m_outputDevice;  // handle of the output device
  BYTE *m_outputBuffer;   // decompression target, write buffer
  
#ifdef _M_IX86            // special case for legacy versions on arch=i386
  DWORD m_sectorOut;      // next sector to write
  bool m_usingVXD;        // true if we are on Win9x and can't write raw devices directly
#else
#define m_usingVXD  false
#endif

protected:
  bool OpenInputFile(HANDLE);         // map input and calculate hashes
  void CloseInputFile();              // close input file and mapping
  bool VerifyInput();                 // return TRUE if "WriteToDisk" button has been enabled
  void CalcHashes(CString &out);      // calculate hahes and format proper output
    // wait for all objects passed, handling paint and other important messages
  bool WaitAndPoll(vector<HANDLE>&, DWORD timeout); // return true if timed out, false if we don't need to wait more
  void Poll();                        // handle all waiting messages
  bool MapInputView();                // map the next block at the current input offset
  bool AdvanceMapOffset();            // advance the input offset, return false if past end of file
  void FormatSize(DWORD64, CString&, DWORD addFactor = 1) const; // format a human readable size from a value in bytes
  void ShowError(DWORD err, UINT id, LPCTSTR = NULL);
  void ShowOutput();
  void UpdateWriteProgress();
  void EnableDlgChildControls(bool enable);
  void FillDriveCombo();
  void EnumPhysicalDrives();
  void EnumLogicalVolumes();
  void CollectDebugInfo() const;

  // support for hidden main menu
  virtual BOOL PreTranslateMessage(MSG *pMsg);
  void ShowMainMenu(bool showIt);

protected:
  // background decompression thread
  static UINT __cdecl dcompressionStarter(void *token);
  UINT BackgroundDecompressor();
protected:
  interface IGenericDecompressor *m_decomp; // decompressor to use
  HANDLE m_decompOutputSpaceAvailable;  // event: decompressor input available
  HANDLE m_decompOutputAvailable;     // event: write buffer ready for output
  DWORD m_decompOutputLen;            // amount of decompressor output ready
  DWORD m_curDecompTarget;            // 0 or 1, decompressor writes to first or second half of the output buffer
  volatile LONG m_decompForcedExit;   // if != 0 the decompressor exits ASAP
  volatile LONG m_writerIdle;         // if != 0 the writer is waiting for more data

  int m_origHeight;
  // device selection
  bool m_writeTargetLogicalVolume;    // if true, write to logical volumes (instead of physical disk devices)

protected:
	// Generated message map functions
	//{{AFX_MSG(CRawrite32Dlg)
	virtual BOOL OnInitDialog();
	afx_msg void OnPaint();
	afx_msg HCURSOR OnQueryDragIcon();
  afx_msg void OnExitMenuLoop(BOOL);
	afx_msg void OnDestroy();
	afx_msg void OnBrowse();
	afx_msg void OnNewImage();
	afx_msg void OnWriteImage();
  afx_msg void OnAboutDlg();
  afx_msg void OnSurfHome();
  afx_msg void OnHashOptions();
  afx_msg void OnSecSkipOptions();
  afx_msg void OnUseVolumes();
  afx_msg void OnUsePhysDisks();
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
  
  DECLARE_INTERFACE_MAP()

  BEGIN_INTERFACE_PART(Drop, IDropTarget)
    STDMETHOD(DragEnter)(IDataObject*, DWORD, POINTL, LPDWORD);
    STDMETHOD(DragOver)(DWORD, POINTL, LPDWORD);
    STDMETHOD(DragLeave)();
    STDMETHOD(Drop)(IDataObject*, DWORD, POINTL, LPDWORD);
  END_INTERFACE_PART(Drop)
};

//{{AFX_INSERT_LOCATION}}
// Microsoft Visual C++ will insert additional declarations immediately before the previous line.

#endif // !defined(AFX_RAWRITE32DLG_H__84A6A733_0EA9_40E6_AEC1_353C157AA5C8__INCLUDED_)