mirror of
https://github.com/mruby/mruby
synced 2026-06-08 16:11:16 +00:00
add mruby-strip tool to strip irep's debug info
This commit is contained in:
@@ -0,0 +1,53 @@
|
||||
require 'tempfile'
|
||||
|
||||
assert('no files') do
|
||||
o = `bin/mruby-strip 2>&1`
|
||||
assert_equal 1, $?.exitstatus
|
||||
assert_equal "no files to strip\n", o
|
||||
end
|
||||
|
||||
assert('file not found') do
|
||||
o = `bin/mruby-strip not_found.mrb 2>&1`
|
||||
assert_equal 1, $?.exitstatus
|
||||
assert_equal "can't open file not_found.mrb\n", o
|
||||
end
|
||||
|
||||
assert('not irep file') do
|
||||
t = Tempfile.new('script.rb')
|
||||
t.write 'p test\n'
|
||||
t.flush
|
||||
o = `bin/mruby-strip #{t.path} 2>&1`
|
||||
assert_equal 1, $?.exitstatus
|
||||
assert_equal "can't read irep file #{t.path}\n", o
|
||||
end
|
||||
|
||||
assert('success') do
|
||||
script_file, compiled1, compiled2 =
|
||||
Tempfile.new('script.rb'), Tempfile.new('c1.mrb'), Tempfile.new('c2.mrb')
|
||||
script_file.write "p 'test'\n"
|
||||
script_file.flush
|
||||
`bin/mrbc -g -o #{compiled1.path} #{script_file.path}`
|
||||
`bin/mrbc -g -o #{compiled2.path} #{script_file.path}`
|
||||
|
||||
o = `bin/mruby-strip #{compiled1.path}`
|
||||
assert_equal 0, $?.exitstatus
|
||||
assert_equal "", o
|
||||
|
||||
o = `bin/mruby-strip #{compiled1.path} #{compiled2.path}`
|
||||
assert_equal 0, $?.exitstatus
|
||||
assert_equal "", o
|
||||
end
|
||||
|
||||
assert('check debug section') do
|
||||
script_file, with_debug, without_debug =
|
||||
Tempfile.new('script.rb'), Tempfile.new('c1.mrb'), Tempfile.new('c2.mrb')
|
||||
script_file.write "p 'test'\n"
|
||||
script_file.flush
|
||||
`bin/mrbc -o #{without_debug.path} #{script_file.path}`
|
||||
`bin/mrbc -g -o #{with_debug.path} #{script_file.path}`
|
||||
|
||||
assert_true with_debug.size >= without_debug.size
|
||||
|
||||
`bin/mruby-strip #{with_debug.path}`
|
||||
assert_equal without_debug.size, with_debug.size
|
||||
end
|
||||
@@ -0,0 +1,5 @@
|
||||
MRuby::Gem::Specification.new('mruby-bin-strip') do |spec|
|
||||
spec.license = 'MIT'
|
||||
spec.author = 'mruby developers'
|
||||
spec.bins = %w(mruby-strip)
|
||||
end
|
||||
@@ -0,0 +1,58 @@
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include "mruby.h"
|
||||
#include "mruby/irep.h"
|
||||
#include "mruby/dump.h"
|
||||
|
||||
int
|
||||
main(int argc, char **argv)
|
||||
{
|
||||
int i, dump_result;
|
||||
FILE **files;
|
||||
mrb_irep **ireps;
|
||||
mrb_state *mrb;
|
||||
|
||||
if (argc <= 1) {
|
||||
fprintf(stderr, "no files to strip\n");
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
files = (FILE**)malloc(sizeof(FILE*) * argc);
|
||||
for (i = 1; i < argc; ++i) {
|
||||
files[i] = fopen(argv[i], "rb");
|
||||
|
||||
if (!files[i]) {
|
||||
fprintf(stderr, "can't open file %s\n", argv[i]);
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
}
|
||||
|
||||
mrb = mrb_open();
|
||||
|
||||
ireps = (mrb_irep**)malloc(sizeof(mrb_irep*) * argc);
|
||||
for (i = 1; i < argc; ++i) {
|
||||
ireps[i] = mrb_read_irep_file(mrb, files[i]);
|
||||
if (!ireps[i]) {
|
||||
fprintf(stderr, "can't read irep file %s\n", argv[i]);
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
fclose(files[i]);
|
||||
files[i] = fopen(argv[i], "wb");
|
||||
if (!ireps[i]) {
|
||||
fprintf(stderr, "can't reopen irep file %s\n", argv[i]);
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
}
|
||||
|
||||
for (i = 1; i < argc; ++i) {
|
||||
/* debug flag must be alway false */
|
||||
dump_result = mrb_dump_irep_binary(mrb, ireps[i], FALSE, files[i]);
|
||||
if (dump_result != MRB_DUMP_OK) {
|
||||
fprintf(stderr, "error occur when dumping %s", argv[i]);
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
}
|
||||
|
||||
mrb_close(mrb);
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
Reference in New Issue
Block a user