1/*
2 *  Licensed to the Apache Software Foundation (ASF) under one or more
3 *  contributor license agreements.  See the NOTICE file distributed with
4 *  this work for additional information regarding copyright ownership.
5 *  The ASF licenses this file to You under the Apache License, Version 2.0
6 *  (the "License"); you may not use this file except in compliance with
7 *  the License.  You may obtain a copy of the License at
8 *
9 *     http://www.apache.org/licenses/LICENSE-2.0
10 *
11 *  Unless required by applicable law or agreed to in writing, software
12 *  distributed under the License is distributed on an "AS IS" BASIS,
13 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 *  See the License for the specific language governing permissions and
15 *  limitations under the License.
16 */
17
18/**
19 * @author Vera Y. Petrashkova
20 */
21
22package org.apache.harmony.crypto.tests.support;
23
24import java.security.InvalidAlgorithmParameterException;
25import java.security.InvalidKeyException;
26import java.security.Key;
27import java.security.spec.AlgorithmParameterSpec;
28
29import javax.crypto.MacSpi;
30import javax.crypto.spec.SecretKeySpec;
31
32/**
33 * Additional class for verification of MacGeneratorSpi and MacSpi
34 */
35
36public class MyMacSpi extends MacSpi {
37
38    private int length = 0;
39
40    @Override
41    protected int engineGetMacLength() {
42        return length;
43    }
44
45    @Override
46    protected void engineInit(Key key, AlgorithmParameterSpec params)
47            throws InvalidKeyException, InvalidAlgorithmParameterException {
48        if (params == null) {
49            if (!(key instanceof SecretKeySpec)) {
50                throw new IllegalArgumentException("params is null and key is SecretKeySpec");
51            }
52        }
53    }
54
55    @Override
56    protected void engineUpdate(byte input) {
57    }
58
59    @Override
60    protected void engineUpdate(byte[] input, int offset, int len) {
61        if (offset >= 0 && len >= 0) {
62            length = len;
63        }
64    }
65
66    @Override
67    protected byte[] engineDoFinal() {
68        return new byte[length];
69    }
70
71    @Override
72    protected void engineReset() {
73        length++;
74    }
75}