src/common/varint.rs

   1 /*
   2  * Copyright (c) 2017 Tilman Sauerbeck (tilman at code-monkey de)
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining
   5  * a copy of this software and associated documentation files (the
   6  * "Software"), to deal in the Software without restriction, including
   7  * without limitation the rights to use, copy, modify, merge, publish,
   8  * distribute, sublicense, and/or sell copies of the Software, and to
   9  * permit persons to whom the Software is furnished to do so, subject to
  10  * the following conditions:
  11  *
  12  * The above copyright notice and this permission notice shall be
  13  * included in all copies or substantial portions of the Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  16  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  17  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  18  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  19  * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  20  * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  21  * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  22  */
  23
  24 pub fn write_u32(buffer: &mut [u8], _n: u32) -> usize {
  25     let shift = 7;
  26     let mask = 1 << shift;
  27     let mut offset = 0;
  28     let mut n = _n;
  29
  30     while n >= mask {
  31         let b = (n | mask) as u8;
  32
  33         buffer[offset] = b;
  34         offset += 1;
  35
  36         n >>= shift;
  37     }
  38
  39     buffer[offset] = n as u8;
  40     offset += 1;
  41
  42     offset
  43 }
  44
  45 pub fn write_s32(buffer: &mut [u8], n: i32) -> usize {
  46     let u = (n << 1) ^ (n >> 31);
  47
  48     write_u32(buffer, u as u32)
  49 }