easy_pow 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: d04027026e757693648d73bcb5cb6f2afbee99fe
+  data.tar.gz: 276d10200683279b0a4e5300b0fc38e3d9cc7365
+SHA512:
+  metadata.gz: 8803663673acadacddad3153773306232928db56dd1058c9d2b77da5fc380089c5326e27c69a5a201301220b480d521927dd792543ce68bd1d014b3a17672580
+  data.tar.gz: ffaeab32708fefa98404c99e084fb6226d81fcbe7322f181384b87f77dd5812a447b78c2f7adf25b1caea256360ea6d0552045da96958c544bd465c5a093ac21

data/.gitignore

@@ -0,0 +1,9 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in easy_pow.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2017 nomeaning
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,60 @@
+# EasyPow
+
+Simple PoW System/Solver for CTFs.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'easy_pow'
+```
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install easy_pow
+
+## Usage
+
+### Client
+
+```ruby
+require 'easy_pow'
+TCPSocket.open('127.0.0.1', '1234') do |s|
+  s.puts EasyPow.solve(s.gets)
+end
+```
+
+### Server
+
+```ruby
+require 'easy_pow'
+
+# use STDIN / STDOUT
+exit 0 unless EasyPow.easy_pow(27) # 27bit SHA256 PoW
+
+# use TCPServer
+TCPServer.open('0.0.0.0', 1234) do |socket|
+    # ...
+    exit unless EasyPow.easy_pow(25, socket) # 25bit SHA256 PoW with Socket
+end
+```
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and tags, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/nomeaning777/easy_pow.
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](http://opensource.org/licenses/MIT).
+

data/Rakefile

@@ -0,0 +1,2 @@
+require "bundler/gem_tasks"
+task :default => :spec

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "easy_pow"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start(__FILE__)

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/easy_pow.gemspec

@@ -0,0 +1,36 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'easy_pow/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "easy_pow"
+  spec.version       = EasyPow::VERSION
+  spec.authors       = ["nomeaning"]
+  spec.email         = ["nomeaning777@gmail.com"]
+
+  spec.summary       = %q{Simple PoW for CTF}
+  spec.description   = %q{Simple PoW for CTF }
+  spec.homepage      = "http://github.com/nomeaning777/easy_pow"
+  spec.license       = "MIT"
+
+  # Prevent pushing this gem to RubyGems.org. To allow pushes either set the 'allowed_push_host'
+  # to allow pushing to a single host or delete this section to allow pushing to any host.
+  if spec.respond_to?(:metadata)
+    # spec.metadata['allowed_push_host'] = "TODO: Set to 'http://mygemserver.com'"
+  else
+    raise "RubyGems 2.0 or newer is required to protect against " \
+      "public gem pushes."
+  end
+
+  spec.files         = `git ls-files -z`.split("\x0").reject do |f|
+    f.match(%r{^(test|spec|features)/})
+  end
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib", "ext"]
+  spec.extensions << "ext/easy_pow/extconf.rb"
+
+  spec.add_development_dependency "bundler", "~> 1.14"
+  spec.add_development_dependency "rake", "~> 10.0"
+end

data/ext/easy_pow/Makefile

@@ -0,0 +1,263 @@
+
+SHELL = /bin/sh
+
+# V=0 quiet, V=1 verbose.  other values don't work.
+V = 0
+Q1 = $(V:1=)
+Q = $(Q1:0=@)
+ECHO1 = $(V:1=@ :)
+ECHO = $(ECHO1:0=@ echo)
+NULLCMD = :
+
+#### Start of system configuration section. ####
+
+srcdir = .
+topdir = /home/nomeaning/.rbenv/versions/2.4.0/include/ruby-2.4.0
+hdrdir = $(topdir)
+arch_hdrdir = /home/nomeaning/.rbenv/versions/2.4.0/include/ruby-2.4.0/x86_64-linux
+PATH_SEPARATOR = :
+VPATH = $(srcdir):$(arch_hdrdir)/ruby:$(hdrdir)/ruby
+prefix = $(DESTDIR)/home/nomeaning/.rbenv/versions/2.4.0
+rubysitearchprefix = $(rubylibprefix)/$(sitearch)
+rubyarchprefix = $(rubylibprefix)/$(arch)
+rubylibprefix = $(libdir)/$(RUBY_BASE_NAME)
+exec_prefix = $(prefix)
+vendorarchhdrdir = $(vendorhdrdir)/$(sitearch)
+sitearchhdrdir = $(sitehdrdir)/$(sitearch)
+rubyarchhdrdir = $(rubyhdrdir)/$(arch)
+vendorhdrdir = $(rubyhdrdir)/vendor_ruby
+sitehdrdir = $(rubyhdrdir)/site_ruby
+rubyhdrdir = $(includedir)/$(RUBY_VERSION_NAME)
+vendorarchdir = $(vendorlibdir)/$(sitearch)
+vendorlibdir = $(vendordir)/$(ruby_version)
+vendordir = $(rubylibprefix)/vendor_ruby
+sitearchdir = $(sitelibdir)/$(sitearch)
+sitelibdir = $(sitedir)/$(ruby_version)
+sitedir = $(rubylibprefix)/site_ruby
+rubyarchdir = $(rubylibdir)/$(arch)
+rubylibdir = $(rubylibprefix)/$(ruby_version)
+sitearchincludedir = $(includedir)/$(sitearch)
+archincludedir = $(includedir)/$(arch)
+sitearchlibdir = $(libdir)/$(sitearch)
+archlibdir = $(libdir)/$(arch)
+ridir = $(datarootdir)/$(RI_BASE_NAME)
+mandir = $(datarootdir)/man
+localedir = $(datarootdir)/locale
+libdir = $(exec_prefix)/lib
+psdir = $(docdir)
+pdfdir = $(docdir)
+dvidir = $(docdir)
+htmldir = $(docdir)
+infodir = $(datarootdir)/info
+docdir = $(datarootdir)/doc/$(PACKAGE)
+oldincludedir = $(DESTDIR)/usr/include
+includedir = $(prefix)/include
+localstatedir = $(prefix)/var
+sharedstatedir = $(prefix)/com
+sysconfdir = $(prefix)/etc
+datadir = $(datarootdir)
+datarootdir = $(prefix)/share
+libexecdir = $(exec_prefix)/libexec
+sbindir = $(exec_prefix)/sbin
+bindir = $(exec_prefix)/bin
+archdir = $(rubyarchdir)
+
+
+CC = gcc
+CXX = g++
+LIBRUBY = $(LIBRUBY_A)
+LIBRUBY_A = lib$(RUBY_SO_NAME)-static.a
+LIBRUBYARG_SHARED = -Wl,-R$(libdir) -L$(libdir) 
+LIBRUBYARG_STATIC = -Wl,-R$(libdir) -L$(libdir) -l$(RUBY_SO_NAME)-static
+empty =
+OUTFLAG = -o $(empty)
+COUTFLAG = -o $(empty)
+CSRCFLAG = $(empty)
+
+RUBY_EXTCONF_H = 
+cflags   = $(optflags) $(debugflags) $(warnflags)
+cxxflags = $(optflags) $(debugflags) $(warnflags)
+optflags = -O3 -fno-fast-math
+debugflags = -ggdb3
+warnflags = -Wall -Wextra -Wno-unused-parameter -Wno-parentheses -Wno-long-long -Wno-missing-field-initializers -Wno-tautological-compare -Wno-parentheses-equality -Wno-constant-logical-operand -Wno-self-assign -Wunused-variable -Wimplicit-int -Wpointer-arith -Wwrite-strings -Wdeclaration-after-statement -Wimplicit-function-declaration -Wdeprecated-declarations -Wno-packed-bitfield-compat -Wsuggest-attribute=noreturn -Wsuggest-attribute=format
+CCDLFLAGS = -fPIC
+CFLAGS   = $(CCDLFLAGS) $(cflags) -O3 -std=c11 -march=native -fopenmp $(ARCH_FLAG)
+INCFLAGS = -I. -I$(arch_hdrdir) -I$(hdrdir)/ruby/backward -I$(hdrdir) -I$(srcdir)
+DEFS     = 
+CPPFLAGS = -DHAVE_OPENSSL_SHA_H -I/home/nomeaning/.rbenv/versions/2.4.0/include  $(DEFS) $(cppflags)
+CXXFLAGS = $(CCDLFLAGS) $(cxxflags) $(ARCH_FLAG)
+ldflags  = -L. -L/home/nomeaning/.rbenv/versions/2.4.0/lib  -fstack-protector -rdynamic -Wl,-export-dynamic
+dldflags = -L/home/nomeaning/.rbenv/versions/2.4.0/lib  
+ARCH_FLAG = 
+DLDFLAGS = $(ldflags) $(dldflags) $(ARCH_FLAG)
+LDSHARED = $(CC) -shared
+LDSHAREDXX = $(CXX) -shared
+AR = ar
+EXEEXT = 
+
+RUBY_INSTALL_NAME = $(RUBY_BASE_NAME)
+RUBY_SO_NAME = ruby
+RUBYW_INSTALL_NAME = 
+RUBY_VERSION_NAME = $(RUBY_BASE_NAME)-$(ruby_version)
+RUBYW_BASE_NAME = rubyw
+RUBY_BASE_NAME = ruby
+
+arch = x86_64-linux
+sitearch = $(arch)
+ruby_version = 2.4.0
+ruby = $(bindir)/$(RUBY_BASE_NAME)
+RUBY = $(ruby)
+ruby_headers = $(hdrdir)/ruby.h $(hdrdir)/ruby/backward.h $(hdrdir)/ruby/ruby.h $(hdrdir)/ruby/defines.h $(hdrdir)/ruby/missing.h $(hdrdir)/ruby/intern.h $(hdrdir)/ruby/st.h $(hdrdir)/ruby/subst.h $(arch_hdrdir)/ruby/config.h
+
+RM = rm -f
+RM_RF = $(RUBY) -run -e rm -- -rf
+RMDIRS = rmdir --ignore-fail-on-non-empty -p
+MAKEDIRS = /bin/mkdir -p
+INSTALL = /usr/bin/install -c
+INSTALL_PROG = $(INSTALL) -m 0755
+INSTALL_DATA = $(INSTALL) -m 644
+COPY = cp
+TOUCH = exit >
+
+#### End of system configuration section. ####
+
+preload = 
+libpath = . $(libdir)
+LIBPATH =  -L. -L$(libdir) -Wl,-R$(libdir)
+DEFFILE = 
+
+CLEANFILES = mkmf.log
+DISTCLEANFILES = 
+DISTCLEANDIRS = 
+
+extout = 
+extout_prefix = 
+target_prefix = /easy_pow
+LOCAL_LIBS = 
+LIBS =  -lgomp -lcrypto -lstdc++  -lpthread -lgmp -ldl -lcrypt -lm   -lc
+ORIG_SRCS = ext.c
+SRCS = $(ORIG_SRCS) 
+OBJS = ext.o
+HDRS = 
+LOCAL_HDRS = 
+TARGET = ext
+TARGET_NAME = ext
+TARGET_ENTRY = Init_$(TARGET_NAME)
+DLLIB = $(TARGET).so
+EXTSTATIC = 
+STATIC_LIB = 
+
+TIMESTAMP_DIR = .
+BINDIR        = $(bindir)
+RUBYCOMMONDIR = $(sitedir)$(target_prefix)
+RUBYLIBDIR    = $(sitelibdir)$(target_prefix)
+RUBYARCHDIR   = $(sitearchdir)$(target_prefix)
+HDRDIR        = $(rubyhdrdir)/ruby$(target_prefix)
+ARCHHDRDIR    = $(rubyhdrdir)/$(arch)/ruby$(target_prefix)
+TARGET_SO_DIR =
+TARGET_SO     = $(TARGET_SO_DIR)$(DLLIB)
+CLEANLIBS     = $(TARGET_SO) 
+CLEANOBJS     = *.o  *.bak
+
+all:    $(DLLIB)
+static: $(STATIC_LIB) install-rb
+.PHONY: all install static install-so install-rb
+.PHONY: clean clean-so clean-static clean-rb
+
+clean-static::
+clean-rb-default::
+clean-rb::
+clean-so::
+clean: clean-so clean-static clean-rb-default clean-rb
+		-$(Q)$(RM) $(CLEANLIBS) $(CLEANOBJS) $(CLEANFILES) .*.time
+
+distclean-rb-default::
+distclean-rb::
+distclean-so::
+distclean-static::
+distclean: clean distclean-so distclean-static distclean-rb-default distclean-rb
+		-$(Q)$(RM) Makefile $(RUBY_EXTCONF_H) conftest.* mkmf.log
+		-$(Q)$(RM) core ruby$(EXEEXT) *~ $(DISTCLEANFILES)
+		-$(Q)$(RMDIRS) $(DISTCLEANDIRS) 2> /dev/null || true
+
+realclean: distclean
+install: install-so install-rb
+
+install-so: $(DLLIB) $(TIMESTAMP_DIR)/.sitearchdir.-.easy_pow.time
+	$(INSTALL_PROG) $(DLLIB) $(RUBYARCHDIR)
+clean-static::
+	-$(Q)$(RM) $(STATIC_LIB)
+install-rb: pre-install-rb do-install-rb install-rb-default
+install-rb-default: pre-install-rb-default do-install-rb-default
+pre-install-rb: Makefile
+pre-install-rb-default: Makefile
+do-install-rb:
+do-install-rb-default:
+pre-install-rb-default:
+	@$(NULLCMD)
+$(TIMESTAMP_DIR)/.sitearchdir.-.easy_pow.time:
+	$(Q) $(MAKEDIRS) $(@D) $(RUBYARCHDIR)
+	$(Q) $(TOUCH) $@
+
+site-install: site-install-so site-install-rb
+site-install-so: install-so
+site-install-rb: install-rb
+
+.SUFFIXES: .c .m .cc .mm .cxx .cpp .o .S
+
+.cc.o:
+	$(ECHO) compiling $(<)
+	$(Q) $(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) $(COUTFLAG)$@ -c $(CSRCFLAG)$<
+
+.cc.S:
+	$(ECHO) translating $(<)
+	$(Q) $(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) $(COUTFLAG)$@ -S $(CSRCFLAG)$<
+
+.mm.o:
+	$(ECHO) compiling $(<)
+	$(Q) $(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) $(COUTFLAG)$@ -c $(CSRCFLAG)$<
+
+.mm.S:
+	$(ECHO) translating $(<)
+	$(Q) $(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) $(COUTFLAG)$@ -S $(CSRCFLAG)$<
+
+.cxx.o:
+	$(ECHO) compiling $(<)
+	$(Q) $(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) $(COUTFLAG)$@ -c $(CSRCFLAG)$<
+
+.cxx.S:
+	$(ECHO) translating $(<)
+	$(Q) $(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) $(COUTFLAG)$@ -S $(CSRCFLAG)$<
+
+.cpp.o:
+	$(ECHO) compiling $(<)
+	$(Q) $(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) $(COUTFLAG)$@ -c $(CSRCFLAG)$<
+
+.cpp.S:
+	$(ECHO) translating $(<)
+	$(Q) $(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) $(COUTFLAG)$@ -S $(CSRCFLAG)$<
+
+.c.o:
+	$(ECHO) compiling $(<)
+	$(Q) $(CC) $(INCFLAGS) $(CPPFLAGS) $(CFLAGS) $(COUTFLAG)$@ -c $(CSRCFLAG)$<
+
+.c.S:
+	$(ECHO) translating $(<)
+	$(Q) $(CC) $(INCFLAGS) $(CPPFLAGS) $(CFLAGS) $(COUTFLAG)$@ -S $(CSRCFLAG)$<
+
+.m.o:
+	$(ECHO) compiling $(<)
+	$(Q) $(CC) $(INCFLAGS) $(CPPFLAGS) $(CFLAGS) $(COUTFLAG)$@ -c $(CSRCFLAG)$<
+
+.m.S:
+	$(ECHO) translating $(<)
+	$(Q) $(CC) $(INCFLAGS) $(CPPFLAGS) $(CFLAGS) $(COUTFLAG)$@ -S $(CSRCFLAG)$<
+
+$(TARGET_SO): $(OBJS) Makefile
+	$(ECHO) linking shared-object easy_pow/$(DLLIB)
+	-$(Q)$(RM) $(@)
+	$(Q) $(LDSHARED) -o $@ $(OBJS) $(LIBPATH) $(DLDFLAGS) $(LOCAL_LIBS) $(LIBS)
+
+
+
+$(OBJS): $(HDRS) $(ruby_headers)

data/ext/easy_pow/ext.c

@@ -0,0 +1,173 @@
+#include <openssl/sha.h>
+#include <openssl/md5.h>
+#include "ruby.h"
+#include <stdint.h>
+#include <assert.h>
+#include <stdlib.h>
+#include <mm_malloc.h>
+
+struct search_condition {
+  unsigned char (*hash)(const unsigned char *d, size_t n, unsigned char *md);
+  size_t md_len;
+  unsigned char * restrict mask;
+  unsigned char *target;
+  unsigned char *result;
+  size_t chars_len;
+  unsigned char *chars;
+  size_t search_length;
+  size_t length;
+};
+
+unsigned char * safe_malloc(size_t length) {
+  unsigned char *data = _mm_malloc(length + 15, 16);
+  if(data == 0) {
+    rb_raise(rb_eNoMemError, "Failed to allocate memory");
+  }
+  return data;
+}
+
+void search(unsigned char *data, size_t pos, unsigned char * restrict md, struct search_condition *cond) {
+  if(cond->result != NULL) return; // Found Already
+  if(pos == cond->search_length) {
+	  int64_t *mdc = __builtin_assume_aligned(md, 16);
+	  int64_t *mask = __builtin_assume_aligned(cond->mask, 16);
+    // ハッシュ関数
+    (cond->hash)(data, cond->length, md);
+
+    for(size_t i = 0; i < cond->md_len / 8; i++) {
+      mdc[i] &= mask[i];
+    }
+
+    if(memcmp(md, cond->target, cond->md_len) == 0) {
+      #pragma omp critical
+      {
+        if(cond->result == NULL) {
+          cond->result = malloc(cond->length);
+          if(cond->result == 0) {
+            rb_raise(rb_eNoMemError, "Failed to allocate memory");
+          }
+          memcpy(cond->result, data, cond->length);
+        }
+      }
+    }
+  } else {
+    for(size_t i = 0; i < cond->chars_len; i++) {
+      data[pos] = cond->chars[i];
+      search(data, pos + 1, md, cond);
+    }
+  }
+}
+
+void search_parallel(unsigned char *data, size_t pos, unsigned char * restrict md, struct search_condition *cond) {
+#pragma omp parallel for
+  for(size_t i = 0; i < cond->chars_len; i++) {
+    unsigned char *data2 = safe_malloc(cond->length);
+    unsigned char *md2 = safe_malloc(cond->md_len);
+    memcpy(data2, data, cond->length);
+    data2[pos] = cond->chars[i];
+    search(data2, pos + 1, md2, cond);
+    _mm_free(data2);
+    _mm_free(md2);
+  }
+}
+
+
+static VALUE search_general(VALUE self, VALUE prefix, VALUE suffix, VALUE length, VALUE target, VALUE mask, VALUE chars, VALUE paralell, struct search_condition *cond) {
+  Check_Type(prefix, T_STRING);
+  Check_Type(suffix, T_STRING);
+  FIXNUM_P(length);
+  Check_Type(target, T_STRING);
+  Check_Type(mask, T_STRING);
+  Check_Type(chars, T_STRING);
+  if((size_t)RSTRING_LEN(mask) != cond->md_len)
+    rb_raise(rb_eArgError, "Invalid Mask length");
+  if((size_t)RSTRING_LEN(target) != cond->md_len)
+    rb_raise(rb_eArgError, "Invalid Target length");
+  cond->mask = safe_malloc(cond->md_len); 
+  memcpy(cond->mask, (unsigned char*)RSTRING_PTR(mask), cond->md_len);
+  cond->target = (unsigned char*)RSTRING_PTR(target);
+  cond->result = NULL;
+  cond->chars = (unsigned char*)RSTRING_PTR(chars);
+  cond->chars_len = RSTRING_LEN(chars);
+  cond->search_length = FIX2INT(length) + RSTRING_LEN(prefix);
+  cond->length = cond->search_length + RSTRING_LEN(suffix);
+#ifdef _OPENMP
+  if(RTEST(paralell) && length > 0) { // openmp
+    unsigned char *data = safe_malloc(cond->length);
+    unsigned char *md = safe_malloc(cond->md_len);
+    memcpy(data, RSTRING_PTR(prefix), RSTRING_LEN(prefix));
+    memcpy(data + cond->search_length, RSTRING_PTR(suffix), RSTRING_LEN(suffix));
+    search_parallel(data, RSTRING_LEN(prefix), md, cond);
+    _mm_free(data);
+    _mm_free(md);
+  } else {
+#endif
+    unsigned char *data = safe_malloc(cond->length);
+    unsigned char *md = safe_malloc(cond->md_len);
+    memcpy(data, RSTRING_PTR(prefix), RSTRING_LEN(prefix));
+    memcpy(data + cond->search_length, RSTRING_PTR(suffix), RSTRING_LEN(suffix));
+    search(data, RSTRING_LEN(prefix), md, cond);
+    _mm_free(data);
+    _mm_free(md);
+#ifdef _OPENMP
+  }
+#endif
+  _mm_free(cond->mask);
+  if(cond->result) {
+    return rb_str_new((char*)cond->result, cond->length);
+  } else {
+    return Qnil;
+  }
+}
+
+static VALUE search_md5(VALUE self, VALUE prefix, VALUE suffix, VALUE length, VALUE target, VALUE mask, VALUE chars, VALUE paralell) {
+  struct search_condition cond;
+  cond.hash = MD5;
+  cond.md_len = 128 / 8;
+  return search_general(self, prefix, suffix, length, target, mask, chars, paralell, &cond);
+}
+
+static VALUE search_sha1(VALUE self, VALUE prefix, VALUE suffix, VALUE length, VALUE target, VALUE mask, VALUE chars, VALUE paralell) {
+  struct search_condition cond;
+  cond.hash = SHA1;
+  cond.md_len = 160 / 8;
+  return search_general(self, prefix, suffix, length, target, mask, chars, paralell, &cond);
+}
+
+static VALUE search_sha224(VALUE self, VALUE prefix, VALUE suffix, VALUE length, VALUE target, VALUE mask, VALUE chars, VALUE paralell) {
+  struct search_condition cond;
+  cond.hash = SHA224;
+  cond.md_len = 224 / 8;
+  return search_general(self, prefix, suffix, length, target, mask, chars, paralell, &cond);
+}
+
+static VALUE search_sha256(VALUE self, VALUE prefix, VALUE suffix, VALUE length, VALUE target, VALUE mask, VALUE chars, VALUE paralell) {
+  struct search_condition cond;
+  cond.hash = SHA256;
+  cond.md_len = 256 / 8;
+  return search_general(self, prefix, suffix, length, target, mask, chars, paralell, &cond);
+}
+
+static VALUE search_sha384(VALUE self, VALUE prefix, VALUE suffix, VALUE length, VALUE target, VALUE mask, VALUE chars, VALUE paralell) {
+  struct search_condition cond;
+  cond.hash = SHA384;
+  cond.md_len = 384 / 8;
+  return search_general(self, prefix, suffix, length, target, mask, chars, paralell, &cond);
+}
+
+static VALUE search_sha512(VALUE self, VALUE prefix, VALUE suffix, VALUE length, VALUE target, VALUE mask, VALUE chars, VALUE paralell) {
+  struct search_condition cond;
+  cond.hash = SHA512;
+  cond.md_len = 512 / 8;
+  return search_general(self, prefix, suffix, length, target, mask, chars, paralell, &cond);
+}
+void Init_ext()
+{
+  VALUE module = rb_define_module("EasyPow");
+  rb_define_module_function(module, "search_md5_ext", search_md5, 7);
+  rb_define_module_function(module, "search_sh1_ext", search_sha1, 7);
+  rb_define_module_function(module, "search_sha224_ext", search_sha224, 7);
+  rb_define_module_function(module, "search_sha256_ext", search_sha256, 7);
+  rb_define_module_function(module, "search_sha384_ext", search_sha384, 7);
+  rb_define_module_function(module, "search_sha512_ext", search_sha512, 7);
+}

data/ext/easy_pow/extconf.rb

@@ -0,0 +1,8 @@
+require 'mkmf'
+$CFLAGS += " -O3 -std=c11 -march=native -fopenmp" 
+
+have_library("stdc++")
+have_header('openssl/sha.h')
+have_library('crypto')
+have_library('gomp')
+create_makefile("easy_pow/ext")  

data/ext/easy_pow/mkmf.log

@@ -0,0 +1,92 @@
+have_library: checking for -lstdc++... -------------------- yes
+
+"gcc -o conftest -I/home/nomeaning/.rbenv/versions/2.4.0/include/ruby-2.4.0/x86_64-linux -I/home/nomeaning/.rbenv/versions/2.4.0/include/ruby-2.4.0/ruby/backward -I/home/nomeaning/.rbenv/versions/2.4.0/include/ruby-2.4.0 -I. -I/home/nomeaning/.rbenv/versions/2.4.0/include    -O3 -fno-fast-math -ggdb3 -Wall -Wextra -Wno-unused-parameter -Wno-parentheses -Wno-long-long -Wno-missing-field-initializers -Wno-tautological-compare -Wno-parentheses-equality -Wno-constant-logical-operand -Wno-self-assign -Wunused-variable -Wimplicit-int -Wpointer-arith -Wwrite-strings -Wdeclaration-after-statement -Wimplicit-function-declaration -Wdeprecated-declarations -Wno-packed-bitfield-compat -Wsuggest-attribute=noreturn -Wsuggest-attribute=format -O3 -std=c11 -march=native -fopenmp conftest.c  -L. -L/home/nomeaning/.rbenv/versions/2.4.0/lib -Wl,-R/home/nomeaning/.rbenv/versions/2.4.0/lib -L. -L/home/nomeaning/.rbenv/versions/2.4.0/lib  -fstack-protector -rdynamic -Wl,-export-dynamic     -Wl,-R/home/nomeaning/.rbenv/versions/2.4.0/lib -L/home/nomeaning/.rbenv/versions/2.4.0/lib -lruby-static  -lpthread -lgmp -ldl -lcrypt -lm   -lc"
+checked program was:
+/* begin */
+1: #include "ruby.h"
+2: 
+3: int main(int argc, char **argv)
+4: {
+5:   return 0;
+6: }
+/* end */
+
+"gcc -o conftest -I/home/nomeaning/.rbenv/versions/2.4.0/include/ruby-2.4.0/x86_64-linux -I/home/nomeaning/.rbenv/versions/2.4.0/include/ruby-2.4.0/ruby/backward -I/home/nomeaning/.rbenv/versions/2.4.0/include/ruby-2.4.0 -I. -I/home/nomeaning/.rbenv/versions/2.4.0/include    -O3 -fno-fast-math -ggdb3 -Wall -Wextra -Wno-unused-parameter -Wno-parentheses -Wno-long-long -Wno-missing-field-initializers -Wno-tautological-compare -Wno-parentheses-equality -Wno-constant-logical-operand -Wno-self-assign -Wunused-variable -Wimplicit-int -Wpointer-arith -Wwrite-strings -Wdeclaration-after-statement -Wimplicit-function-declaration -Wdeprecated-declarations -Wno-packed-bitfield-compat -Wsuggest-attribute=noreturn -Wsuggest-attribute=format -O3 -std=c11 -march=native -fopenmp conftest.c  -L. -L/home/nomeaning/.rbenv/versions/2.4.0/lib -Wl,-R/home/nomeaning/.rbenv/versions/2.4.0/lib -L. -L/home/nomeaning/.rbenv/versions/2.4.0/lib  -fstack-protector -rdynamic -Wl,-export-dynamic     -Wl,-R/home/nomeaning/.rbenv/versions/2.4.0/lib -L/home/nomeaning/.rbenv/versions/2.4.0/lib -lruby-static -lstdc++  -lpthread -lgmp -ldl -lcrypt -lm   -lc"
+checked program was:
+/* begin */
+ 1: #include "ruby.h"
+ 2: 
+ 3: /*top*/
+ 4: extern int t(void);
+ 5: int main(int argc, char **argv)
+ 6: {
+ 7:   if (argc > 1000000) {
+ 8:     printf("%p", &t);
+ 9:   }
+10: 
+11:   return 0;
+12: }
+13: 
+14: int t(void) { ; return 0; }
+/* end */
+
+--------------------
+
+have_header: checking for openssl/sha.h... -------------------- yes
+
+"gcc -E -I/home/nomeaning/.rbenv/versions/2.4.0/include/ruby-2.4.0/x86_64-linux -I/home/nomeaning/.rbenv/versions/2.4.0/include/ruby-2.4.0/ruby/backward -I/home/nomeaning/.rbenv/versions/2.4.0/include/ruby-2.4.0 -I. -I/home/nomeaning/.rbenv/versions/2.4.0/include    -O3 -fno-fast-math -ggdb3 -Wall -Wextra -Wno-unused-parameter -Wno-parentheses -Wno-long-long -Wno-missing-field-initializers -Wno-tautological-compare -Wno-parentheses-equality -Wno-constant-logical-operand -Wno-self-assign -Wunused-variable -Wimplicit-int -Wpointer-arith -Wwrite-strings -Wdeclaration-after-statement -Wimplicit-function-declaration -Wdeprecated-declarations -Wno-packed-bitfield-compat -Wsuggest-attribute=noreturn -Wsuggest-attribute=format -O3 -std=c11 -march=native -fopenmp  conftest.c -o conftest.i"
+checked program was:
+/* begin */
+1: #include "ruby.h"
+2: 
+3: #include <openssl/sha.h>
+/* end */
+
+--------------------
+
+have_library: checking for -lcrypto... -------------------- yes
+
+"gcc -o conftest -I/home/nomeaning/.rbenv/versions/2.4.0/include/ruby-2.4.0/x86_64-linux -I/home/nomeaning/.rbenv/versions/2.4.0/include/ruby-2.4.0/ruby/backward -I/home/nomeaning/.rbenv/versions/2.4.0/include/ruby-2.4.0 -I. -I/home/nomeaning/.rbenv/versions/2.4.0/include    -O3 -fno-fast-math -ggdb3 -Wall -Wextra -Wno-unused-parameter -Wno-parentheses -Wno-long-long -Wno-missing-field-initializers -Wno-tautological-compare -Wno-parentheses-equality -Wno-constant-logical-operand -Wno-self-assign -Wunused-variable -Wimplicit-int -Wpointer-arith -Wwrite-strings -Wdeclaration-after-statement -Wimplicit-function-declaration -Wdeprecated-declarations -Wno-packed-bitfield-compat -Wsuggest-attribute=noreturn -Wsuggest-attribute=format -O3 -std=c11 -march=native -fopenmp conftest.c  -L. -L/home/nomeaning/.rbenv/versions/2.4.0/lib -Wl,-R/home/nomeaning/.rbenv/versions/2.4.0/lib -L. -L/home/nomeaning/.rbenv/versions/2.4.0/lib  -fstack-protector -rdynamic -Wl,-export-dynamic    -lstdc++  -Wl,-R/home/nomeaning/.rbenv/versions/2.4.0/lib -L/home/nomeaning/.rbenv/versions/2.4.0/lib -lruby-static -lcrypto -lstdc++  -lpthread -lgmp -ldl -lcrypt -lm   -lc"
+checked program was:
+/* begin */
+ 1: #include "ruby.h"
+ 2: 
+ 3: /*top*/
+ 4: extern int t(void);
+ 5: int main(int argc, char **argv)
+ 6: {
+ 7:   if (argc > 1000000) {
+ 8:     printf("%p", &t);
+ 9:   }
+10: 
+11:   return 0;
+12: }
+13: 
+14: int t(void) { ; return 0; }
+/* end */
+
+--------------------
+
+have_library: checking for -lgomp... -------------------- yes
+
+"gcc -o conftest -I/home/nomeaning/.rbenv/versions/2.4.0/include/ruby-2.4.0/x86_64-linux -I/home/nomeaning/.rbenv/versions/2.4.0/include/ruby-2.4.0/ruby/backward -I/home/nomeaning/.rbenv/versions/2.4.0/include/ruby-2.4.0 -I. -I/home/nomeaning/.rbenv/versions/2.4.0/include    -O3 -fno-fast-math -ggdb3 -Wall -Wextra -Wno-unused-parameter -Wno-parentheses -Wno-long-long -Wno-missing-field-initializers -Wno-tautological-compare -Wno-parentheses-equality -Wno-constant-logical-operand -Wno-self-assign -Wunused-variable -Wimplicit-int -Wpointer-arith -Wwrite-strings -Wdeclaration-after-statement -Wimplicit-function-declaration -Wdeprecated-declarations -Wno-packed-bitfield-compat -Wsuggest-attribute=noreturn -Wsuggest-attribute=format -O3 -std=c11 -march=native -fopenmp conftest.c  -L. -L/home/nomeaning/.rbenv/versions/2.4.0/lib -Wl,-R/home/nomeaning/.rbenv/versions/2.4.0/lib -L. -L/home/nomeaning/.rbenv/versions/2.4.0/lib  -fstack-protector -rdynamic -Wl,-export-dynamic    -lcrypto -lstdc++  -Wl,-R/home/nomeaning/.rbenv/versions/2.4.0/lib -L/home/nomeaning/.rbenv/versions/2.4.0/lib -lruby-static -lgomp -lcrypto -lstdc++  -lpthread -lgmp -ldl -lcrypt -lm   -lc"
+checked program was:
+/* begin */
+ 1: #include "ruby.h"
+ 2: 
+ 3: /*top*/
+ 4: extern int t(void);
+ 5: int main(int argc, char **argv)
+ 6: {
+ 7:   if (argc > 1000000) {
+ 8:     printf("%p", &t);
+ 9:   }
+10: 
+11:   return 0;
+12: }
+13: 
+14: int t(void) { ; return 0; }
+/* end */
+
+--------------------
+

data/lib/easy_pow/version.rb

@@ -0,0 +1,3 @@
+module EasyPow
+  VERSION = "0.1.0"
+end

data/lib/easy_pow.rb

@@ -0,0 +1,73 @@
+require 'easy_pow/version'
+require 'easy_pow/ext'
+require 'securerandom'
+require 'digest/sha2'
+
+module EasyPow
+  @@hashes = [
+    ['sha1', 160],
+    ['md5', 128],
+    ['sha224', 224],
+    ['sha256', 256],
+    ['sha384', 384],
+    ['sha512', 512]
+  ]
+
+  def search_prefix(hash, bin, length, prefix, suffix = '', chars = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789', parallel = true)
+    len = @@hashes.select{|a|a[0] == hash.to_s.downcase}[0]
+    raise ArgumentError.new('Unknown hash type: %s' % [hash.to_s.downcase]) unless len
+    len = len[1] / 8
+    raise ArgumentError.new('Too long prefix') if bin.size > len * 8
+    target = Array.new(len){0}
+    mask = Array.new(len){0}
+    bin.each_char.with_index do |b, i|
+      mask[i / 8] |= 1 << (7 - i % 8)
+      target[i / 8] |= b.to_i << (7 - i % 8)
+    end
+    send 'search_' + hash.to_s.downcase + '_ext', prefix, suffix, length, target.pack("C*"), mask.pack("C*"), chars, parallel
+  end
+
+  def search_suffix(hash, bin, length, prefix, suffix = '', chars = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789', parallel = true)
+    len = @@hashes.select{|a|a[0] == hash.to_s.downcase}[0]
+    raise ArgumentError.new('Unknown hash type: %s' % [hash.to_s.downcase]) unless len
+    len = len[1] / 8
+    raise ArgumentError.new('Too long prefix') if bin.size > len * 8
+    target = Array.new(len){0}
+    mask = Array.new(len){0}
+    bin.each_char.with_index do |b, i|
+      mask[-(i / 8) - 1] |= 1 << (i % 8)
+      target[-(i / 8) - 1] |= b.to_i << (i % 8)
+    end
+    send 'search_' + hash.to_s.downcase + '_ext', prefix, suffix, length, target.pack("C*"), mask.pack("C*"), chars, parallel
+  end
+
+  def easy_pow(bits, socket = nil)
+    in_s = out_s = socket
+    if socket == nil
+      in_s = STDIN
+      out_s = STDOUT
+    end
+    prefix = SecureRandom.hex(8)
+    out_s.print "Send me proof-of-work: The first #{bits}-bits of sha256(\"#{prefix}\" + input.rstrip) is \"111...1\"\n"
+    out_s.flush
+    input = in_s.gets.strip.force_encoding('ASCII-8BIT')
+    digest = Digest::SHA256.digest(prefix + input.rstrip).unpack("C*")
+    bits.times do |i|
+      if (digest[i / 8] >> (7 - i % 8) & 1) == 0
+        return false
+      end
+    end
+    return true
+  end
+
+  def solve(input)
+    if /The first (\d+)-bits of sha256\("(.{16})"/ =~ input
+      search_prefix('sha256', '1' * $1.to_i, 12, $2)[16..-1]
+    else
+      raise ArgumentError.new("Invalid format")
+    end
+  end
+
+  module_function :search_prefix, :search_suffix
+  module_function :easy_pow, :solve
+end

metadata

@@ -0,0 +1,90 @@
+--- !ruby/object:Gem::Specification
+name: easy_pow
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- nomeaning
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2017-04-25 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.14'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.14'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+description: 'Simple PoW for CTF '
+email:
+- nomeaning777@gmail.com
+executables: []
+extensions:
+- ext/easy_pow/extconf.rb
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- easy_pow.gemspec
+- ext/easy_pow/Makefile
+- ext/easy_pow/ext.c
+- ext/easy_pow/ext.o
+- ext/easy_pow/ext.so
+- ext/easy_pow/extconf.rb
+- ext/easy_pow/mkmf.log
+- lib/easy_pow.rb
+- lib/easy_pow/version.rb
+homepage: http://github.com/nomeaning777/easy_pow
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+- ext
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.6.8
+signing_key: 
+specification_version: 4
+summary: Simple PoW for CTF
+test_files: []