1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
/*
* Copyright (c) 2018 HUACHENTEL and/or its affiliates.
* 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.
*/
#include "sc_plugins.h"
#include <dirent.h>
#include <vpp-api/client/stat_client.h>
sc_plugin_main_t sc_plugin_main;
static int vpp_pid_start;
sc_plugin_main_t *sc_get_plugin_main()
{
return &sc_plugin_main;
}
/* get vpp pid in system */
int get_vpp_pid()
{
DIR *dir;
struct dirent *ptr;
FILE *fp;
char filepath[50];
char filetext[20];
dir = opendir("/proc");
int vpp_pid = 0;
/* read vpp pid file in proc, return pid of vpp */
if (NULL != dir)
{
while (NULL != (ptr =readdir(dir)))
{
if ((0 == strcmp(ptr->d_name, ".")) || (0 == strcmp(ptr->d_name, "..")))
continue;
if (DT_DIR != ptr->d_type)
continue;
sprintf(filepath, "/proc/%s/cmdline",ptr->d_name);
fp = fopen(filepath, "r");
if (NULL != fp)
{
fread(filetext, 1, 13, fp);
filetext[12] = '\0';
if (filetext == strstr(filetext, "/usr/bin/vpp"))
vpp_pid = atoi(ptr->d_name);
fclose(fp);
}
}
closedir(dir);
}
return vpp_pid;
}
int sr_plugin_init_cb(sr_session_ctx_t *session, void **private_ctx)
{
int rc;
sc_plugin_main.session = session;
/* Connect to VAPI */
rc = sc_connect_vpp();
if (0 != rc) {
SRP_LOG_ERR("vpp vapi connect error , with return %d.", rc);
return SR_ERR_INTERNAL;
}
/* Connect to STAT API */
rc = stat_segment_connect(STAT_SEGMENT_SOCKET_FILE);
if (rc != 0) {
SRP_LOG_ERR("vpp stat connect error , with return %d.", rc);
return SR_ERR_INTERNAL;
}
rc = sc_call_all_init_function(&sc_plugin_main);
if (rc != SR_ERR_OK) {
SRP_LOG_ERR("Call all init function error: %d", rc);
return rc;
}
/* set subscription as our private context */
*private_ctx = sc_plugin_main.subscription;
/* get the vpp pid sweetcomb connected, we assumed that only one vpp is run in system */
vpp_pid_start = get_vpp_pid();
return SR_ERR_OK;
}
void sr_plugin_cleanup_cb(sr_session_ctx_t *session, void *private_ctx)
{
sc_call_all_exit_function(&sc_plugin_main);
/* subscription was set as our private context */
if (private_ctx != NULL)
sr_unsubscribe(session, private_ctx);
SRP_LOG_DBG_MSG("unload plugin ok.");
/* Disconnect from STAT API */
stat_segment_disconnect();
/* Disconnect from VAPI */
sc_disconnect_vpp();
SRP_LOG_DBG_MSG("plugin disconnect vpp ok.");
}
int sr_plugin_health_check_cb(sr_session_ctx_t *session, void *private_ctx)
{
/* health check, will use shell to detect vpp when plugin is loaded */
/* health_check will run every 10 seconds in loop*/
int vpp_pid_now = get_vpp_pid();
if(vpp_pid_now == vpp_pid_start)
{
return SR_ERR_OK;
}
else
{
return -1;
}
}
|
