* Transport.cpp
* Copyright 1999-2000 Y.Takagi. All Rights Reserved.
*/
#include <FindDirectory.h>
#include <Message.h>
#include <Directory.h>
#include <DataIO.h>
#include <File.h>
#include <Path.h>
#include <image.h>
#include "Transport.h"
#include "PrinterData.h"
#include "DbgMsg.h"
using namespace std;
TransportException::TransportException(const string& what)
:
fWhat(what)
{
}
const char*
TransportException::What() const
{
return fWhat.c_str();
}
Transport::Transport(const PrinterData *printerData)
:
fImage(-1),
fInitTransport(0),
fExitTransport(0),
fDataStream(0),
fAbort(false)
{
const directory_which paths[] = {
B_USER_NONPACKAGED_ADDONS_DIRECTORY,
B_USER_ADDONS_DIRECTORY,
B_SYSTEM_NONPACKAGED_ADDONS_DIRECTORY,
B_BEOS_ADDONS_DIRECTORY,
};
BPath path;
for (uint32 i = 0; i < sizeof(paths) / sizeof(paths[0]); ++i) {
if (find_directory(paths[i], &path) != B_OK)
continue;
path.Append("Print/transport");
path.Append(printerData->GetTransport().c_str());
DBGMSG(("load_add_on: %s\n", path.Path()));
fImage = load_add_on(path.Path());
if (fImage >= 0)
break;
}
if (fImage < 0) {
SetLastError("cannot load a transport add-on");
return;
}
DBGMSG(("image id = %d\n", (int)fImage));
get_image_symbol(fImage, "init_transport", B_SYMBOL_TYPE_TEXT, (void **)&fInitTransport);
get_image_symbol(fImage, "exit_transport", B_SYMBOL_TYPE_TEXT, (void **)&fExitTransport);
if (fInitTransport == NULL) {
SetLastError("get_image_symbol failed.");
DBGMSG(("init_transport is NULL\n"));
}
if (fExitTransport == NULL) {
SetLastError("get_image_symbol failed.");
DBGMSG(("exit_transport is NULL\n"));
}
if (fInitTransport) {
string spool_path;
printerData->GetPath(spool_path);
BMessage *msg = new BMessage('TRIN');
msg->AddString("printer_file", spool_path.c_str());
fDataStream = (*fInitTransport)(msg);
delete msg;
if (fDataStream == 0) {
SetLastError("init_transport failed.");
}
}
}
Transport::~Transport()
{
if (fExitTransport) {
(*fExitTransport)();
}
if (fImage >= 0) {
unload_add_on(fImage);
}
}
bool
Transport::CheckAbort() const
{
return fDataStream == 0;
}
const
string &Transport::LastError() const
{
return fLastErrorString;
}
bool
Transport::IsPrintToFileCanceled() const
{
BFile* file = dynamic_cast<BFile*>(fDataStream);
return file != NULL && file->InitCheck() != B_OK;
}
void
Transport::SetLastError(const char *e)
{
fLastErrorString = e;
fAbort = true;
}
void
Transport::Write(const void* buffer, size_t size)
{
if (fDataStream) {
if (size == (size_t)fDataStream->Write(buffer, size)) {
return;
}
SetLastError("BDataIO::Write failed.");
}
throw TransportException(LastError());
}
void
Transport::Read(void* buffer, size_t size)
{
if (fDataStream) {
if (size == (size_t)fDataStream->Read(buffer, size)) {
return;
}
SetLastError("BDataIO::Read failed.");
}
throw TransportException(LastError());
}
