| Index: syzygy/experimental/protect/protect_lib/protect_util.cc
|
| diff --git a/syzygy/experimental/protect/protect_lib/protect_util.cc b/syzygy/experimental/protect/protect_lib/protect_util.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..71df4c6712f6dc8af1df6a6b42ef37df916c718d
|
| --- /dev/null
|
| +++ b/syzygy/experimental/protect/protect_lib/protect_util.cc
|
| @@ -0,0 +1,51 @@
|
| +// Copyright 2015 Google Inc. All Rights Reserved.
|
| +//
|
| +// Licensed under the Apache License, Version 2.0 (the "License");
|
| +// you may not use this file except in compliance with the License.
|
| +// You may obtain a copy of the License at
|
| +//
|
| +// http://www.apache.org/licenses/LICENSE-2.0
|
| +//
|
| +// Unless required by applicable law or agreed to in writing, software
|
| +// distributed under the License is distributed on an "AS IS" BASIS,
|
| +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
| +// See the License for the specific language governing permissions and
|
| +// limitations under the License.
|
| +
|
| +#include "syzygy/protect/protect_lib/protect_util.h"
|
| +#include <algorithm>
|
| +#include <ctime>
|
| +#include <iostream>
|
| +#include <vector>
|
| +
|
| +unsigned int VectGenerator::_kSum = 256;
|
| +
|
| +std::vector<uint8>* VectGenerator::Generate(uint8 x, int len)
|
| +{
|
| + unsigned int max_sum = _kSum - x;
|
| + int rand_num;
|
| + std::vector<uint8> *values = new std::vector<uint8>();
|
| +
|
| + if (max_sum < 0) {
|
| + return NULL;
|
| + }
|
| +
|
| + std::srand(unsigned(std::time(0)));
|
| +
|
| + for (int i = 0; i < len - 1; ++i) {
|
| + rand_num = std::rand() % max_sum;
|
| + values->push_back(rand_num);
|
| +
|
| + max_sum -= rand_num;
|
| +
|
| + if (max_sum < 0) {
|
| + return NULL;
|
| + }
|
| + }
|
| +
|
| + values->push_back(max_sum);
|
| +
|
| + std::random_shuffle(values->begin(), values->end());
|
| +
|
| + return values;
|
| +}
|
|
|