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 LoadStoreSem2.
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 INVLemmaContainer.
Require Import INVLemmaMemory.
Require Import INVLemmaThread.
Require Import INVLemmaProc.
Require Import INVLemmaIntel.

Require Import AbstractDataType.

Require Export VVMCSIntro.
Require Import LoadStoreSem3.

Require Import CalRealPTPool.
Require Import CalRealPT.
Require Import CalRealIDPDE.
Require Import CalRealInitPTE.
Require Import CalRealProcModule.
Require Import INVLemmaMemory.
Require Import INVLemmaThread.
Require Import INVLemmaProc.
Require Import AbstractDataType.

Require Import CalTicketLock.
Require Import INVLemmaInterrupt.
Require Import INVLemmaDriver.
Require Import DeviceStateDataType.
Require Import FutureTactic.

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.

    Section INTERCEPT_INTWIN.

      Lemma vmx_set_intercept_intwin_high_inv:
        ∀ d d´ i,
          vmx_set_intercept_intwin_spec i d = Some d´→
          high_level_invariant d →
          high_level_invariant d´.
      Proof.
        intros. inv H0; functional inversion H; subst; simpl;
        constructor; simpl; intros; try congruence; eauto 2.
      Qed.

      Lemma vmx_set_intercept_intwin_low_inv:
        ∀ d d´ n i,
          vmx_set_intercept_intwin_spec i d = Some d´→
          low_level_invariant n d →
          low_level_invariant n d´.
      Proof.
        intros. inv H0; functional inversion H; subst; simpl;
        constructor; simpl; intros; try congruence; eauto 2.
        - unfold VMCSPool_inject_neutral; intros.
          case_eq (zeq vmid (CPU_ID d)); intros; subst.
          + rewrite ZMap.gss.
            eapply VMCS_inject_neutral_gss_Vint; eauto.
          + rewrite ZMap.gso; eauto.
        - unfold VMCSPool_inject_neutral; intros.
          case_eq (zeq vmid (CPU_ID d)); intros; subst.
          + rewrite ZMap.gss.
            eapply VMCS_inject_neutral_gss_Vint; eauto.
          + rewrite ZMap.gso; eauto.
      Qed.

      Lemma vmx_set_intercept_intwin_kernel_mode:
        ∀ d d´ i,
          vmx_set_intercept_intwin_spec i d = Some d´→
          kernel_mode d →
          kernel_mode d´.
      Proof.
        intros. functional inversion H; subst; constructor; trivial.
      Qed.

      Global Instance vmx_set_intercept_intwin_inv: PreservesInvariants vmx_set_intercept_intwin_spec.
      Proof.
        preserves_invariants_simpl´.
        - eapply vmx_set_intercept_intwin_low_inv; eauto.
        - eapply vmx_set_intercept_intwin_high_inv; eauto.
        - eapply vmx_set_intercept_intwin_kernel_mode; eauto.
      Qed.

    End INTERCEPT_INTWIN.

    Section INJECT_EVENT.

      Lemma vmx_inject_event_high_inv:
        ∀ d d´ i i0 i1 i2,
          vmx_inject_event_spec i i0 i1 i2 d = Some d´→
          high_level_invariant d →
          high_level_invariant d´.
      Proof.
        intros. inv H0; functional inversion H; subst; simpl;
        constructor; simpl; intros; try congruence; eauto 2.
      Qed.

      Lemma vmx_inject_event_low_inv:
        ∀ d d´ n i i0 i1 i2,
          vmx_inject_event_spec i i0 i1 i2 d = Some d´→
          low_level_invariant n d →
          low_level_invariant n d´.
      Proof.
        intros. inv H0; functional inversion H; subst; simpl;
        constructor; simpl; intros; try congruence; eauto 2.
        - unfold VMCSPool_inject_neutral; intros.
          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.
        - unfold VMCSPool_inject_neutral; intros.
          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.

      Lemma vmx_inject_event_kernel_mode:
        ∀ d d´ i i0 i1 i2,
          vmx_inject_event_spec i i0 i1 i2 d = Some d´→
          kernel_mode d →
          kernel_mode d´.
      Proof.
        intros. functional inversion H; subst; constructor; trivial.
      Qed.

      Global Instance vmx_inject_event_inv: PreservesInvariants vmx_inject_event_spec.
      Proof.
        preserves_invariants_simpl´.
        - eapply vmx_inject_event_low_inv; eauto.
        - eapply vmx_inject_event_high_inv; eauto.
        - eapply vmx_inject_event_kernel_mode; eauto.
      Qed.

    End INJECT_EVENT.

    Section SET_DESC1.

      Lemma vmx_set_desc1_high_inv:
        ∀ d d´ i i0 i1,
          vmx_set_desc1_spec i i0 i1 d = Some d´→
          high_level_invariant d →
          high_level_invariant d´.
      Proof.
        intros. inv H0; functional inversion H; subst; simpl;
        constructor; simpl; intros; try congruence; eauto 2.
      Qed.

      Lemma vmx_set_desc1_low_inv:
        ∀ d d´ n i i0 i1,
          vmx_set_desc1_spec i i0 i1 d = Some d´→
          low_level_invariant n d →
          low_level_invariant n d´.
      Proof.
        intros. inv H0; functional inversion H; subst; simpl;
        constructor; simpl; intros; try congruence; eauto 2.
        functional inversion H6.
        - unfold VMCSPool_inject_neutral; intros.
          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.
        - unfold VMCSPool_inject_neutral; intros.
          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.

      Lemma vmx_set_desc1_kernel_mode:
        ∀ d d´ i i0 i1,
          vmx_set_desc1_spec i i0 i1 d = Some d´→
          kernel_mode d →
          kernel_mode d´.
      Proof.
        intros. functional inversion H; subst; constructor; trivial.
      Qed.

      Global Instance vmx_set_desc1_inv: PreservesInvariants vmx_set_desc1_spec.
      Proof.
        preserves_invariants_simpl´.
        - eapply vmx_set_desc1_low_inv; eauto.
        - eapply vmx_set_desc1_high_inv; eauto.
        - eapply vmx_set_desc1_kernel_mode; eauto.
      Qed.

    End SET_DESC1.

    Section SET_DESC2.

      Lemma vmx_set_desc2_high_inv:
        ∀ d d´ i i0 i1,
          vmx_set_desc2_spec i i0 i1 d = Some d´→
          high_level_invariant d →
          high_level_invariant d´.
      Proof.
        intros. inv H0; functional inversion H; subst; simpl;
        constructor; simpl; intros; try congruence; eauto 2.
      Qed.

      Lemma vmx_set_desc2_low_inv:
        ∀ d d´ n i i0 i1,
          vmx_set_desc2_spec i i0 i1 d = Some d´→
          low_level_invariant n d →
          low_level_invariant n d´.
      Proof.
        intros. inv H0; functional inversion H; subst; simpl;
        constructor; simpl; intros; try congruence; eauto 2.
        functional inversion H6.
        - unfold VMCSPool_inject_neutral; intros.
          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.
        - unfold VMCSPool_inject_neutral; intros.
          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.

      Lemma vmx_set_desc2_kernel_mode:
        ∀ d d´ i i0 i1,
          vmx_set_desc2_spec i i0 i1 d = Some d´→
          kernel_mode d →
          kernel_mode d´.
      Proof.
        intros. functional inversion H; subst; constructor; trivial.
      Qed.

      Global Instance vmx_set_desc2_inv: PreservesInvariants vmx_set_desc2_spec.
      Proof.
        preserves_invariants_simpl´.
        - eapply vmx_set_desc2_low_inv; eauto.
        - eapply vmx_set_desc2_high_inv; eauto.
        - eapply vmx_set_desc2_kernel_mode; eauto.
      Qed.

    End SET_DESC2.

    Section SET_TSC_OFFSET.

      Lemma vmx_set_tsc_offset_high_inv:
        ∀ d d´ i,
          vmx_set_tsc_offset_spec i d = Some d´→
          high_level_invariant d →
          high_level_invariant d´.
      Proof.
        intros. inv H0; functional inversion H; subst; simpl;
        constructor; simpl; intros; try congruence; eauto 2.
      Qed.

      Lemma vmx_set_tsc_offset_low_inv:
        ∀ d d´ n i,
          vmx_set_tsc_offset_spec i d = Some d´→
          low_level_invariant n d →
          low_level_invariant n d´.
      Proof.
        intros. inv H0; functional inversion H; subst; simpl;
        constructor; simpl; intros; try congruence; eauto 2.
        - unfold VMCSPool_inject_neutral; intros.
          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.

      Lemma vmx_set_tsc_offset_kernel_mode:
        ∀ d d´ i,
          vmx_set_tsc_offset_spec i d = Some d´→
          kernel_mode d →
          kernel_mode d´.
      Proof.
        intros. functional inversion H; subst; constructor; trivial.
      Qed.

      Global Instance vmx_set_tsc_offset_inv: PreservesInvariants vmx_set_tsc_offset_spec.
      Proof.
        preserves_invariants_simpl´.
        - eapply vmx_set_tsc_offset_low_inv; eauto.
        - eapply vmx_set_tsc_offset_high_inv; eauto.
        - eapply vmx_set_tsc_offset_kernel_mode; eauto.
      Qed.

    End SET_TSC_OFFSET.

    Global Instance vmcs_set_defaults_inv: VMCSSetDefaultsInvariants vmcs_set_defaults_spec.
    Proof.
      constructor; intros; functional inversion H; simpl; auto.
      -
        inv H6. constructor; trivial; intros; simpl in ×.
        eapply real_vmcs_inject_neutral; eauto.
      -
        inv H0. constructor; simpl; try congruence; intros; eauto.
    Qed.

  End INV.

  Definition vmcsinit_fresh : compatlayer (cdata RData) :=
    vmx_set_intercept_intwin ↦ gensem vmx_set_intercept_intwin_spec
                             ⊕ vmx_set_desc1 ↦ gensem vmx_set_desc1_spec
                             ⊕ vmx_set_desc2 ↦ gensem vmx_set_desc2_spec
                             ⊕ vmx_inject_event ↦ gensem vmx_inject_event_spec
                             ⊕ vmx_set_tsc_offset ↦ gensem vmx_set_tsc_offset_spec
                             ⊕ vmx_get_tsc_offset ↦ gensem vmx_get_tsc_offset_spec
                             ⊕ vmx_get_exit_reason ↦ gensem vmx_get_exit_reason_spec
                             ⊕ vmx_get_exit_fault_addr ↦ gensem vmx_get_exit_fault_addr_spec
                             ⊕ vmx_check_pending_event ↦ gensem vmx_check_pending_event_spec
                             ⊕ vmx_check_int_shadow ↦ gensem vmx_check_int_shadow_spec
                             ⊕ vmcs_set_defaults ↦ vmcs_set_defaults_compatsem vmcs_set_defaults_spec.

  Definition vmcsinit_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

          ⊕ vmcs_readz ↦ gensem vmcs_readz_spec
          ⊕ vmcs_writez ↦ gensem vmcs_writez_spec
          ⊕ vmcs_readz64 ↦ gensem vmcs_readz64_spec
          ⊕ vmcs_writez64 ↦ gensem vmcs_writez64_spec
          ⊕ vmptrld ↦ gensem vmptrld_spec
          ⊕ rdmsr ↦ gensem rdmsr_spec
          ⊕ wrmsr ↦ gensem wrmsr_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 vmcsinit : compatlayer (cdata RData) := vmcsinit_fresh ⊕ vmcsinit_passthrough.

End WITHMEM.