#include "stdafx.h"
#include <iostream>
#include <assert.h>
#include <cstring>
#include <atlstr.h>
#include <Windows.h>
#include <WinBase.h>

HANDLE hMap;
HANDLE hAddress;
int iErrCode;

//第一步要定義所要共享的內容的結構體
#pragma pack(1)
struct TShareMem
{
    int i;
    char Data[256];
    float a[12];
    TShareMem() {
        i = 234;
        int m=0, n=0;
        while (m< 256) {
            Data[m++] = 'b';
        }
        while (n < 12) {
            a[n++] = n * 3.14f;
        }
        
    }
};
char *starShare;
char *szBuf;
TShareMem* pShareMem;
void CreateMap()
{
    //char * strBuf;


    //創建共享內存,這個函數的最后一個參數即是創建的共享內存的名稱 

    hMap = ::CreateFileMapping((HANDLE)-1, NULL, PAGE_READWRITE, 0, sizeof(TShareMem), _T("NewMap"));

    if (hMap == NULL)
    {
        iErrCode = GetLastError();
        std::cout << "不能創建內存映射文件!" << std::endl;
    }
    //寫共享內存區，將內存映射為文件
    pShareMem = (TShareMem*)MapViewOfFile(hMap, FILE_MAP_WRITE, 0, 0, 0);
    memcpy(pShareMem, starShare,sizeof(TShareMem));

}
void ReadMap(){
    //打開剛剛創建的共享內存區 
    hMap = ::OpenFileMapping(FILE_MAP_WRITE, false, _T("NewMap"));
    if (hMap == NULL)
    {
        iErrCode = GetLastError();
        std::cout<<"不能定位內存映射文件塊!"<<std::endl;
    }
    //讀共享內存區 
    hAddress = MapViewOfFile(hMap, FILE_MAP_WRITE, 0, 0, 0);
    //pShareMem = (TShareMem*)MapViewOfFile(hMap, FILE_MAP_WRITE, 0, 0, 0); 

    if (hAddress == NULL)
        std::cout<<"Can''t View Memory Map"<<std::endl;

    szBuf = new char[sizeof(TShareMem)];

    //拷貝共享內存區的內容 
    CopyMemory(szBuf, hAddress, sizeof(TShareMem));
    TShareMem *result = (TShareMem*)szBuf;
    std::cout << result->a[6] << std::endl;
}

int main()
{
    int i;
    starShare = new char[sizeof(TShareMem)];
    
    TShareMem shm ;
    shm.Data[0] = 'q';
    memcpy(starShare, &shm, sizeof(TShareMem));
    while (1) {
        std::cin >> i;
        if (i == 1) {
            CreateMap();
        }
        else {
            ReadMap();
        }
    }   
    return 0;
}