Require Import Coqlib.
Require Import Maps.
Require Import ASTExtra.
Require Import Integers.
Require Import Floats.
Require Import Values.
Require Import Memory.
Require Import Events.
Require Import Stacklayout.
Require Import Globalenvs.
Require Import AsmX.
Require Import Smallstep.
Require Import AuxStateDataType.
Require Import Constant.
Require Import GlobIdent.
Require Import FlatMemory.
Require Import CommonTactic.
Require Import AuxLemma.
Require Import RealParams.
Require Import PrimSemantics.
Require Import LAsm.
Require Import XOmega.

Require Import liblayers.logic.PTreeModules.
Require Import liblayers.logic.LayerLogicImpl.
Require Import liblayers.compat.CompatLayers.
Require Import liblayers.compat.CompatGenSem.

Require Import CalRealPTPool.
Require Import CalRealPT.
Require Import CalRealIDPDE.
Require Import CalRealInitPTE.

Require Import CalRealSMSPool.
Require Import CalRealProcModule.

Require Import INVLemmaMemory.
Require Import INVLemmaThread.
Require Import INVLemmaProc.
Require Import INVLemmaIntel.

Require Import AbstractDataType.

Require Export VEPTInit.
Require Export ObjVMCS.
Require Import LoadStoreSem3.
Require Export IntelPrimSemantics.

Section WITHMEM.

  Local Open Scope Z_scope.

  Context `{real_params: RealParams}.
  Context `{multi_oracle_prop: MultiOracleProp}.

  Context `{Hstencil: Stencil}.
  Context `{Hmem: Mem.MemoryModelX}.
  Context `{Hmwd: UseMemWithData mem}.

  Section INV.

    Global Instance vmcs_set_revid_inv: PreservesInvariants vmcs_set_revid_spec.
    Proof.
      preserves_invariants_simpl low_level_invariant high_level_invariant; eauto 2.
      unfold VMCSPool_inject_neutral; intros.
      simpl in H6.
      case_eq (zeq vmid (CPU_ID d)); intros; subst.
      - rewrite ZMap.gss.
        eapply VMCS_inject_neutral_gso; eauto.
      - rewrite ZMap.gso; eauto.
    Qed.

    Global Instance vmcs_set_abrtid_inv: PreservesInvariants vmcs_set_abrtid_spec.
    Proof.
      preserves_invariants_simpl low_level_invariant high_level_invariant; eauto 2.
      unfold VMCSPool_inject_neutral; intros.
      simpl in H6.
      case_eq (zeq vmid (CPU_ID d)); intros; subst.
      - rewrite ZMap.gss.
        eapply VMCS_inject_neutral_gso; eauto.
      - rewrite ZMap.gso; eauto.
    Qed.

    Global Instance vmcs_writez_inv: PreservesInvariants vmcs_writez_spec.
    Proof.
      preserves_invariants_simpl low_level_invariant high_level_invariant; eauto 2.
      unfold VMCSPool_inject_neutral; intros.
      simpl in H7.
      case_eq (zeq vmid (CPU_ID d)); intros; subst.
      - rewrite ZMap.gss.
        eapply VMCS_inject_neutral_gss_Vint; eauto.
      - rewrite ZMap.gso; eauto.
    Qed.

    Global Instance vmcs_writez64_inv: PreservesInvariants vmcs_writez64_spec.
    Proof.
      preserves_invariants_simpl low_level_invariant high_level_invariant; eauto 2.
      unfold VMCSPool_inject_neutral; intros.
      simpl in H7.
      case_eq (zeq vmid (CPU_ID d)); intros; subst.
      - rewrite ZMap.gss.
        repeat eapply VMCS_inject_neutral_gss_Vint_same; eauto.
      - rewrite ZMap.gso; eauto.
    Qed.


    Global Instance vmcs_write_ident_inv: VMCS_writebInvariants vmcs_write_ident_spec.
    Proof.
      constructor; intros; functional inversion H; simpl; auto.
      -
        inv H1. constructor; trivial; intros; simpl in ×.
        unfold VMCSPool_inject_neutral; intros.
        case_eq (zeq vmid (CPU_ID d)); intros; subst.
        + rewrite ZMap.gss.
          eapply VMCS_inject_neutral_gss; eauto.
          split; econstructor; eauto. rewrite Int.add_zero. trivial.
        + rewrite ZMap.gso; eauto.
      -
        inv H0. constructor; simpl; eauto 2; try congruence; intros.
    Qed.

    Global Instance vmptrld_inv: PreservesInvariants vmptrld_spec.
    Proof.
      preserves_invariants_simpl low_level_invariant high_level_invariant; eauto 2.
    Qed.

  End INV.

  Definition vmcsintro_fresh : compatlayer (cdata RData) :=
    (vmcs_get_revid ↦ gensem vmcs_get_revid_spec
                   ⊕ vmcs_set_revid ↦ gensem vmcs_set_revid_spec
                   ⊕ vmcs_get_abrtid ↦ gensem vmcs_get_abrtid_spec
                   ⊕ vmcs_set_abrtid ↦ gensem vmcs_set_abrtid_spec)
                   ⊕ vmcs_readz ↦ gensem vmcs_readz_spec
                   ⊕ vmcs_writez ↦ gensem vmcs_writez_spec
                   ⊕ vmcs_readz64 ↦ gensem vmcs_readz64_spec
                   ⊕ vmcs_writez64 ↦ gensem vmcs_writez64_spec
                   
                   ⊕ vmcs_write_ident ↦ vmcs_writeb_compatsem vmcs_write_ident_spec
                   ⊕ rcr0 ↦ gensem rcr0_spec
                   ⊕ rcr3 ↦ gensem rcr3_spec
                   ⊕ rcr4 ↦ gensem rcr4_spec
                   ⊕ rdmsr ↦ gensem rdmsr_spec
                   ⊕ wrmsr ↦ gensem wrmsr_spec
                   ⊕ vmptrld ↦ gensem vmptrld_spec
                   ⊕ vmx_enable ↦ gensem vmx_enable_spec.

  Definition vmcsintro_passthrough : compatlayer (cdata RData) :=
    fload ↦ gensem fload_spec
          ⊕ fstore ↦ gensem fstore_spec
          ⊕ vmxinfo_get ↦ gensem vmxinfo_get_spec
          ⊕ palloc ↦ gensem palloc_spec
          ⊕ set_pt ↦ gensem setPT_spec
          ⊕ pt_read ↦ gensem ptRead_spec
          ⊕ pt_resv ↦ gensem ptResv_spec
          ⊕ shared_mem_status ↦ gensem shared_mem_status_spec
          ⊕ offer_shared_mem ↦ gensem offer_shared_mem_spec
          ⊕ thread_wakeup ↦ gensem biglow_thread_wakeup_spec
          ⊕ thread_yield ↦ primcall_thread_schedule_compatsem biglow_thread_yield_spec (prim_ident:= thread_yield)
          ⊕ thread_sleep ↦ primcall_thread_transfer_compatsem biglow_thread_sleep_spec
          ⊕ sched_init ↦ gensem biglow_sched_init_spec

          ⊕ uctx_get ↦ gensem uctx_get_spec
          ⊕ uctx_set ↦ gensem uctx_set_spec
          ⊕ proc_create ↦ proc_create_compatsem biglow_proc_create_spec

          ⊕ container_get_nchildren ↦ gensem container_get_nchildren_spec
          ⊕ container_get_quota ↦ gensem container_get_quota_spec
          ⊕ container_get_usage ↦ gensem container_get_usage_spec
          ⊕ container_can_consume ↦ gensem container_can_consume_spec

          ⊕ get_CPU_ID ↦ gensem get_CPU_ID_spec
          ⊕ get_curid ↦ gensem get_curid_spec

          ⊕ acquire_lock_CHAN ↦ gensem acquire_lock_SC_spec
          ⊕ release_lock_CHAN ↦ gensem release_lock_SC_spec
          ⊕ get_sync_chan_busy ↦ gensem get_sync_chan_busy_spec
          ⊕ set_sync_chan_busy ↦ gensem set_sync_chan_busy_spec

          ⊕ ipc_send_body ↦ gensem ipc_send_body_spec
          ⊕ ipc_receive_body ↦ gensem ipc_receive_body_spec

          ⊕ ept_invalidate_mappings ↦ gensem ept_invalidate_mappings_spec
          ⊕ ept_add_mapping ↦ gensem ept_add_mapping_spec
          ⊕ ept_init ↦ gensem ept_init_spec

          ⊕ cli ↦ gensem cli_spec
          ⊕ sti ↦ gensem sti_spec
          ⊕ serial_intr_disable ↦ gensem serial_intr_disable_spec
          ⊕ serial_intr_enable ↦ gensem serial_intr_enable_spec
          ⊕ serial_putc ↦ gensem serial_putc_spec
          ⊕ cons_buf_read ↦ gensem cons_buf_read_spec

          ⊕ host_in ↦ primcall_general_compatsem hostin_spec
          ⊕ host_out ↦ primcall_general_compatsem hostout_spec
          ⊕ proc_create_postinit ↦ gensem proc_create_postinit_spec
          ⊕ trap_get ↦ primcall_trap_info_get_compatsem trap_info_get_spec
          ⊕ trap_set ↦ primcall_trap_info_ret_compatsem trap_info_ret_spec
          ⊕ proc_start_user ↦ primcall_start_user_compatsem proc_start_user_spec
          ⊕ proc_exit_user ↦ primcall_exit_user_compatsem proc_exit_user_spec
          ⊕ proc_start_user2 ↦ primcall_start_user_compatsem2 proc_start_user_spec
          ⊕ proc_exit_user2 ↦ primcall_exit_user_compatsem2 proc_exit_user_spec

          ⊕ accessors ↦ {| exec_load := (@exec_loadex _ _ _ _ _ _ _ _ _ _ _ _ _ _ Hmwd);
                           exec_store := (@exec_storeex _ _ _ _ _ _ _ _ _ _ _ _ _ _ Hmwd) |}.

  Definition vmcsintro : compatlayer (cdata RData) := vmcsintro_fresh ⊕ vmcsintro_passthrough.

End WITHMEM.