hc
2024-08-16 62c46c9150c4afde7e5b25436263fddf79d66f0b
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
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
/*
 * Copyright (C) 2015 Spreadtrum Communications Inc.
 *
 * Authors    :
 * Xianwei.Zhao <xianwei.zhao@spreadtrum.com>
 *
 * This software is licensed under the terms of the GNU General Public
 * License version 2, as published by the Free Software Foundation, and
 * may be copied, distributed, and modified under those terms.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 */
 
#include <linux/kernel.h>
#include <linux/platform_device.h>
#include <linux/module.h>
#include <linux/uaccess.h>
#include <net/genetlink.h>
#include <linux/types.h>
#include <linux/version.h>
#include <marlin_platform.h>
 
#include "sprdwl.h"
#include "npi.h"
 
static int sprdwl_nl_send_generic(struct genl_info *info, u8 attr, u8 cmd,
                 u32 len, u8 *data);
static struct genl_family sprdwl_nl_genl_family;
 
static int sprdwl_get_flag(void)
{
   struct file *fp = NULL;
#ifdef setfs
   mm_segment_t fs;
#endif
   loff_t *pos;
   int flag = 0;
   char file_data[2];
   unsigned long long tmp;
 
   fp = filp_open(SPRDWL_PSM_PATH, O_RDONLY, 0);
   if (IS_ERR(fp)) {
       wl_err("open file:%s failed\n", SPRDWL_PSM_PATH);
       return PTR_ERR(fp);
   }
#ifdef setfs
   fs = get_fs();
   set_fs(KERNEL_DS);
#endif
 
   pos = &fp->f_pos;
   kernel_read(fp, file_data, 1, pos);
 
   filp_close(fp, NULL);
#ifdef setfs
   set_fs(fs);
#endif
 
   file_data[1] = 0;
   if (kstrtoull(file_data, 10, &tmp)) {
       wl_err("%s: get value invald\n", __func__);
       return flag;
   }
   if (tmp)
       flag = SPRDWL_STA_GC_EN_SLEEP;
   else
       flag = SPRDWL_STA_GC_NO_SLEEP;
   return flag;
}
 
static int sprdwl_cmd_set_psm_cap(struct sprdwl_vif *vif)
{
   struct sprdwl_priv *priv = NULL;
   struct sprdwl_npi_cmd_hdr *msg;
   unsigned char r_buf[512] = {0}, s_buf[8];
   unsigned short r_len = 512;
   int s_len, flag, ret;
 
   if (!vif) {
       wl_err("%s: parameters invalid\n", __func__);
       return -EINVAL;
   }
   priv = vif->priv;
   flag = sprdwl_get_flag();
   if (flag < 0) {
       wl_err("%s: get flag failed\n", __func__);
       return 0;
   }
 
   msg = (struct sprdwl_npi_cmd_hdr *)s_buf;
   msg->type = SPRDWL_HT2CP_CMD;
   msg->subtype = SPRDWL_NPI_CMD_SET_WLAN_CAP;
   msg->len = sizeof(flag);
   s_len = msg->len + sizeof(*msg);
   memcpy(s_buf + sizeof(*msg), &flag, sizeof(flag));
   ret = sprdwl_npi_send_recv(priv, vif->ctx_id, s_buf,
                  s_len, r_buf, &r_len);
 
   wl_info("[%s psm is:%s]\n", __func__, flag ? "normal mode" : "rf mode");
 
   return ret;
}
 
static int sprdwl_npi_pre_doit(const struct genl_ops *ops,
                  struct sk_buff *skb, struct genl_info *info)
{
   struct net_device *ndev;
   struct sprdwl_vif *vif;
   struct sprdwl_priv *priv;
   int ifindex;
 
   if (!info) {
       wl_err("%s NULL info!\n", __func__);
       return -EINVAL;
   }
 
   if (info->attrs[SPRDWL_NL_ATTR_IFINDEX]) {
       ifindex = nla_get_u32(info->attrs[SPRDWL_NL_ATTR_IFINDEX]);
       ndev = dev_get_by_index(genl_info_net(info), ifindex);
       if (!ndev) {
           wl_err("NPI: Could not find ndev\n");
           return -EFAULT;
       }
       vif = netdev_priv(ndev);
       priv = vif->priv;
       info->user_ptr[0] = ndev;
       info->user_ptr[1] = priv;
   } else {
       wl_err("nl80211_pre_doit: Not have attr_ifindex\n");
       return -EFAULT;
   }
   return 0;
}
 
static void sprdwl_npi_post_doit(const struct genl_ops *ops,
                struct sk_buff *skb, struct genl_info *info)
{
   if (info->user_ptr[0])
       dev_put(info->user_ptr[0]);
}
 
static bool sprdwl_npi_cmd_is_start(void *buf)
{
   struct sprdwl_npi_cmd_hdr *msg;
 
   msg = (struct sprdwl_npi_cmd_hdr *)buf;
   if ((msg->type == SPRDWL_HT2CP_CMD) &&
       (msg->subtype == SPRDWL_NPI_CMD_START))
       return true;
   else
       return false;
}
 
static bool sta_or_p2p_is_opened(void)
{
   return false;
}
 
static int sprdwl_nl_npi_handler(struct sk_buff *skb_2, struct genl_info *info)
{
   struct net_device *ndev = NULL;
   struct sprdwl_vif *vif = NULL;
   struct sprdwl_priv *priv = NULL;
   struct sprdwl_npi_cmd_hdr *hdr = NULL;
   unsigned short r_len = 1024, s_len;
   unsigned char *s_buf = NULL, *r_buf = NULL;
   unsigned char dbgstr[64] = { 0 };
   int err = -100, ret = 0;
#if LINUX_VERSION_CODE >= KERNEL_VERSION(4, 14, 0)
   const char *id_name = NULL;
   unsigned char status = 0;
#endif
 
   ndev = info->user_ptr[0];
   vif = netdev_priv(ndev);
   priv = info->user_ptr[1];
   if (!info->attrs[SPRDWL_NL_ATTR_AP2CP]) {
       wl_err("%s: invalid content\n", __func__);
       return -EPERM;
   }
   r_buf = kmalloc(1024, GFP_KERNEL);
   if (!r_buf)
       return -ENOMEM;
 
   s_buf = nla_data(info->attrs[SPRDWL_NL_ATTR_AP2CP]);
   s_len = nla_len(info->attrs[SPRDWL_NL_ATTR_AP2CP]);
   if (sprdwl_npi_cmd_is_start(s_buf) && sta_or_p2p_is_opened()) {
       hdr = kzalloc(sizeof(*hdr), GFP_KERNEL);
       if (!hdr) {
           wl_err("%s: failed to alloc hdr!\n", __func__);
           kfree(r_buf);
           return -ENOMEM;
       }
       hdr->type = SPRDWL_CP2HT_REPLY;
       hdr->subtype = SPRDWL_NPI_CMD_START;
       hdr->len = sizeof(err);
       r_len = sizeof(*hdr) + hdr->len;
       memcpy(r_buf, hdr, sizeof(*hdr));
       memcpy(r_buf + sizeof(*hdr), &err, hdr->len);
       ret = sprdwl_nl_send_generic(info, SPRDWL_NL_ATTR_CP2AP,
                        SPRDWL_NL_CMD_NPI, r_len, r_buf);
       kfree(hdr);
       kfree(r_buf);
       return ret;
   }
 
   sprintf(dbgstr, "[iwnpi][SEND][%d]:", s_len);
   hdr = (struct sprdwl_npi_cmd_hdr *)s_buf;
   wl_err("%s type is %d, subtype %d\n", dbgstr, hdr->type, hdr->subtype);
 
#if LINUX_VERSION_CODE >= KERNEL_VERSION(4, 14, 0)
   if (hdr->subtype == SPRDWL_NPI_CMD_GET_CHIPID) {
       id_name = wcn_get_chip_name();
       sprintf(r_buf, "%d", status);
       strcat(r_buf, id_name);
       r_len = strlen(r_buf);
       wl_err("r_len = %d, %s\n", r_len, __func__);
   } else {
       sprdwl_npi_send_recv(priv, vif->ctx_id, s_buf, s_len, r_buf, &r_len);
 
       sprintf(dbgstr, "[iwnpi][RECV][%d]:", r_len);
       hdr = (struct sprdwl_npi_cmd_hdr *)r_buf;
       wl_err("%s type is %d, subtype %d\n", dbgstr, hdr->type, hdr->subtype);
   }
#else
   sprdwl_npi_send_recv(priv, vif->ctx_id, s_buf, s_len, r_buf, &r_len);
 
   sprintf(dbgstr, "[iwnpi][RECV][%d]:", r_len);
   hdr = (struct sprdwl_npi_cmd_hdr *)r_buf;
   wl_err("%s type is %d, subtype %d\n", dbgstr, hdr->type, hdr->subtype);
#endif
 
   ret = sprdwl_nl_send_generic(info, SPRDWL_NL_ATTR_CP2AP,
                    SPRDWL_NL_CMD_NPI, r_len, r_buf);
 
   if (sprdwl_npi_cmd_is_start(s_buf)) {
       msleep(100);
       ret = sprdwl_cmd_set_psm_cap(vif);
   }
   kfree(r_buf);
   return ret;
}
 
static int sprdwl_nl_get_info_handler(struct sk_buff *skb_2,
                     struct genl_info *info)
{
   struct net_device *ndev = info->user_ptr[0];
   struct sprdwl_vif *vif = netdev_priv(ndev);
   unsigned char r_buf[64] = { 0 };
   unsigned short r_len = 0;
   int ret = 0;
 
   if (vif) {
       ether_addr_copy(r_buf, vif->ndev->dev_addr);
       sprdwl_put_vif(vif);
       r_len = 6;
       ret = sprdwl_nl_send_generic(info, SPRDWL_NL_ATTR_CP2AP,
                        SPRDWL_NL_CMD_GET_INFO, r_len,
                        r_buf);
   } else {
       wl_err("%s NULL vif!\n", __func__);
       ret = -1;
   }
   return ret;
}
 
static struct nla_policy sprdwl_genl_policy[SPRDWL_NL_ATTR_MAX + 1] = {
   [SPRDWL_NL_ATTR_AP2CP] = {.type = NLA_BINARY, .len = 1024},
   [SPRDWL_NL_ATTR_CP2AP] = {.type = NLA_BINARY, .len = 1024}
};
 
static struct genl_ops sprdwl_nl_ops[] = {
   {
       .cmd = SPRDWL_NL_CMD_NPI,
#if KERNEL_VERSION(5, 2, 0) > LINUX_VERSION_CODE
       .policy = sprdwl_genl_policy,
#endif
       .doit = sprdwl_nl_npi_handler,
   },
   {
       .cmd = SPRDWL_NL_CMD_GET_INFO,
#if KERNEL_VERSION(5, 2, 0) > LINUX_VERSION_CODE
       .policy = sprdwl_genl_policy,
#endif
       .doit = sprdwl_nl_get_info_handler,
   }
};
 
static struct genl_family sprdwl_nl_genl_family = {
   .id = SPRDWL_NL_GENERAL_SOCK_ID,
   .hdrsize = 0,
   .name = "SPRDWL_NL",
   .version = 1,
   .maxattr = SPRDWL_NL_ATTR_MAX,
   .pre_doit = sprdwl_npi_pre_doit,
   .post_doit = sprdwl_npi_post_doit,
#if LINUX_VERSION_CODE >= KERNEL_VERSION(4, 14, 0)
   .module = THIS_MODULE,
   .n_ops = ARRAY_SIZE(sprdwl_nl_ops),
   .ops = sprdwl_nl_ops,
#endif
};
 
static int sprdwl_nl_send_generic(struct genl_info *info, u8 attr,
                 u8 cmd, u32 len, u8 *data)
{
   struct sk_buff *skb;
   void *hdr;
   int ret;
 
   skb = nlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
   if (!skb)
       return -ENOMEM;
   hdr = genlmsg_put(skb, info->snd_portid, info->snd_seq,
             &sprdwl_nl_genl_family, 0, cmd);
   if (IS_ERR(hdr)) {
       ret = PTR_ERR(hdr);
       goto err_put;
   }
   if (nla_put(skb, attr, len, data)) {
       ret = -1;
       goto err_put;
   }
 
   genlmsg_end(skb, hdr);
   return genlmsg_reply(skb, info);
 
err_put:
   nlmsg_free(skb);
   return ret;
}
 
void sprdwl_init_npi(void)
{
#if LINUX_VERSION_CODE >= KERNEL_VERSION(4, 14, 0)
   int ret = genl_register_family(&sprdwl_nl_genl_family);
 
   if (ret)
       wl_err("genl_register_family error: %d\n", ret);
#else
#if LINUX_VERSION_CODE >= KERNEL_VERSION(3, 13, 0)
   int ret = genl_register_family_with_ops(&sprdwl_nl_genl_family,
                       sprdwl_nl_ops);
#else
   int ret = genl_register_family_with_ops(&sprdwl_nl_genl_family,
                       sprdwl_nl_ops,
                       ARRAY_SIZE(sprdwl_nl_ops));
#endif
   if (ret)
       wl_err("genl_register_family_with_ops error: %d\n", ret);
#endif
}
 
void sprdwl_deinit_npi(void)
{
   int ret = genl_unregister_family(&sprdwl_nl_genl_family);
 
   if (ret)
       wl_err("genl_unregister_family error:%d\n", ret);
}