auth/auth-core/src/main/java/org/onap/aaf/auth/local/DataFile.java

   1 /**
   2  * ============LICENSE_START====================================================
   3  * org.onap.aaf
   4  * ===========================================================================
   5  * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
   6  * ===========================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END====================================================
  19  *
  20  */
  21
  22 package org.onap.aaf.auth.local;
  23
  24 import java.io.File;
  25 import java.io.FileNotFoundException;
  26 import java.io.IOException;
  27 import java.io.RandomAccessFile;
  28 import java.nio.ByteBuffer;
  29 import java.nio.IntBuffer;
  30 import java.nio.MappedByteBuffer;
  31 import java.nio.channels.FileChannel;
  32 import java.nio.channels.FileChannel.MapMode;
  33
  34 public class DataFile {
  35     private RandomAccessFile rafile;
  36     private FileChannel channel;
  37     public MappedByteBuffer mapBuff;
  38     private final File file;
  39     private final String access;
  40
  41     public DataFile(File file, String access)  {
  42         this.file = file;
  43         this.access = access;
  44     }
  45     public void open() throws IOException {
  46         if (!file.exists()) throw new FileNotFoundException();
  47         rafile = new RandomAccessFile(file,access);
  48         channel = rafile.getChannel();
  49         mapBuff = channel.map("r".equals(access)?MapMode.READ_ONLY:MapMode.READ_WRITE,0,channel.size());
  50     }
  51     public boolean isOpened() {
  52         return mapBuff!=null;
  53     }
  54     public void close() throws IOException {
  55         if (channel!=null){
  56             channel.close();
  57         }
  58         if (rafile!=null) {
  59             rafile.close();
  60         }
  61         mapBuff = null;
  62     }
  63
  64     public long size() throws IOException {
  65         return channel==null?0:channel.size();
  66     }
  67
  68     private synchronized int load(Token t) {
  69         int len = Math.min(mapBuff.limit()-t.next,t.buff.length);
  70         if (len>0) {
  71             mapBuff.position(t.next);
  72             mapBuff.get(t.buff,0,len);
  73         }
  74         return len<0?0:len;
  75     }
  76
  77     public class Token {
  78         private byte[] buff;
  79         int pos, next, end;
  80
  81         public Token(int size) {
  82             buff = new byte[size];
  83             pos = next = end = 0;
  84         }
  85
  86         public boolean pos(int to) {
  87             pos = next = to;
  88             return (end=load(this))>0;
  89         }
  90
  91         public boolean nextLine() {
  92             end = load(this);
  93             pos = next;
  94             for (int i=0;i<end;++i) {
  95                 if (buff[i]=='\n') {
  96                     end = i;
  97                     next += i+1;
  98                     return true;
  99                 }
 100             }
 101             return false;
 102         }
 103
 104         public IntBuffer getIntBuffer() {
 105             return ByteBuffer.wrap(buff).asIntBuffer();
 106         }
 107
 108         public String toString() {
 109             return new String(buff,0,end);
 110         }
 111
 112         public class Field {
 113             char delim;
 114             int idx;
 115             ByteBuffer bb;
 116
 117             public Field(char delimiter) {
 118                 delim = delimiter;
 119                 idx = 0;
 120                 bb = null;
 121             }
 122
 123             public Field reset() {
 124                 idx = 0;
 125                 return this;
 126             }
 127
 128             public String next() {
 129                 if (idx>=end)return null;
 130                 int start = idx;
 131                 byte c=0;
 132                 int endStr = -1;
 133                 while (idx<end && idx<buff.length && (c=buff[idx])!=delim && c!='\n') { // for DOS
 134                     if (c=='\r')endStr=idx;
 135                     ++idx;
 136                 }
 137
 138                 if (endStr<0) {
 139                     endStr=idx-start;
 140                 } else {
 141                     endStr=endStr-start;
 142                 }
 143                 ++idx;
 144                 return new String(buff,start,endStr);
 145             }
 146
 147             public String at(int fieldOffset) {
 148                 int start;
 149                 byte c=0;
 150                 for (int count = idx = start = 0; idx<end && idx<buff.length; ++idx) {
 151                     if ((c=buff[idx])==delim || c=='\n') {
 152                         if (count++ == fieldOffset) {
 153                             break;
 154                         }
 155                         start = idx+1;
 156                     }
 157                 }
 158                 return new String(buff,start,(idx-start-(c=='\r'?1:0)));
 159             }
 160
 161             public String atToEnd(int fieldOffset) {
 162                 int start;
 163                 byte c=0;
 164                 for (int count = idx = start = 0; idx<end && idx<buff.length; ++idx) {
 165                     if ((c=buff[idx])==delim || c=='\n') {
 166                         if (count++ == fieldOffset) {
 167                             break;
 168                         }
 169                         start = idx+1;
 170                     }
 171                 }
 172
 173                 for (; idx<end && idx<buff.length && (c=buff[idx])!='\n'; ++idx) {
 174                     ++idx;
 175                 }
 176                 return new String(buff,start,(idx-start-((c=='\r' || idx>=end)?1:0)));
 177             }
 178
 179         }
 180
 181         public int pos() {
 182             return pos;
 183         }
 184     }
 185
 186     public File file() {
 187         return file;
 188     }
 189
 190 }