1 files changed, 134 insertions, 0 deletions
diff --git a/src/framework/include/nsfw_ps_api.h b/src/framework/include/nsfw_ps_api.h
new file mode 100644
index 0000000..69fa992
--- /dev/null
+++ b/src/framework/include/nsfw_ps_api.h
@@ -0,0 +1,134 @@
+/*
+*
+* Copyright (c) 2018 Huawei Technologies Co.,Ltd.
+* Licensed under the Apache License, Version 2.0 (the "License");
+* you may not use this file except in compliance with the License.
+* You may obtain a copy of the License at:
+*
+*     http://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing, software
+* distributed under the License is distributed on an "AS IS" BASIS,
+* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+* See the License for the specific language governing permissions and
+* limitations under the License.
+*/
+
+#ifndef _NSFW_PS_API_H
+#define _NSFW_PS_API_H
+
+#include "list.h"
+#ifdef __cplusplus
+/* *INDENT-OFF* */
+extern "C"{
+/* *INDENT-ON* */
+#endif /* __cplusplus */
+
+#define NSFW_PS_MODULE      "nsfw_ps"
+
+#define NSFW_PS_MAX_CALLBACK 16
+
+#define NSFW_MAX_HBT_CHK_COUNT_DEF 60
+
+typedef enum _nsfw_ps_state
+{
+  NSFW_PS_NULL = 0,
+  NSFW_PS_START,
+  NSFW_PS_RUNNING,
+  NSFW_PS_PARENT_FORK,
+  NSFW_PS_CHILD_FORK,
+  NSFW_PS_HBT_FAILED,
+  NSFW_PS_EXITING,
+  NSFW_PS_EXIT,
+  NSFW_PS_MAX
+} nsfw_ps_state;
+
+/* for state change call back proc*/
+typedef int (*nsfw_ps_proc_fun) (void *pps_info, void *argv);
+typedef struct _nsfw_ps_callback
+{
+  u8 state;
+  nsfw_ps_proc_fun fun;
+  void *argv;
+} nsfw_ps_callback;
+
+/* for value in ps_info get/set*/
+typedef enum _nsfw_ps_user_value
+{
+  NSFW_REC_IDX = 1,
+  NSFW_REC_TIMER = 2,
+  NSFW_PS_UV_MAX = 16
+} nsfw_ps_user_value;
+#define nsfw_ps_set_uv(_pps_info, _type, _value) (_pps_info)->value[(_type)] = (_value)
+#define nsfw_ps_get_uv(_pps_info, _type) ((_pps_info)->value[(_type)])
+
+typedef struct _nsfw_ps_info
+{
+  struct list_head node;
+  u8 alloc_flag;
+  u8 state;                     /*nsfw_ps_state */
+  u8 proc_type;                 /*fw_poc_type */
+  u8 rechk_flg;
+  u32 host_pid;
+  u32 parent_pid;               /* only use for fork */
+  u32 cur_child_pid;            /* only use for fork */
+  void *exit_timer_ptr;
+  void *resend_timer_ptr;
+  void *hbt_timer_ptr;
+  u32 hbt_failed_count;
+  nsfw_ps_callback callback[NSFW_PS_MAX_CALLBACK];
+  void *value[NSFW_PS_UV_MAX];
+} nsfw_ps_info;
+
+typedef struct _nsfw_thread_dogs
+{
+  u8 alloc_flag;
+  i32 count;
+  u32 thread_id;
+} nsfw_thread_dogs;
+
+extern nsfw_ps_info *nsfw_ps_info_alloc (u32 pid, u8 proc_type);
+extern nsfw_ps_info *nsfw_ps_info_get (u32 pid);
+extern void nsfw_ps_info_free (nsfw_ps_info * ps_info);
+
+extern u8 nsfw_ps_reg_fun (nsfw_ps_info * pps_info, u8 ps_state,
+                           nsfw_ps_proc_fun fun, void *argv);
+
+/* will auto reg after ps_info alloc*/
+extern u8 nsfw_ps_reg_global_fun (u8 proc_type, u8 ps_state,
+                                  nsfw_ps_proc_fun fun, void *argv);
+
+typedef struct _nsfw_ps_info_msg
+{
+  u32 host_pid;
+  u32 parent_pid;
+  u64 reserve;
+} nsfw_ps_info_msg;
+extern u8 nsfw_ps_exit_end_notify (u32 pid);
+
+/*for heartbeat check*/
+extern u8 nsfw_ps_check_dst_init (u8 dst_proc_type);
+extern u8 nsfw_thread_chk ();
+extern nsfw_thread_dogs *nsfw_thread_getDog ();
+extern u8 nsfw_thread_chk_unreg ();
+extern u8 nsfw_ps_hbt_start (nsfw_ps_info * ps_info);
+extern u8 nsfw_ps_hbt_stop (nsfw_ps_info * ps_info);
+
+extern u32 nsfw_ps_iterator (nsfw_ps_proc_fun fun, void *argv);
+
+#define MAX_THREAD 16
+extern pthread_t g_all_thread[];
+extern u8 nsfw_reg_trace_thread (pthread_t tid);
+
+typedef int (*nsfw_ps_pid_fun) (u32 pid, u8 proc_type, void *argv);
+extern int nsfw_ps_rechk_pid_exit (nsfw_ps_pid_fun fun, void *argv);
+extern nsfw_ps_info *nsfw_share_ps_info_get (u32 pid);
+extern void nsfw_ps_cfg_set_chk_count (u16 count);
+
+#ifdef __cplusplus
+/* *INDENT-OFF* */
+}
+/* *INDENT-ON* */
+#endif /* __cplusplus */
+
+#endif /* _NSFW_PS_API_H  */