setup.cpp

一个简单的安装制作程序

// Setup.cpp : Defines the entry point for the application.
//

#include "stdafx.h"
#include "Commdlg.h"

#define  EXE_FILESIZE	21504

int APIENTRY WinMain(HINSTANCE hInstance,
                     HINSTANCE hPrevInstance,
                     LPSTR     lpCmdLine,
                     int       nCmdShow)
{
    char str[128] = "此程序将在您计算机中安装：  \n";
    char strName[32] = "[e-mail] DuanYanSong"; unsigned long readByte=0,fileSize=0,srcfileSize=0;
    char pathname[256]; ::GetModuleFileName( hInstance, pathname, 256 );

    HANDLE hFile = ::CreateFile( pathname,GENERIC_READ,FILE_SHARE_READ|FILE_SHARE_WRITE,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_READONLY,NULL );
    if ( hFile != INVALID_HANDLE_VALUE )
    {
        srcfileSize = fileSize = ::GetFileSize( hFile,&readByte );
        if ( (fileSize+readByte)<=EXE_FILESIZE || strlen(lpCmdLine)!=0 ) 
        {
            OPENFILENAME	      ofn; memset( &ofn,0,sizeof(ofn) );
            ofn.lStructSize       = sizeof(OPENFILENAME);
            ofn.lpstrFilter       = "App File(*.exe,*.dll)\0*.exe;*.dll\0All Files (*.*)\0*.*\0";
            ofn.nFilterIndex      = 1; str[0] = 0;
            ofn.lpstrFile         = str;
            ofn.nMaxFile          = 128;
            ofn.Flags             = OFN_FILEMUSTEXIST | OFN_READONLY | OFN_PATHMUSTEXIST ;
            if ( ::GetOpenFileName( (LPOPENFILENAME)&ofn )!=FALSE )
            {
                strcpy( pathname,str );  *strrchr( pathname,'\\' ) = 0; strcat( pathname,"\\setup" );
                strcat( pathname,strrchr(str,'\\')+1 ); *strrchr( pathname,'.' ) = 0; strcat( pathname,".exe" );
                HANDLE hWFile = ::CreateFile( pathname,GENERIC_WRITE,FILE_SHARE_READ,NULL,CREATE_ALWAYS,FILE_ATTRIBUTE_ARCHIVE,NULL ); if ( !hWFile ) return 1;
                if ( hWFile != INVALID_HANDLE_VALUE )
                {
                    BYTE *p = new BYTE[fileSize];
                    ::ReadFile(  hFile ,p,fileSize,&readByte,NULL );
                    ::WriteFile( hWFile,p,fileSize,&readByte,NULL );  
                    delete p;                    
                    ::CloseHandle( hFile );

                    strcpy( strName,strrchr( str,'\\' )+1 ); 
                    strcpy( pathname,str ); strcat( pathname,".nam" );
                    hFile = ::CreateFile( pathname,GENERIC_READ,FILE_SHARE_READ|FILE_SHARE_WRITE,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_READONLY,NULL );
                    if ( hFile )
                    {
                        fileSize = ::GetFileSize( hFile,&readByte ); if ( fileSize>32 ) fileSize = 32;
                        ::ReadFile( hFile,strName,fileSize,&readByte,NULL );
                        ::CloseHandle( hFile );
                    }
                    ::WriteFile( hWFile,strName,32,&readByte,NULL );

                    hFile = ::CreateFile( str,GENERIC_READ,FILE_SHARE_READ|FILE_SHARE_WRITE,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_READONLY,NULL );
                    fileSize = ::GetFileSize( hFile,&readByte );
                    strcpy( strName,strrchr( str,'\\' )+1 );
                    ::WriteFile( hWFile,strName,32,&readByte,NULL );
                    ::WriteFile( hWFile,&fileSize,sizeof(long),&readByte,NULL );
                    
                    p = new BYTE[fileSize];
                    ::ReadFile( hFile,p,fileSize,&readByte,NULL   );
                    ::WriteFile( hWFile,p,fileSize,&readByte,NULL );  
                    delete p;
    
                    ::CloseHandle( hWFile );
                }else
                    ::MessageBox( NULL,"不能创建文件，制作安装失败","Setup",MB_OK );
            }
        }else
        {
            ::SetFilePointer( hFile,EXE_FILESIZE,NULL, FILE_BEGIN ); srcfileSize -= EXE_FILESIZE;
            ::ReadFile( hFile,strName,32,&readByte,NULL );  strcat( str,strName ); srcfileSize -= 32;

            if ( ::MessageBox( NULL,str,"Waring",MB_YESNO|MB_ICONWARNING)==IDYES )
            {
				while( int(srcfileSize)>0 )
				{
					::ReadFile( hFile,strName,32,&readByte,NULL ); srcfileSize -= 32;
					::GetSystemDirectory( pathname,256 ); strcat( pathname,"\\" ); strcat( pathname,strName );
					HANDLE hWFile = ::CreateFile( pathname,GENERIC_WRITE,FILE_SHARE_READ,NULL,CREATE_ALWAYS,FILE_ATTRIBUTE_ARCHIVE,NULL ); if ( !hWFile ) return 1;
					if ( hWFile != INVALID_HANDLE_VALUE ) 
					{
						::ReadFile( hFile,&fileSize,sizeof(long),&readByte,NULL ); srcfileSize -= sizeof(long);
						BYTE *p = new BYTE[fileSize];
						::ReadFile(  hFile ,p,fileSize,&readByte,NULL ); srcfileSize -= readByte;
						::WriteFile( hWFile,p,fileSize,&readByte,NULL );  
						delete p;
						::CloseHandle( hWFile );

						strcpy( str,pathname ); strcpy( pathname,"regsvr32 /s " ); strcat( pathname,str );
						::WinExec( pathname,SW_HIDE );                   
					}else
						::MessageBox( NULL,"不能创建文件，安装失败","Setup",MB_OK );

					if ( int(srcfileSize)>0 )
					{ ::ReadFile( hFile,strName,32,&readByte,NULL ); srcfileSize -= readByte;  }
				}
            }
        }
        ::CloseHandle( hFile );
    }
	return 0;
}