WZFile.java
上传用户:gwt600
上传日期:2021-06-03
资源大小:704k
文件大小:7k
源码类别:

游戏

开发平台:

Java

  1. /*
  2. This file is part of the OdinMS Maple Story Server
  3.     Copyright (C) 2008 Patrick Huy <patrick.huy@frz.cc> 
  4.                        Matthias Butz <matze@odinms.de>
  5.                        Jan Christian Meyer <vimes@odinms.de>
  6.     This program is free software: you can redistribute it and/or modify
  7.     it under the terms of the GNU Affero General Public License version 3
  8.     as published by the Free Software Foundation. You may not use, modify
  9.     or distribute this program under any other version of the
  10.     GNU Affero General Public License.
  11.     This program is distributed in the hope that it will be useful,
  12.     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13.     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14.     GNU Affero General Public License for more details.
  15.     You should have received a copy of the GNU Affero General Public License
  16.     along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17. */
  18. package net.sf.odinms.provider.wz;
  19. import java.io.BufferedInputStream;
  20. import java.io.File;
  21. import java.io.FileInputStream;
  22. import java.io.IOException;
  23. import java.io.RandomAccessFile;
  24. import net.sf.odinms.provider.MapleData;
  25. import net.sf.odinms.provider.MapleDataDirectoryEntry;
  26. import net.sf.odinms.provider.MapleDataFileEntry;
  27. import net.sf.odinms.provider.MapleDataProvider;
  28. import net.sf.odinms.tools.data.input.GenericLittleEndianAccessor;
  29. import net.sf.odinms.tools.data.input.GenericSeekableLittleEndianAccessor;
  30. import net.sf.odinms.tools.data.input.InputStreamByteStream;
  31. import net.sf.odinms.tools.data.input.LittleEndianAccessor;
  32. import net.sf.odinms.tools.data.input.RandomAccessByteStream;
  33. import net.sf.odinms.tools.data.input.SeekableLittleEndianAccessor;
  34. import org.slf4j.Logger;
  35. import org.slf4j.LoggerFactory;
  36. /*
  37.  * This is a rather straightforward port from Maplext xentax.com/uploads/author/mrmouse/Maplext.zip unfortunately I do
  38.  * not know who the original author is. In any case: Thanks, you rock.
  39.  */
  40. public class WZFile implements MapleDataProvider {
  41. static {
  42. ListWZFile.init();
  43. }
  44. private File wzfile;
  45. private LittleEndianAccessor lea;
  46. private SeekableLittleEndianAccessor slea;
  47. // private LittleEndianOutputStream leo;
  48. private Logger log = LoggerFactory.getLogger(WZFile.class);
  49. private int headerSize;
  50. private WZDirectoryEntry root;
  51. private boolean provideImages;
  52. private int cOffset;
  53. public WZFile(File wzfile, boolean provideImages) throws IOException {
  54. this.wzfile = wzfile;
  55. lea = new GenericLittleEndianAccessor(new InputStreamByteStream(new BufferedInputStream(new FileInputStream(wzfile))));
  56. RandomAccessFile raf = new RandomAccessFile(wzfile, "r");
  57. slea = new GenericSeekableLittleEndianAccessor(new RandomAccessByteStream(raf));
  58. root = new WZDirectoryEntry(wzfile.getName(), 0, 0, null);
  59. this.provideImages = provideImages;
  60. load();
  61. }
  62. @SuppressWarnings("unused")
  63. private void load() throws IOException {
  64. String sPKG = lea.readAsciiString(4);
  65. int size1 = lea.readInt();
  66. int size2 = lea.readInt();
  67. headerSize = lea.readInt();
  68. String copyright = lea.readNullTerminatedAsciiString();
  69. short version = lea.readShort();
  70. parseDirectory(root);
  71. cOffset = (int) lea.getBytesRead();
  72. getOffsets(root);
  73. }
  74. // private void writeHeader(short version) throws IOException { // pseudo header leo.writeBytes("PKG1");
  75. // leo.writeInt(0);
  76. // leo.writeInt(0);
  77. // leo.writeInt(0);
  78. // leo.writeBytes("Package file v1.0 Copyright OdinMS, Mtz");
  79. // leo.writeByte(0);
  80. // leo.writeShort(version);
  81. // writeDirectory(root);
  82. // cOffset = leo.size();
  83. // writeOffsets(root);
  84. // }
  85. private void getOffsets(MapleDataDirectoryEntry dir) {
  86. for (MapleDataFileEntry file : dir.getFiles()) {
  87. file.setOffset(cOffset);
  88. cOffset += file.getSize();
  89. }
  90. for (MapleDataDirectoryEntry sdir : dir.getSubdirectories()) {
  91. getOffsets(sdir);
  92. }
  93. }
  94. private void parseDirectory(WZDirectoryEntry dir) {
  95. int entries = WZTool.readValue(lea);
  96. for (int i = 0; i < entries; i++) {
  97. byte marker = lea.readByte();
  98. String name = null;
  99. @SuppressWarnings("unused")
  100. int dummyInt;
  101. int size, checksum;
  102. switch (marker) {
  103. case 0x02:
  104. name = WZTool.readDecodedStringAtOffset(slea, lea.readInt() + this.headerSize + 1,true);
  105. size = WZTool.readValue(lea);
  106. checksum = WZTool.readValue(lea);
  107. dummyInt = lea.readInt();
  108. dir.addFile(new WZFileEntry(name, size, checksum, dir));
  109. break;
  110. case 0x03:
  111. case 0x04:
  112. name = WZTool.readDecodedString(lea);
  113. size = WZTool.readValue(lea);
  114. checksum = WZTool.readValue(lea);
  115. dummyInt = lea.readInt();
  116. if (marker == 3) {
  117. dir.addDirectory(new WZDirectoryEntry(name, size, checksum, dir));
  118. } else {
  119. dir.addFile(new WZFileEntry(name, size, checksum, dir));
  120. }
  121. break;
  122. default:
  123. log.error("Default case in marker ({}):/", marker);
  124. }
  125. }
  126. for (MapleDataDirectoryEntry idir : dir.getSubdirectories()) {
  127. parseDirectory((WZDirectoryEntry) idir);
  128. }
  129. }
  130. // private void writeDirectory(MapleDataDirectoryEntry dir) {
  131. // // leo.writeInt(dir.getSize());
  132. //
  133. // for (int i = 0; i < dir.getSize(); i++) {
  134. // byte marker = lea.readByte();
  135. //
  136. // // if ()
  137. //
  138. // String name = null;
  139. // @SuppressWarnings("unused")
  140. // int dummyInt;
  141. // int size, checksum;
  142. //
  143. // switch (marker) {
  144. // case 0x02:
  145. // name = WZTool.readDecodedStringAtOffset(slea, lea.readInt() + this.headerSize + 1);
  146. // size = WZTool.readValue(lea);
  147. // checksum = WZTool.readValue(lea);
  148. // dummyInt = lea.readInt();
  149. // dir.addFile(new WZFileEntry(name, size, checksum));
  150. // break;
  151. //
  152. // case 0x03:
  153. // case 0x04:
  154. // name = WZTool.readDecodedString(lea);
  155. // size = WZTool.readValue(lea);
  156. // checksum = WZTool.readValue(lea);
  157. // dummyInt = lea.readInt();
  158. // if (marker == 3) {
  159. // dir.addDirectory(new WZDirectoryEntry(name, size, checksum));
  160. // } else {
  161. // dir.addFile(new WZFileEntry(name, size, checksum));
  162. // }
  163. // break;
  164. // default:
  165. // log.error("Default case in marker ({}):/", marker);
  166. // }
  167. // }
  168. //
  169. // for (MapleDataDirectoryEntry idir : dir.getSubdirectories()) {
  170. // parseDirectory(idir);
  171. // }
  172. // }
  173. public WZIMGFile getImgFile(String path) throws IOException {
  174. String segments[] = path.split("/");
  175. WZDirectoryEntry dir = root;
  176. for (int x = 0; x < segments.length - 1; x++) {
  177. dir = (WZDirectoryEntry) dir.getEntry(segments[x]);
  178. if (dir == null) {
  179. // throw new IllegalArgumentException("File " + path + " not found in " + root.getName());
  180. return null;
  181. }
  182. }
  183. WZFileEntry entry = (WZFileEntry) dir.getEntry(segments[segments.length - 1]);
  184. if (entry == null) {
  185. return null;
  186. }
  187. String fullPath = wzfile.getName().substring(0, wzfile.getName().length() - 3).toLowerCase() + "/" + path;
  188. return new WZIMGFile(this.wzfile, entry, provideImages, ListWZFile.isModernImgFile(fullPath));
  189. }
  190. // XXX see if we can prevent locking here without keeping multiple handles :/
  191. public synchronized MapleData getData(String path) {
  192. try {
  193. WZIMGFile imgFile = getImgFile(path);
  194. if (imgFile == null) {
  195. // throw new IllegalArgumentException("File " + path + " not found in " + root.getName());
  196. return null;
  197. }
  198. MapleData ret = imgFile.getRoot();
  199. return ret;
  200. } catch (IOException e) {
  201. log.error("THROW", e);
  202. }
  203. return null;
  204. }
  205. public MapleDataDirectoryEntry getRoot() {
  206. return root;
  207. }
  208. }