| /* |
| * Copyright (c) 2012 Neratec Solutions AG |
| * |
| * Permission to use, copy, modify, and/or distribute this software for any |
| * purpose with or without fee is hereby granted, provided that the above |
| * copyright notice and this permission notice appear in all copies. |
| * |
| * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES |
| * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF |
| * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR |
| * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES |
| * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN |
| * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF |
| * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. |
| */ |
| |
| #include <linux/slab.h> |
| |
| #include "dfs_pattern_detector.h" |
| #include "dfs_pri_detector.h" |
| |
| /** |
| * struct pri_sequence - sequence of pulses matching one PRI |
| * @head: list_head |
| * @pri: pulse repetition interval (PRI) in usecs |
| * @dur: duration of sequence in usecs |
| * @count: number of pulses in this sequence |
| * @count_falses: number of not matching pulses in this sequence |
| * @first_ts: time stamp of first pulse in usecs |
| * @last_ts: time stamp of last pulse in usecs |
| * @deadline_ts: deadline when this sequence becomes invalid (first_ts + dur) |
| */ |
| struct pri_sequence { |
| struct list_head head; |
| u32 pri; |
| u32 dur; |
| u32 count; |
| u32 count_falses; |
| u64 first_ts; |
| u64 last_ts; |
| u64 deadline_ts; |
| }; |
| |
| /** |
| * struct pulse_elem - elements in pulse queue |
| * @ts: time stamp in usecs |
| */ |
| struct pulse_elem { |
| struct list_head head; |
| u64 ts; |
| }; |
| |
| /** |
| * pde_get_multiple() - get number of multiples considering a given tolerance |
| * @return factor if abs(val - factor*fraction) <= tolerance, 0 otherwise |
| */ |
| static u32 pde_get_multiple(u32 val, u32 fraction, u32 tolerance) |
| { |
| u32 remainder; |
| u32 factor; |
| u32 delta; |
| |
| if (fraction == 0) |
| return 0; |
| |
| delta = (val < fraction) ? (fraction - val) : (val - fraction); |
| |
| if (delta <= tolerance) |
| /* val and fraction are within tolerance */ |
| return 1; |
| |
| factor = val / fraction; |
| remainder = val % fraction; |
| if (remainder > tolerance) { |
| /* no exact match */ |
| if ((fraction - remainder) <= tolerance) |
| /* remainder is within tolerance */ |
| factor++; |
| else |
| factor = 0; |
| } |
| return factor; |
| } |
| |
| /** |
| * DOC: Singleton Pulse and Sequence Pools |
| * |
| * Instances of pri_sequence and pulse_elem are kept in singleton pools to |
| * reduce the number of dynamic allocations. They are shared between all |
| * instances and grow up to the peak number of simultaneously used objects. |
| * |
| * Memory is freed after all references to the pools are released. |
| */ |
| static u32 singleton_pool_references; |
| static LIST_HEAD(pulse_pool); |
| static LIST_HEAD(pseq_pool); |
| |
| static struct pulse_elem *pulse_queue_get_tail(struct pri_detector *pde) |
| { |
| struct list_head *l = &pde->pulses; |
| if (list_empty(l)) |
| return NULL; |
| return list_entry(l->prev, struct pulse_elem, head); |
| } |
| |
| static bool pulse_queue_dequeue(struct pri_detector *pde) |
| { |
| struct pulse_elem *p = pulse_queue_get_tail(pde); |
| if (p != NULL) { |
| list_del_init(&p->head); |
| pde->count--; |
| /* give it back to pool */ |
| list_add(&p->head, &pulse_pool); |
| } |
| return (pde->count > 0); |
| } |
| |
| /* remove pulses older than window */ |
| static void pulse_queue_check_window(struct pri_detector *pde) |
| { |
| u64 min_valid_ts; |
| struct pulse_elem *p; |
| |
| /* there is no delta time with less than 2 pulses */ |
| if (pde->count < 2) |
| return; |
| |
| if (pde->last_ts <= pde->window_size) |
| return; |
| |
| min_valid_ts = pde->last_ts - pde->window_size; |
| while ((p = pulse_queue_get_tail(pde)) != NULL) { |
| if (p->ts >= min_valid_ts) |
| return; |
| pulse_queue_dequeue(pde); |
| } |
| } |
| |
| static bool pulse_queue_enqueue(struct pri_detector *pde, u64 ts) |
| { |
| struct pulse_elem *p; |
| if (!list_empty(&pulse_pool)) { |
| p = list_first_entry(&pulse_pool, struct pulse_elem, head); |
| list_del(&p->head); |
| } else { |
| p = kmalloc(sizeof(*p), GFP_KERNEL); |
| if (p == NULL) { |
| pr_err("failed to allocate pulse_elem\n"); |
| return false; |
| } |
| } |
| INIT_LIST_HEAD(&p->head); |
| p->ts = ts; |
| list_add(&p->head, &pde->pulses); |
| pde->count++; |
| pde->last_ts = ts; |
| pulse_queue_check_window(pde); |
| if (pde->count >= pde->max_count) |
| pulse_queue_dequeue(pde); |
| return true; |
| } |
| |
| static bool pseq_handler_create_sequences(struct pri_detector *pde, |
| u64 ts, u32 min_count) |
| { |
| struct pulse_elem *p; |
| list_for_each_entry(p, &pde->pulses, head) { |
| struct pri_sequence ps, *new_ps; |
| struct pulse_elem *p2; |
| u32 tmp_false_count; |
| u64 min_valid_ts; |
| u32 delta_ts = ts - p->ts; |
| |
| if (delta_ts < pde->rs->pri_min) |
| /* ignore too small pri */ |
| continue; |
| |
| if (delta_ts > pde->rs->pri_max) |
| /* stop on too large pri (sorted list) */ |
| break; |
| |
| /* build a new sequence with new potential pri */ |
| ps.count = 2; |
| ps.count_falses = 0; |
| ps.first_ts = p->ts; |
| ps.last_ts = ts; |
| ps.pri = ts - p->ts; |
| ps.dur = ps.pri * (pde->rs->ppb - 1) |
| + 2 * pde->rs->max_pri_tolerance; |
| |
| p2 = p; |
| tmp_false_count = 0; |
| min_valid_ts = ts - ps.dur; |
| /* check which past pulses are candidates for new sequence */ |
| list_for_each_entry_continue(p2, &pde->pulses, head) { |
| u32 factor; |
| if (p2->ts < min_valid_ts) |
| /* stop on crossing window border */ |
| break; |
| /* check if pulse match (multi)PRI */ |
| factor = pde_get_multiple(ps.last_ts - p2->ts, ps.pri, |
| pde->rs->max_pri_tolerance); |
| if (factor > 0) { |
| ps.count++; |
| ps.first_ts = p2->ts; |
| /* |
| * on match, add the intermediate falses |
| * and reset counter |
| */ |
| ps.count_falses += tmp_false_count; |
| tmp_false_count = 0; |
| } else { |
| /* this is a potential false one */ |
| tmp_false_count++; |
| } |
| } |
| if (ps.count < min_count) |
| /* did not reach minimum count, drop sequence */ |
| continue; |
| |
| /* this is a valid one, add it */ |
| ps.deadline_ts = ps.first_ts + ps.dur; |
| |
| if (!list_empty(&pseq_pool)) { |
| new_ps = list_first_entry(&pseq_pool, |
| struct pri_sequence, head); |
| list_del(&new_ps->head); |
| } else { |
| new_ps = kmalloc(sizeof(*new_ps), GFP_KERNEL); |
| if (new_ps == NULL) |
| return false; |
| } |
| memcpy(new_ps, &ps, sizeof(ps)); |
| INIT_LIST_HEAD(&new_ps->head); |
| list_add(&new_ps->head, &pde->sequences); |
| } |
| return true; |
| } |
| |
| /* check new ts and add to all matching existing sequences */ |
| static u32 |
| pseq_handler_add_to_existing_seqs(struct pri_detector *pde, u64 ts) |
| { |
| u32 max_count = 0; |
| struct pri_sequence *ps, *ps2; |
| list_for_each_entry_safe(ps, ps2, &pde->sequences, head) { |
| u32 delta_ts; |
| u32 factor; |
| |
| /* first ensure that sequence is within window */ |
| if (ts > ps->deadline_ts) { |
| list_del_init(&ps->head); |
| list_add(&ps->head, &pseq_pool); |
| continue; |
| } |
| |
| delta_ts = ts - ps->last_ts; |
| factor = pde_get_multiple(delta_ts, ps->pri, |
| pde->rs->max_pri_tolerance); |
| if (factor > 0) { |
| ps->last_ts = ts; |
| ps->count++; |
| |
| if (max_count < ps->count) |
| max_count = ps->count; |
| } else { |
| ps->count_falses++; |
| } |
| } |
| return max_count; |
| } |
| |
| static struct pri_sequence * |
| pseq_handler_check_detection(struct pri_detector *pde) |
| { |
| struct pri_sequence *ps; |
| |
| if (list_empty(&pde->sequences)) |
| return NULL; |
| |
| list_for_each_entry(ps, &pde->sequences, head) { |
| /* |
| * we assume to have enough matching confidence if we |
| * 1) have enough pulses |
| * 2) have more matching than false pulses |
| */ |
| if ((ps->count >= pde->rs->ppb_thresh) && |
| (ps->count * pde->rs->num_pri >= ps->count_falses)) |
| return ps; |
| } |
| return NULL; |
| } |
| |
| |
| /* free pulse queue and sequences list and give objects back to pools */ |
| static void pri_detector_reset(struct pri_detector *pde, u64 ts) |
| { |
| struct pri_sequence *ps, *ps0; |
| struct pulse_elem *p, *p0; |
| list_for_each_entry_safe(ps, ps0, &pde->sequences, head) { |
| list_del_init(&ps->head); |
| list_add(&ps->head, &pseq_pool); |
| } |
| list_for_each_entry_safe(p, p0, &pde->pulses, head) { |
| list_del_init(&p->head); |
| list_add(&p->head, &pulse_pool); |
| } |
| pde->count = 0; |
| pde->last_ts = ts; |
| } |
| |
| static void pri_detector_exit(struct pri_detector *de) |
| { |
| pri_detector_reset(de, 0); |
| |
| singleton_pool_references--; |
| if (singleton_pool_references == 0) { |
| /* free singleton pools with no references left */ |
| struct pri_sequence *ps, *ps0; |
| struct pulse_elem *p, *p0; |
| |
| list_for_each_entry_safe(p, p0, &pulse_pool, head) { |
| list_del(&p->head); |
| kfree(p); |
| } |
| list_for_each_entry_safe(ps, ps0, &pseq_pool, head) { |
| list_del(&ps->head); |
| kfree(ps); |
| } |
| } |
| kfree(de); |
| } |
| |
| static bool pri_detector_add_pulse(struct pri_detector *de, |
| struct pulse_event *event) |
| { |
| u32 max_updated_seq; |
| struct pri_sequence *ps; |
| u64 ts = event->ts; |
| const struct radar_detector_specs *rs = de->rs; |
| |
| /* ignore pulses not within width range */ |
| if ((rs->width_min > event->width) || (rs->width_max < event->width)) |
| return false; |
| |
| if ((ts - de->last_ts) < rs->max_pri_tolerance) |
| /* if delta to last pulse is too short, don't use this pulse */ |
| return false; |
| de->last_ts = ts; |
| |
| max_updated_seq = pseq_handler_add_to_existing_seqs(de, ts); |
| |
| if (!pseq_handler_create_sequences(de, ts, max_updated_seq)) { |
| pr_err("failed to create pulse sequences\n"); |
| pri_detector_reset(de, ts); |
| return false; |
| } |
| |
| ps = pseq_handler_check_detection(de); |
| |
| if (ps != NULL) { |
| pr_info("DFS: radar found: pri=%d, count=%d, count_false=%d\n", |
| ps->pri, ps->count, ps->count_falses); |
| pri_detector_reset(de, ts); |
| return true; |
| } |
| pulse_queue_enqueue(de, ts); |
| return false; |
| } |
| |
| struct pri_detector * |
| pri_detector_init(const struct radar_detector_specs *rs) |
| { |
| struct pri_detector *de; |
| de = kzalloc(sizeof(*de), GFP_KERNEL); |
| if (de == NULL) |
| return NULL; |
| de->exit = pri_detector_exit; |
| de->add_pulse = pri_detector_add_pulse; |
| de->reset = pri_detector_reset; |
| |
| INIT_LIST_HEAD(&de->sequences); |
| INIT_LIST_HEAD(&de->pulses); |
| de->window_size = rs->pri_max * rs->ppb * rs->num_pri; |
| de->max_count = rs->ppb * 2; |
| de->rs = rs; |
| |
| singleton_pool_references++; |
| return de; |
| } |