Line data    Source code

       1             : /*
       2             :  * Copyright (c) 2016 Cisco and/or its affiliates.
       3             :  * Licensed under the Apache License, Version 2.0 (the "License");
       4             :  * you may not use this file except in compliance with the License.
       5             :  * You may obtain a copy of the License at:
       6             :  *
       7             :  *     http://www.apache.org/licenses/LICENSE-2.0
       8             :  *
       9             :  * Unless required by applicable law or agreed to in writing, software
      10             :  * distributed under the License is distributed on an "AS IS" BASIS,
      11             :  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
      12             :  * See the License for the specific language governing permissions and
      13             :  * limitations under the License.
      14             :  */
      15             : 
      16             : #include <vnet/util/refcount.h>
      17             : 
      18           3 : void __vlib_refcount_resize(vlib_refcount_per_cpu_t *per_cpu, u32 size)
      19             : {
      20           3 :   u32 *new_counter = 0, *old_counter;
      21           3 :   vec_validate(new_counter, size);
      22           3 :   vlib_refcount_lock(per_cpu->counter_lock);
      23           3 :   memcpy(new_counter, per_cpu->counters, vec_len(per_cpu->counters)*4);
      24           3 :   old_counter = per_cpu->counters;
      25           3 :   per_cpu->counters = new_counter;
      26           3 :   vlib_refcount_unlock(per_cpu->counter_lock);
      27           3 :   CLIB_MEMORY_BARRIER();
      28           3 :   vec_free(old_counter);
      29           3 : }
      30             : 
      31         455 : u64 vlib_refcount_get(vlib_refcount_t *r, u32 index)
      32             : {
      33         455 :   u64 count = 0;
      34         455 :   vlib_thread_main_t *tm = vlib_get_thread_main ();
      35             :   u32 thread_index;
      36         910 :   for (thread_index = 0; thread_index < tm->n_vlib_mains; thread_index++) {
      37         455 :     vlib_refcount_lock(r->per_cpu[thread_index].counter_lock);
      38         455 :     if (index < vec_len(r->per_cpu[thread_index].counters))
      39             :       {
      40         455 :         count += r->per_cpu[thread_index].counters[index];
      41             :       }
      42         455 :     vlib_refcount_unlock(r->per_cpu[thread_index].counter_lock);
      43             :   }
      44         455 :   return count;
      45             : }
      46             :